test: Handle kB units in size_test.go (#5900)

hypermodeinc · Jul 23, 2020 · 80e6c19 · 80e6c19
1 parent b89e355
commit 80e6c19
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/posting/size_test.go b/posting/size_test.go
@@ -218,7 +218,8 @@ func Test21MillionDataSetSize(t *testing.T) {
 func filterUnit(line string) (string, error) {
 	words := strings.Split(line, " ")
 	for _, word := range words {
-		if strings.Contains(word, "MB") || strings.Contains(word, "GB") {
+		if strings.Contains(word, "MB") || strings.Contains(word, "GB") ||
+			strings.Contains(word, "kB") {
 			return strings.TrimSpace(word), nil
 		}
 	}
@@ -227,6 +228,13 @@ func filterUnit(line string) (string, error) {
 
 // convertToBytes converts the unit into bytes.
 func convertToBytes(unit string) (uint32, error) {
+	if strings.Contains(unit, "kB") {
+		kb, err := strconv.ParseFloat(unit[0:len(unit)-2], 64)
+		if err != nil {
+			return 0, err
+		}
+		return uint32(kb * 1024.0), nil
+	}
 	if strings.Contains(unit, "MB") {
 		mb, err := strconv.ParseFloat(unit[0:len(unit)-2], 64)
 		if err != nil {