commit a5e7e453bb39396c1b019f81c28c3d6cb052389d Author: David Fifield david@bamsoftware.com Date: Tue Nov 26 23:27:21 2013 -0800
Add tests for extOrPortSendTransport. --- pt_test.go | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+)
diff --git a/pt_test.go b/pt_test.go index 5628a96..4579fbd 100644 --- a/pt_test.go +++ b/pt_test.go @@ -471,6 +471,35 @@ func TestExtOrSendUserAddr(t *testing.T) { } }
+func TestExtOrPortSendTransport(t *testing.T) { + tests := [...]struct { + methodName string + expected []byte + }{ + {"", []byte("\x00\x02\x00\x00")}, + {"a", []byte("\x00\x02\x00\x01a")}, + {"alpha", []byte("\x00\x02\x00\x05alpha")}, + } + + for _, test := range tests { + var buf bytes.Buffer + err := extOrPortSendTransport(&buf, test.methodName) + if err != nil { + t.Errorf("%q unexpectedly returned an error: %s", test.methodName, err) + } + p := make([]byte, 1024) + n, err := buf.Read(p) + if err != nil { + t.Fatal(err) + } + output := p[:n] + if !bytes.Equal(output, test.expected) { + t.Errorf("%q → %s (expected %s)", test.methodName, + fmtBytes(output), fmtBytes(test.expected)) + } + } +} + func TestExtOrPortSendDone(t *testing.T) { expected := []byte("\x00\x00\x00\x00")