refactor: mid refactor

1 files changed, 58 insertions, 0 deletions

diff --git a/internal/packet/packet_test.go b/internal/packet/packet_test.go
index 20370af..a523d59 100644
--- a/internal/packet/packet_test.go
+++ b/internal/packet/packet_test.go
@@ -1,7 +1,10 @@
 package packet
 
 import (
+	"context"
+	"io"
 	"testing"
+	"time"
 
 	"github.com/vmihailenco/msgpack/v5"
 )
@@ -48,3 +51,58 @@ func TestPacketMsgPackEncoding(t *testing.T) {
 		return
 	}
 }
+
+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
+	}
+}