X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/98d137c3e945a09f76e85bd18238558c668fdca4..f4509310b5abe5012cb83bed3960404e8fa92217:/daemon_test.go

diff --git a/daemon_test.go b/daemon_test.go
index a329778..5670f6f 100644
--- a/daemon_test.go
+++ b/daemon_test.go
@@ -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())
 }