Support batched relay,unrelay,local,unlocal messages

websocket/version_test.go

@@ -0,0 +1,47 @@
+package websocket
+
+import "testing"
+
+func TestSupportsBatchedSiteMessages(t *testing.T) {
+	cases := []struct {
+		version string
+		want    bool
+	}{
+		{"", false},
+		{"1.23.0", false},
+		{"1.23.9", false},
+		{"1.24.0", true},
+		{"1.24.1", true},
+		{"1.25.0", true},
+		{"2.0.0", true},
+		{"1.24.0-s.5", true},   // cloud build of a supported base version
+		{"1.23.0-s.99", false}, // cloud build of an unsupported base version
+		{"not-a-version", false},
+	}
+
+	for _, c := range cases {
+		if got := supportsBatchedSiteMessages(c.version); got != c.want {
+			t.Errorf("supportsBatchedSiteMessages(%q) = %v, want %v", c.version, got, c.want)
+		}
+	}
+}
+
+func TestCompareVersions(t *testing.T) {
+	cases := []struct {
+		a, b string
+		want int
+	}{
+		{"1.24.0", "1.24.0", 0},
+		{"1.23.0", "1.24.0", -1},
+		{"1.24.0", "1.23.0", 1},
+		{"1.24", "1.24.0", 0},
+		{"1.24.1", "1.24", 1},
+		{"2.0.0", "1.99.99", 1},
+	}
+
+	for _, c := range cases {
+		if got := compareVersions(c.a, c.b); got != c.want {
+			t.Errorf("compareVersions(%q, %q) = %d, want %d", c.a, c.b, got, c.want)
+		}
+	}
+}