From cf105501c7eafb6e08e46d9674de61e3e849a45a Mon Sep 17 00:00:00 2001
From: Kyren223 <ulmliad223@gmail.com>
Date: Wed, 16 Oct 2024 16:04:26 +0300
Subject: test: add test for framer

---
 internal/packet/framer_test.go | 63 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 internal/packet/framer_test.go

(limited to 'internal')

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
+	}
+}
-- 
cgit v1.3.1