1 files changed, 95 insertions, 0 deletions
diff --git a/internal/config/args_test.go b/internal/config/args_test.go
new file mode 100644
index 0000000..5e351a5
--- /dev/null
+++ b/internal/config/args_test.go
@@ -0,0 +1,95 @@
+package config
+
+import (
+	"encoding/base64"
+	"strings"
+	"testing"
+
+	"github.com/mimecast/dtail/internal/lcontext"
+)
+
+func TestSerializeOptionsUsesStableOrdering(t *testing.T) {
+	args := Args{
+		LContext:   lcontextForTest(3, 1, 2),
+		Plain:      true,
+		Quiet:      true,
+		Serverless: true,
+	}
+
+	got := args.SerializeOptions()
+	want := "after=2:before=1:max=3:plain=true:quiet=true:serverless=true"
+	if got != want {
+		t.Fatalf("unexpected serialized options:\nwant %q\ngot  %q", want, got)
+	}
+}
+
+func TestSerializeOptionsRoundTripsReservedValues(t *testing.T) {
+	unsafeValue := "a:b=c|d"
+	encodedValue := "base64%" + base64.StdEncoding.EncodeToString([]byte(unsafeValue))
+
+	got := serializeOptions(map[string]string{
+		"plain": "true",
+		"note":  unsafeValue,
+		"quiet": "false",
+	})
+
+	want := strings.Join([]string{
+		"note=" + encodedValue,
+		"plain=true",
+		"quiet=false",
+	}, ":")
+	if got != want {
+		t.Fatalf("unexpected serialized options:\nwant %q\ngot  %q", want, got)
+	}
+
+	options, ltx, err := DeserializeOptions(strings.Split(got, ":"))
+	if err != nil {
+		t.Fatalf("DeserializeOptions failed: %v", err)
+	}
+	if ltx != (lcontext.LContext{}) {
+		t.Fatalf("unexpected lcontext: %#v", ltx)
+	}
+	if options["note"] != unsafeValue {
+		t.Fatalf("expected note to round-trip, got %q", options["note"])
+	}
+	if options["plain"] != "true" {
+		t.Fatalf("expected plain to round-trip, got %q", options["plain"])
+	}
+	if options["quiet"] != "false" {
+		t.Fatalf("expected quiet to round-trip, got %q", options["quiet"])
+	}
+}
+
+func TestDeserializeOptionsAcceptsRawSerializedBlob(t *testing.T) {
+	unsafeValue := "a:b=c|d"
+	serialized := serializeOptions(map[string]string{
+		"note":  unsafeValue,
+		"plain": "true",
+		"quiet": "false",
+	})
+
+	options, ltx, err := DeserializeOptions([]string{serialized})
+	if err != nil {
+		t.Fatalf("DeserializeOptions failed: %v", err)
+	}
+	if ltx != (lcontext.LContext{}) {
+		t.Fatalf("unexpected lcontext: %#v", ltx)
+	}
+	if options["note"] != unsafeValue {
+		t.Fatalf("expected note to round-trip, got %q", options["note"])
+	}
+	if options["plain"] != "true" {
+		t.Fatalf("expected plain to round-trip, got %q", options["plain"])
+	}
+	if options["quiet"] != "false" {
+		t.Fatalf("expected quiet to round-trip, got %q", options["quiet"])
+	}
+}
+
+func lcontextForTest(max, before, after int) lcontext.LContext {
+	return lcontext.LContext{
+		MaxCount:      max,
+		BeforeContext: before,
+		AfterContext:  after,
+	}
+}