Add tests for GraphicsCommand serialization

tools/tui/graphics/command_test.go

@@ -0,0 +1,103 @@
+// License: GPLv3 Copyright: 2022, Kovid Goyal, <kovid at kovidgoyal.net>
+
+package graphics
+
+import (
+	"bytes"
+	"compress/zlib"
+	"encoding/base64"
+	"fmt"
+	"io"
+	"strings"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"golang.org/x/exp/rand"
+)
+
+var _ = fmt.Print
+
+func from_full_apc_escape_code(raw string) *GraphicsCommand {
+	return GraphicsCommandFromAPC([]byte(raw[2 : len(raw)-2]))
+}
+
+func TestGraphicsCommandSerialization(t *testing.T) {
+	gc := &GraphicsCommand{}
+
+	test_serialize := func(payload string, vals ...string) {
+		expected := "\033_G" + strings.Join(vals, ",")
+		if payload != "" {
+			expected += ";" + base64.StdEncoding.EncodeToString([]byte(payload))
+		}
+		expected += "\033\\"
+		if diff := cmp.Diff(expected, gc.AsAPC([]byte(payload))); diff != "" {
+			t.Fatalf("Failed to write vals: %#v with payload: %#v\n%s", vals, payload, diff)
+		}
+	}
+
+	test_chunked_payload := func(payload []byte) {
+		c := &GraphicsCommand{}
+		data := c.AsAPC([]byte(payload))
+		encoded := strings.Builder{}
+		compressed := false
+		var data_size uint64
+		for {
+			idx := strings.Index(data, "\033_")
+			if idx < 0 {
+				break
+			}
+			l := strings.Index(data, "\033\\")
+			apc := data[idx+2 : l]
+			data = data[l+2:]
+			g := GraphicsCommandFromAPC([]byte(apc))
+			if data_size == 0 {
+				data_size = g.DataSize()
+				compressed = g.Compression() != 0
+			}
+			encoded.WriteString(g.ResponseMessage())
+			if g.m == GRT_more_nomore {
+				break
+			}
+		}
+		if len(data) > 0 {
+			t.Fatalf("Unparsed remnant: %#v", string(data))
+		}
+		decoded, err := base64.StdEncoding.DecodeString(encoded.String())
+		if err != nil {
+			t.Fatalf("Encoded data not valid base-64 with error: %v", err)
+		}
+		if data_size > 0 && uint64(len(decoded)) != data_size {
+			t.Fatalf("Data size %d != decoded size %d", data_size, len(decoded))
+		}
+		if compressed {
+			b := bytes.Buffer{}
+			b.Write(decoded)
+			r, _ := zlib.NewReader(&b)
+			o := bytes.Buffer{}
+			io.Copy(&o, r)
+			r.Close()
+			decoded = o.Bytes()
+		}
+		if diff := cmp.Diff(payload, decoded); diff != "" {
+			t.Fatalf("Decoded payload does not match original\nlen decoded=%d len payload=%d", len(decoded), len(payload))
+		}
+	}
+
+	test_serialize("")
+	gc.SetTransmission(GRT_transmission_sharedmem).SetAction(GRT_action_query).SetZIndex(-3).SetWidth(33).SetImageId(11)
+	test_serialize("abcd", "a=q", "t=s", "w=33", "i=11", "z=-3")
+	q := from_full_apc_escape_code(gc.AsAPC([]byte("abcd")))
+	if diff := cmp.Diff(gc.AsAPC(nil), q.AsAPC(nil)); diff != "" {
+		t.Fatalf("Parsing failed:\n%s", diff)
+	}
+	if diff := cmp.Diff(q.response_message, base64.StdEncoding.EncodeToString([]byte("abcd"))); diff != "" {
+		t.Fatalf("Failed to parse payload:\n%s", diff)
+	}
+
+	test_chunked_payload([]byte("abcd"))
+	data := make([]byte, 8111)
+	rand.Read(data)
+	test_chunked_payload(data)
+	test_chunked_payload([]byte(strings.Repeat("a", 8007)))
+
+}