test: add test for framer

1 files changed, 63 insertions, 0 deletions

diff --git a/internal/packet/framer_test.go b/internal/packet/framer_test.go
new file mode 100644
index 0000000..41a4396
--- /dev/null
+++ b/internal/packet/framer_test.go
@@ -0,0 +1,63 @@
+package packet
+
+import (
+	"context"
+	"io"
+	"testing"
+	"time"
+)
+
+type TestIoReader struct {
+	data []byte
+	len  int
+}
+
+func (r *TestIoReader) Read(data []byte) (int, error) {
+	if len(r.data) <= r.len {
+		// log.Println("RETURNING EOF:", len(data), r.len)
+		return 0, io.EOF
+	}
+	n := copy(data, r.data[r.len:])
+	r.len += n
+	// log.Println("RETURNING N:", len(r.data), r.len)
+	return n, nil
+}
+
+func (r *TestIoReader) start(msg TypedMessage, t *testing.T) {
+	encoder, err := NewMsgPackEncoder(msg)
+	if err != nil {
+		t.Errorf("encoding error: %v", err)
+		return
+	}
+	r.data = NewPacket(encoder).data
+	// log.Println("LEN:", len(r.data))
+}
+
+func TestPacketFramer(t *testing.T) {
+	reader := &TestIoReader{}
+
+	// Long msg to test multiple
+	msg := EkoMessage{"Testing FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting FramerTesting Framer"}
+	reader.start(&msg, t)
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	out, outErr := RunFramer(ctx, reader)
+
+	var message EkoMessage
+	select {
+	case packet := <-out:
+		if err := packet.DecodePayload(&message); err != nil {
+			t.Errorf("error decoding response: %v", err)
+			return
+		}
+		if msg.Message != message.Message {
+			t.Errorf("%v != %v", msg.Message, message.Message)
+			return
+		}
+
+	case err := <-outErr:
+		t.Errorf("error receiving packet: %v", err)
+		return
+	}
+}