From e48663b03ad42ae5737c1198b92b1daee8202e12 Mon Sep 17 00:00:00 2001
From: Nikolay Turpitko <nikolay@turpitko.com>
Date: Thu, 21 Jan 2016 22:07:36 +0600
Subject: [PATCH] RESP writer test

---
 server/client_resp.go      |  4 ++
 server/client_resp_test.go | 75 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 79 insertions(+)
 create mode 100644 server/client_resp_test.go

diff --git a/server/client_resp.go b/server/client_resp.go
index 55e0db5..5719ad7 100644
--- a/server/client_resp.go
+++ b/server/client_resp.go
@@ -280,6 +280,10 @@ func (w *respWriter) writeArray(lst []interface{}) {
 				w.writeBulk(nil)
 			case int64:
 				w.writeInteger(v)
+			case string:
+				w.writeStatus(v)
+			case error:
+				w.writeError(v)
 			default:
 				panic(fmt.Sprintf("invalid array type %T %v", lst[i], v))
 			}
diff --git a/server/client_resp_test.go b/server/client_resp_test.go
new file mode 100644
index 0000000..9a96104
--- /dev/null
+++ b/server/client_resp_test.go
@@ -0,0 +1,75 @@
+package server
+
+import (
+	"bufio"
+	"bytes"
+	"errors"
+	"testing"
+)
+
+func TestRespWriter(t *testing.T) {
+	for _, fixture := range []struct {
+		v interface{}
+		e string
+	}{
+		{
+			v: errors.New("Some error"),
+			//e: "-Some error\r\n", // as described at http://redis.io/topics/protocol
+			e: "- Some error\r\n", // actual
+		},
+		{
+			v: "Some status",
+			e: "+Some status\r\n",
+		},
+		{
+			v: int64(42),
+			e: ":42\r\n",
+		},
+		{
+			v: []byte("ultimate answer"),
+			e: "$15\r\nultimate answer\r\n",
+		},
+		{
+			v: []interface{}{[]byte("aaa"), []byte("bbb"), int64(42)},
+			e: "*3\r\n$3\r\naaa\r\n$3\r\nbbb\r\n:42\r\n",
+		},
+		{
+			v: [][]byte{[]byte("test"), nil, []byte("zzz")},
+			e: "*3\r\n$4\r\ntest\r\n$-1\r\n$3\r\nzzz\r\n",
+		},
+		{
+			v: nil,
+			e: "$-1\r\n",
+		},
+		{
+			v: []interface{}{[]interface{}{int64(1), int64(2), int64(3)}, []interface{}{"Foo", errors.New("Bar")}},
+			//e: "*2\r\n*3\r\n:1\r\n:2\r\n:3\r\n*2\r\n+Foo\r\n-Bar\r\n",
+			e: "*2\r\n*3\r\n:1\r\n:2\r\n:3\r\n*2\r\n+Foo\r\n- Bar\r\n",
+		},
+	} {
+		w := new(respWriter)
+		var b bytes.Buffer
+		w.buff = bufio.NewWriter(&b)
+		switch v := fixture.v.(type) {
+		case error:
+			w.writeError(v)
+		case string:
+			w.writeStatus(v)
+		case int64:
+			w.writeInteger(v)
+		case []byte:
+			w.writeBulk(v)
+		case []interface{}:
+			w.writeArray(v)
+		case [][]byte:
+			w.writeSliceArray(v)
+		default:
+			w.writeBulk(b.Bytes())
+		}
+		w.flush()
+		if b.String() != fixture.e {
+			t.Errorf("respWriter, actual: %v, expected: %v", []byte(b.String()), []byte(fixture.e))
+		}
+	}
+
+}