Add some checking of channel ID before sending websocket event (#7431)

1 files changed, 35 insertions, 0 deletions

diff --git a/model/utils_test.go b/model/utils_test.go
index bc2aa6ce7..fd333b40c 100644
--- a/model/utils_test.go
+++ b/model/utils_test.go
@@ -331,3 +331,38 @@ func TestIsValidAlphaNumHyphenUnderscore(t *testing.T) {
 		}
 	}
 }
+
+func TestIsValidId(t *testing.T) {
+	cases := []struct {
+		Input  string
+		Result bool
+	}{
+		{
+			Input:  NewId(),
+			Result: true,
+		},
+		{
+			Input:  "",
+			Result: false,
+		},
+		{
+			Input:  "junk",
+			Result: false,
+		},
+		{
+			Input:  "qwertyuiop1234567890asdfg{",
+			Result: false,
+		},
+		{
+			Input:  NewId() + "}",
+			Result: false,
+		},
+	}
+
+	for _, tc := range cases {
+		actual := IsValidId(tc.Input)
+		if actual != tc.Result {
+			t.Fatalf("case: %v\tshould returned: %#v", tc, tc.Result)
+		}
+	}
+}