diff --git a/api/client_test.go b/api/client_test.go index 8fae00e..2b28b11 100644 --- a/api/client_test.go +++ b/api/client_test.go @@ -4,6 +4,7 @@ import ( "testing" "github.com/eosswedenorg/thalos/api/message" + "github.com/stretchr/testify/assert" ) type mockReader struct{} @@ -48,3 +49,13 @@ func TestClient_Subscribe(t *testing.T) { }) } } + +func TestClient_SubscribeWithNilHandler(t *testing.T) { + client := NewClient(nil, nil) + client.OnAction = mockActionHandler + client.OnHeartbeat = mockHbHandler + + err := client.Subscribe(TableDeltaChannel{Name: "name"}.Channel()) + + assert.Error(t, err) +}