]> git.giacomolli.uk - onechat/blobdiff - daemon_test.go
Select notification targets by stable ID
[onechat] / daemon_test.go
index a32977890b31777e5075690a359bb5769f83ac52..5670f6fe429bfb3c27a44f94cdc2f002baa084d8 100644 (file)
@@ -85,6 +85,18 @@ func TestNotificationSocketRejectsMissingOrInvalidData(t *testing.T) {
                        ID:     "1",
                        Text:   "data",
                },
+               {
+                       AgentID: "Garden",
+                       Source:  "check",
+                       ID:      "1",
+                       Text:    "data",
+               },
+               {
+                       AgentID: strings.Repeat("f", 2*agentIDBytes),
+                       Source:  "check",
+                       ID:      "1",
+                       Text:    "data",
+               },
        }
 
        for _, tc := range testCases {
@@ -102,6 +114,11 @@ func TestNotificationSocketRejectsMissingOrInvalidData(t *testing.T) {
        require.NoError(t, decoder.Decode(&event))
        require.Equal(t, "error", event.Kind)
 
+       _, err = io.WriteString(conn, `{"type":"notify","notify":{"agent":"Garden","source":"check","id":"1","text":"data"}}`+"\n")
+       require.NoError(t, err)
+       require.NoError(t, decoder.Decode(&event))
+       require.Equal(t, "error", event.Kind)
+
        require.Zero(t, server.Calls())
        require.NoError(t, conn.Close())
 }