From 80e6c199e6af823d5fd08eed0801e29dd87044fc Mon Sep 17 00:00:00 2001
From: Martin Martinez Rivera <martinmr@dgraph.io>
Date: Thu, 23 Jul 2020 11:17:00 -0700
Subject: [PATCH] test: Handle kB units in size_test.go (#5900)
---
posting/size_test.go | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/posting/size_test.go b/posting/size_test.go
index cebed31f835..9e24cf2335f 100644
--- 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 {