adding quickfix

quickfix/connection_internal_test.go

@@ -0,0 +1,73 @@
+// Copyright (c) quickfixengine.org  All rights reserved.
+//
+// This file may be distributed under the terms of the quickfixengine.org
+// license as defined by quickfixengine.org and appearing in the file
+// LICENSE included in the packaging of this file.
+//
+// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING
+// THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A
+// PARTICULAR PURPOSE.
+//
+// See http://www.quickfixengine.org/LICENSE for licensing information.
+//
+// Contact ask@quickfixengine.org if any conditions of this licensing
+// are not clear to you.
+
+package quickfix
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+)
+
+func TestWriteLoop(t *testing.T) {
+	writer := bytes.NewBufferString("")
+	msgOut := make(chan []byte)
+
+	go func() {
+		msgOut <- []byte("test msg 1 ")
+		msgOut <- []byte("test msg 2 ")
+		msgOut <- []byte("test msg 3")
+		close(msgOut)
+	}()
+	writeLoop(writer, msgOut, nullLog{})
+
+	expected := "test msg 1 test msg 2 test msg 3"
+
+	if writer.String() != expected {
+		t.Errorf("expected %v got %v", expected, writer.String())
+	}
+}
+
+func TestReadLoop(t *testing.T) {
+	msgIn := make(chan fixIn)
+	stream := "hello8=FIX.4.0
9=5
blah
10=103
garbage8=FIX.4.0
9=4
foo
10=103
"
+
+	parser := newParser(strings.NewReader(stream))
+	go readLoop(parser, msgIn, nullLog{})
+
+	var tests = []struct {
+		expectedMsg   string
+		channelClosed bool
+	}{
+		{expectedMsg: "8=FIX.4.0
9=5
blah
10=103
"},
+		{expectedMsg: "8=FIX.4.0
9=4
foo
10=103
"},
+		{channelClosed: true},
+	}
+
+	for _, test := range tests {
+		msg, ok := <-msgIn
+		switch {
+		case !ok && !test.channelClosed:
+			t.Error("Channel unexpectedly closed")
+			fallthrough
+		case !ok && test.channelClosed:
+			continue
+		}
+
+		if msg.bytes.String() != test.expectedMsg {
+			t.Errorf("Expected %v got %v", test.expectedMsg, msg.bytes.String())
+		}
+	}
+}