added ui/test package for test helpers

Signed-off-by: Michael Maximilien <[email protected]>

Author: cppforlife

ui/json_ui.go

@@ -11,19 +11,19 @@ import (
 
 type JSONUI struct {
 	parent UI
-	uiResp uiResp
+	uiResp JSONUIResp
 
 	logTag string
 	logger ExternalLogger
 }
 
-type uiResp struct {
-	Tables []tableResp
+type JSONUIResp struct {
+	Tables []JSONUITableResp
 	Blocks []string
 	Lines  []string
 }
 
-type tableResp struct {
+type JSONUITableResp struct {
 	Content string
 	Header  map[string]string
 	Rows    []map[string]string
@@ -88,7 +88,7 @@ func (ui *JSONUI) PrintTable(table Table) {
 		table.Header = rawHeaders
 	}
 
-	resp := tableResp{
+	resp := JSONUITableResp{
 		Content: table.Content,
 		Header:  header,
 		Rows:    ui.stringRows(table.Header, table.AsRows()),
@@ -121,7 +121,7 @@ func (ui *JSONUI) IsInteractive() bool {
 func (ui *JSONUI) Flush() {
 	defer ui.parent.Flush()
 
-	if !reflect.DeepEqual(ui.uiResp, uiResp{}) {
+	if !reflect.DeepEqual(ui.uiResp, JSONUIResp{}) {
 		bytes, err := json.MarshalIndent(ui.uiResp, "", "    ")
 		if err != nil {
 			ui.logger.Error(ui.logTag, "Failed to marshal UI response")

ui/test/json_ui.go

@@ -0,0 +1,19 @@
+package test
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/cppforlife/go-cli-ui/ui"
+)
+
+func JSONUIFromBytes(t *testing.T, bytes []byte) ui.JSONUIResp {
+	resp := ui.JSONUIResp{}
+
+	err := json.Unmarshal(bytes, &resp)
+	if err != nil {
+		t.Fatalf("Expected to successfully unmarshal JSON UI: %s", err)
+	}
+
+	return resp
+}

ui/test/json_ui_test.go

@@ -0,0 +1,80 @@
+package test_test
+
+import (
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
+	"github.com/cppforlife/go-cli-ui/ui"
+	. "github.com/cppforlife/go-cli-ui/ui/test"
+)
+
+var _ = Describe("JSONUIFromBytes", func() {
+	const (
+		example = `
+{
+    "Tables": [
+        {
+            "Content": "services",
+            "Header": {
+                "created_at": "Created At",
+                "domain": "Domain",
+                "internal_domain": "Internal Domain",
+                "name": "Name"
+            },
+            "Rows": [
+                {
+                    "created_at": "2018-07-31T12:27:45-07:00",
+                    "domain": "foo3.default.example.com",
+                    "internal_domain": "foo3.default.svc.cluster.local",
+                    "name": "foo3"
+                },
+                {
+                    "created_at": "2018-07-26T16:47:39-07:00",
+                    "domain": "helloworld-go.default.example.com",
+                    "internal_domain": "helloworld-go.default.svc.cluster.local",
+                    "name": "helloworld-go"
+                }
+            ],
+            "Notes": null
+        }
+    ],
+    "Blocks": null,
+    "Lines": null
+}
+`
+	)
+
+	It("properly parses JSON UI output", func() {
+		resp := JSONUIFromBytes(nil, []byte(example))
+		Expect(resp).To(Equal(ui.JSONUIResp{
+			Tables: []ui.JSONUITableResp{
+				{
+					Content: "services",
+					Header: map[string]string{
+						"domain":          "Domain",
+						"internal_domain": "Internal Domain",
+						"name":            "Name",
+						"created_at":      "Created At",
+					},
+					Rows: []map[string]string{
+						{
+							"internal_domain": "foo3.default.svc.cluster.local",
+							"name":            "foo3",
+							"created_at":      "2018-07-31T12:27:45-07:00",
+							"domain":          "foo3.default.example.com",
+						},
+						{
+							"name":            "helloworld-go",
+							"created_at":      "2018-07-26T16:47:39-07:00",
+							"domain":          "helloworld-go.default.example.com",
+							"internal_domain": "helloworld-go.default.svc.cluster.local",
+						},
+					},
+					Notes: nil,
+				},
+			},
+			Blocks: nil,
+			Lines:  nil,
+		}))
+	})
+})

ui/test/suite_test.go

@@ -0,0 +1,12 @@
+package test_test
+
+import (
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"testing"
+)
+
+func TestReg(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "ui/test")
+}