fix(ws): fix api tests

Signed-off-by: Felix Gateru <[email protected]>
absmach · Jan 24, 2025 · 5fd66b6 · 5fd66b6
1 parent 99afa30
commit 5fd66b6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ws/api/endpoint_test.go b/ws/api/endpoint_test.go
@@ -107,6 +107,8 @@ func TestHandshake(t *testing.T) {
 	require.Nil(t, err)
 	defer ts.Close()
 	pubsub.On("Subscribe", mock.Anything, mock.Anything).Return(nil)
+	es.On("Subscribe", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil)
+	es.On("Publish", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil)
 	pubsub.On("Publish", mock.Anything, mock.Anything, mock.Anything).Return(nil)
 	clients.On("Authenticate", mock.Anything, mock.Anything).Return(&grpcClientsV1.AuthnRes{Authenticated: true}, nil)
 	authn.On("Authenticate", mock.Anything, mock.Anything).Return(smqauthn.Session{}, nil)