cmp: token builder

qb/token_test.go

@@ -0,0 +1,82 @@
+package qb
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func TestToken(t *testing.T) {
+	table := []struct {
+		C Cmp
+		S string
+		N []string
+	}{
+		// Basic comparators
+		{
+			C: Token("a", "b").Eq(),
+			S: "token(a,b)=token(?,?)",
+			N: []string{"a", "b"},
+		},
+		{
+			C: Token("a", "b").Lt(),
+			S: "token(a,b)<token(?,?)",
+			N: []string{"a", "b"},
+		},
+		{
+			C: Token("a", "b").LtOrEq(),
+			S: "token(a,b)<=token(?,?)",
+			N: []string{"a", "b"},
+		},
+		{
+			C: Token("a", "b").Gt(),
+			S: "token(a,b)>token(?,?)",
+			N: []string{"a", "b"},
+		},
+		{
+			C: Token("a", "b").GtOrEq(),
+			S: "token(a,b)>=token(?,?)",
+			N: []string{"a", "b"},
+		},
+
+		// Custom bind names
+		{
+			C: Token("a", "b").EqNamed("c", "d"),
+			S: "token(a,b)=token(?,?)",
+			N: []string{"c", "d"},
+		},
+		{
+			C: Token("a", "b").LtNamed("c", "d"),
+			S: "token(a,b)<token(?,?)",
+			N: []string{"c", "d"},
+		},
+		{
+			C: Token("a", "b").LtOrEqNamed("c", "d"),
+			S: "token(a,b)<=token(?,?)",
+			N: []string{"c", "d"},
+		},
+		{
+			C: Token("a", "b").GtNamed("c", "d"),
+			S: "token(a,b)>token(?,?)",
+			N: []string{"c", "d"},
+		},
+		{
+			C: Token("a", "b").GtOrEqNamed("c", "d"),
+			S: "token(a,b)>=token(?,?)",
+			N: []string{"c", "d"},
+		},
+	}
+
+	buf := bytes.Buffer{}
+	for _, test := range table {
+		buf.Reset()
+		name := test.C.writeCql(&buf)
+		if diff := cmp.Diff(test.S, buf.String()); diff != "" {
+			t.Error(diff)
+		}
+		if diff := cmp.Diff(test.N, name); diff != "" {
+			t.Error(diff)
+		}
+	}
+}