From 356f1cfd1777f1537f1dc0705836e68cfd309af5 Mon Sep 17 00:00:00 2001
From: Alan Bernstein <alan@pilosa.com>
Date: Mon, 12 Dec 2016 22:17:56 -0600
Subject: [PATCH] Add test for prettify

---
 bench/prettify_test.go | 58 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 bench/prettify_test.go

diff --git a/bench/prettify_test.go b/bench/prettify_test.go
new file mode 100644
index 000000000..4da62895b
--- /dev/null
+++ b/bench/prettify_test.go
@@ -0,0 +1,58 @@
+package bench_test
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/pilosa/pilosa/bench"
+)
+
+func TestPrettify(t *testing.T) {
+	res := make(map[string]interface{}, 1)
+	res["0"] = map[string]interface{}{
+		"avg":   time.Duration(12345),
+		"max":   time.Duration(23456),
+		"int":   3456,
+		"slice": []time.Duration{123, 234},
+	}
+	res["runtimes"] = map[string]interface{}{
+		"0":     time.Duration(10e9),
+		"total": time.Duration(10e9),
+	}
+	resPretty := bench.Prettify(res)
+
+	jsonString := new(bytes.Buffer)
+	enc := json.NewEncoder(jsonString)
+	enc.SetIndent("", "  ")
+
+	err := enc.Encode(resPretty)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err)
+	}
+
+	expected := `
+{
+  "0": {
+    "avg": "12.345µs",
+    "max": "23.456µs",
+    "slice": [
+      "123ns",
+      "234ns"
+    ],
+    "int": 3456
+  },
+  "runtimes": {
+    "0": "10s",
+    "total": "10s"
+  }
+}
+`[1:]
+
+	if jsonString.String() != expected {
+		t.Fatalf("failure")
+	}
+}