1 files changed, 28 insertions, 0 deletions
diff --git a/network/protocol/pktline/encode_length_header_test.go b/network/protocol/pktline/encode_length_header_test.go
new file mode 100644
index 00000000..38a980f0
--- /dev/null
+++ b/network/protocol/pktline/encode_length_header_test.go
@@ -0,0 +1,28 @@
+package pktline_test
+
+import (
+	"errors"
+	"testing"
+
+	"codeberg.org/lindenii/furgit/network/protocol/pktline"
+)
+
+func TestEncodeLengthHeader(t *testing.T) {
+	t.Parallel()
+
+	var hdr [4]byte
+
+	err := pktline.EncodeLengthHeader(&hdr, 4)
+	if err != nil {
+		t.Fatalf("unexpected error: %v", err)
+	}
+
+	if got := string(hdr[:]); got != "0004" {
+		t.Fatalf("got %q, want %q", got, "0004")
+	}
+
+	err = pktline.EncodeLengthHeader(&hdr, pktline.LargePacketMax+1)
+	if !errors.Is(err, pktline.ErrInvalidLength) {
+		t.Fatalf("got err %v, want ErrInvalidLength", err)
+	}
+}