status-go/services/subscriptions/subscriptions_test.go

194 lines
4.4 KiB
Go
Raw Normal View History

package subscriptions
import (
"encoding/json"
"errors"
"fmt"
"testing"
"time"
"github.com/status-im/status-go/signal"
"github.com/stretchr/testify/require"
)
const (
filterID = "123"
filterNS = "tst"
)
type mockFilter struct {
filterID string
data []interface{}
filterError error
uninstalled bool
uninstallError error
}
func newMockFilter(filterID string) *mockFilter {
return &mockFilter{
filterID: filterID,
}
}
func (mf *mockFilter) getID() string {
return mf.filterID
}
func (mf *mockFilter) getChanges() ([]interface{}, error) {
if mf.filterError != nil {
err := mf.filterError
mf.filterError = nil
return nil, err
}
data := mf.data
mf.data = nil
return data, nil
}
func (mf *mockFilter) uninstall() error {
mf.uninstalled = true
return mf.uninstallError
}
func (mf *mockFilter) setData(data ...interface{}) {
mf.data = data
}
func (mf *mockFilter) setError(err error) {
mf.data = nil
mf.filterError = err
}
func TestSubscriptionGetData(t *testing.T) {
filter := newMockFilter(filterID)
subs := NewSubscriptions(time.Microsecond)
subID, _ := subs.Create(filterNS, filter)
require.Equal(t, string(subID), fmt.Sprintf("%s-%s", filterNS, filterID))
proceed := make(chan struct{})
signal.SetDefaultNodeNotificationHandler(func(jsonEvent string) {
defer close(proceed)
validateFilterData(t, jsonEvent, string(subID), "1", "2", "3", "4")
})
filter.setData("1", "2", "3", "4")
select {
case <-proceed:
return
case <-time.After(time.Second):
require.NoError(t, errors.New("timeout while waiting for filter results"))
}
require.NoError(t, subs.removeAll())
signal.ResetDefaultNodeNotificationHandler()
}
func TestSubscriptionGetError(t *testing.T) {
filter := newMockFilter(filterID)
subs := NewSubscriptions(time.Microsecond)
subID, _ := subs.Create(filterNS, filter)
require.Equal(t, string(subID), fmt.Sprintf("%s-%s", filterNS, filterID))
proceed := make(chan struct{})
expectedError := errors.New("test-error")
signal.SetDefaultNodeNotificationHandler(func(jsonEvent string) {
defer close(proceed)
validateFilterError(t, jsonEvent, string(subID), expectedError.Error())
})
filter.setError(expectedError)
select {
case <-proceed:
return
case <-time.After(time.Second):
require.NoError(t, errors.New("timeout while waiting for filter results"))
}
require.NoError(t, subs.removeAll())
signal.ResetDefaultNodeNotificationHandler()
}
func TestSubscriptionRemove(t *testing.T) {
filter := newMockFilter(filterID)
subs := NewSubscriptions(time.Microsecond)
subID, _ := subs.Create(filterNS, filter)
require.NoError(t, subs.Remove(subID))
require.True(t, filter.uninstalled)
require.Empty(t, subs.subs)
}
func TestSubscriptionRemoveError(t *testing.T) {
filter := newMockFilter(filterID)
filter.uninstallError = errors.New("uninstall-error-1")
subs := NewSubscriptions(time.Microsecond)
subID, _ := subs.Create(filterNS, filter)
require.Equal(t, subs.Remove(subID), filter.uninstallError)
require.True(t, filter.uninstalled)
require.Equal(t, len(subs.subs), 0)
}
func TestSubscriptionRemoveAll(t *testing.T) {
filter0 := newMockFilter(filterID)
filter1 := newMockFilter(filterID + "1")
subs := NewSubscriptions(time.Microsecond)
_, err := subs.Create(filterNS, filter0)
require.NoError(t, err)
_, err = subs.Create(filterNS, filter1)
require.NoError(t, err)
require.Equal(t, len(subs.subs), 2)
require.NoError(t, subs.removeAll())
require.False(t, filter0.uninstalled)
require.False(t, filter1.uninstalled)
require.Equal(t, len(subs.subs), 0)
}
func validateFilterError(t *testing.T, jsonEvent string, expectedSubID string, expectedErrorMessage string) {
result := struct {
Event signal.SubscriptionErrorEvent `json:"event"`
Type string `json:"type"`
}{}
require.NoError(t, json.Unmarshal([]byte(jsonEvent), &result))
require.Equal(t, signal.EventSubscriptionsError, result.Type)
require.Equal(t, expectedErrorMessage, result.Event.ErrorMessage)
}
func validateFilterData(t *testing.T, jsonEvent string, expectedSubID string, expectedData ...interface{}) {
result := struct {
Event signal.SubscriptionDataEvent `json:"event"`
Type string `json:"type"`
}{}
require.NoError(t, json.Unmarshal([]byte(jsonEvent), &result))
require.Equal(t, signal.EventSubscriptionsData, result.Type)
require.Equal(t, expectedData, result.Event.Data)
require.Equal(t, expectedSubID, result.Event.FilterID)
}