adds tests on response's writer and recorder

conn.go

@@ -16,18 +16,11 @@ type conn struct {
 }
 
 func (c *conn) bySegment(segment string, body io.Reader) {
-	w := c.newResponseWriter()
+	w := newWriter(c.rwc)
 	req := c.newRequest(segment, body)
 	c.srv.ServeTCP(w, req)
 }
 
-func (c *conn) newResponseWriter() *responseWriter {
-	return &responseWriter{
-		ResponseWriter: c.rwc,
-		size:           noWritten,
-	}
-}
-
 func (c *conn) newRequest(segment string, body io.Reader) *Request {
 	req := NewRequest(segment, body)
 	req.RemoteAddr = c.addr

response.go

@@ -13,6 +13,13 @@ type ResponseWriter interface {
 	io.WriteCloser
 }
 
+func newWriter(wc io.WriteCloser) *responseWriter {
+	return &responseWriter{
+		ResponseWriter: wc,
+		size:           noWritten,
+	}
+}
+
 type responseWriter struct {
 	ResponseWriter io.WriteCloser
 	size           int

response_internal_test.go

@@ -0,0 +1,61 @@
+package tcp
+
+import (
+	"testing"
+
+	"github.com/matryer/is"
+)
+
+func TestResponseWriter_Write(t *testing.T) {
+	var (
+		are = is.New(t)
+		dt  = []struct {
+			msg       string
+			len, size int
+		}{
+			{msg: "hi", len: 2, size: 2},
+			{msg: " ", len: 1, size: 3},
+			{msg: "world!", len: 6, size: 9},
+			{msg: "\n", len: 1, size: 10},
+		}
+		w   = newWriter(NewRecorder())
+		n   int
+		err error
+	)
+	for _, tt := range dt {
+		t.Run(tt.msg, func(t *testing.T) {
+			n, err = w.Write([]byte(tt.msg))
+			are.NoErr(err)
+			are.Equal(n, tt.len)         // len mismatch
+			are.Equal(w.Size(), tt.size) // size mismatch
+		})
+	}
+	are.NoErr(w.Close())
+}
+
+func TestResponseWriter_WriteString(t *testing.T) {
+	var (
+		are = is.New(t)
+		dt  = []struct {
+			msg       string
+			len, size int
+		}{
+			{msg: "hi", len: 2, size: 2},
+			{msg: " ", len: 1, size: 3},
+			{msg: "world!", len: 6, size: 9},
+			{msg: "\n", len: 1, size: 10},
+		}
+		w   = newWriter(NewRecorder())
+		n   int
+		err error
+	)
+	for _, tt := range dt {
+		t.Run(tt.msg, func(t *testing.T) {
+			n, err = w.WriteString(tt.msg)
+			are.NoErr(err)
+			are.Equal(n, tt.len)         // len mismatch
+			are.Equal(w.Size(), tt.size) // size mismatch
+		})
+	}
+	are.NoErr(w.Close())
+}

response_test.go

@@ -0,0 +1,46 @@
+package tcp_test
+
+import (
+	"testing"
+
+	"github.com/matryer/is"
+	"github.com/rvflash/tcp"
+)
+
+const (
+	msg     = "hello world\n"
+	msgSize = 12
+)
+
+func TestResponseRecorder_Close(t *testing.T) {
+	is.New(t).True(tcp.NewRecorder().Close() == nil)
+}
+
+func TestResponseRecorder_Write(t *testing.T) {
+	w := tcp.NewRecorder()
+	n, err := w.Write([]byte(msg))
+	are := is.New(t)
+	are.NoErr(err)
+	are.Equal(n, msgSize)
+	are.Equal(w.Size(), msgSize)
+}
+
+func TestResponseRecorder_WriteString(t *testing.T) {
+	w := tcp.NewRecorder()
+	n, err := w.WriteString(msg)
+	are := is.New(t)
+	are.NoErr(err)
+	are.Equal(n, msgSize)
+	are.Equal(w.Size(), msgSize)
+}
+
+func TestResponseRecorder_Size(t *testing.T) {
+	are := is.New(t)
+	// no body
+	w := &tcp.ResponseRecorder{}
+	are.Equal(w.Size(), -1)
+	// with a valid body
+	w = tcp.NewRecorder()
+	are.Equal(w.Size(), 0)
+
+}