Initial commit
This commit is contained in:
Yaser
355
pkg/routing/hub_test.go
Normal file
355
pkg/routing/hub_test.go
Normal file
@@ -0,0 +1,355 @@
|
||||
package routing
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/openware/rango/pkg/message"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/mock"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
type MockedClient struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
func (c *MockedClient) Send(m string) {
|
||||
c.Called(m)
|
||||
}
|
||||
|
||||
func (c *MockedClient) Close() {
|
||||
}
|
||||
|
||||
func (c *MockedClient) GetAuth() Auth {
|
||||
args := c.Called()
|
||||
return args.Get(0).(Auth)
|
||||
}
|
||||
|
||||
func (c *MockedClient) GetSubscriptions() []string {
|
||||
args := c.Called()
|
||||
return args.Get(0).([]string)
|
||||
}
|
||||
|
||||
func (c *MockedClient) SubscribePublic(s string) {
|
||||
c.Called(s)
|
||||
}
|
||||
|
||||
func (c *MockedClient) SubscribePrivate(s string) {
|
||||
c.Called(s)
|
||||
}
|
||||
|
||||
func (c *MockedClient) UnsubscribePublic(s string) {
|
||||
c.Called(s)
|
||||
}
|
||||
|
||||
func (c *MockedClient) UnsubscribePrivate(s string) {
|
||||
c.Called(s)
|
||||
}
|
||||
|
||||
func setup(c *MockedClient, streams []string) *Hub {
|
||||
h := NewHub(nil)
|
||||
h.handleSubscribe(&Request{
|
||||
client: c,
|
||||
Request: message.Request{
|
||||
Streams: streams,
|
||||
},
|
||||
})
|
||||
return h
|
||||
}
|
||||
|
||||
func teardown(h *Hub, c *MockedClient, streams []string) {
|
||||
h.handleUnsubscribe(&Request{
|
||||
client: c,
|
||||
Request: message.Request{
|
||||
Streams: streams,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func TestAnonymous(t *testing.T) {
|
||||
t.Run("subscribe to a public single stream", func(t *testing.T) {
|
||||
c := &MockedClient{}
|
||||
|
||||
streams := []string{
|
||||
"eurusd.trades",
|
||||
}
|
||||
|
||||
c.On("GetAuth").Return(Auth{})
|
||||
c.On("GetSubscriptions").Return(streams).Once()
|
||||
c.On("SubscribePublic", streams[0]).Return().Once()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":["`+streams[0]+`"]}}`).Return()
|
||||
|
||||
h := setup(c, streams)
|
||||
assert.Equal(t, 1, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
|
||||
c.On("UnsubscribePublic", streams[0]).Return()
|
||||
c.On("GetSubscriptions").Return([]string{}).Once()
|
||||
c.On("Send", `{"success":{"message":"unsubscribed","streams":[]}}`).Return()
|
||||
|
||||
teardown(h, c, streams)
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
})
|
||||
|
||||
t.Run("subscribe to multiple public streams", func(t *testing.T) {
|
||||
c := &MockedClient{}
|
||||
streams := []string{
|
||||
"eurusd.trades",
|
||||
"eurusd.updates",
|
||||
}
|
||||
|
||||
c.On("GetAuth").Return(Auth{})
|
||||
c.On("GetSubscriptions").Return(streams).Once()
|
||||
c.On("SubscribePublic", "eurusd.trades").Return()
|
||||
c.On("SubscribePublic", "eurusd.updates").Return()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":["eurusd.trades","eurusd.updates"]}}`).Return()
|
||||
|
||||
h := setup(c, []string{
|
||||
"eurusd.trades",
|
||||
"eurusd.updates",
|
||||
})
|
||||
|
||||
assert.Equal(t, 2, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
|
||||
c.On("UnsubscribePublic", streams[0]).Return().Once()
|
||||
c.On("UnsubscribePublic", streams[1]).Return().Once()
|
||||
c.On("GetSubscriptions").Return([]string{}).Once()
|
||||
c.On("Send", `{"success":{"message":"unsubscribed","streams":[]}}`).Return()
|
||||
|
||||
teardown(h, c, streams)
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
|
||||
})
|
||||
|
||||
t.Run("subscribe to a private single stream", func(t *testing.T) {
|
||||
c := MockedClient{}
|
||||
|
||||
c.On("GetAuth").Return(Auth{})
|
||||
c.On("GetSubscriptions").Return([]string{})
|
||||
c.On("SubscribePrivate", "trades").Return()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":[]}}`).Return()
|
||||
|
||||
h := setup(&c, []string{
|
||||
"trades",
|
||||
})
|
||||
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
})
|
||||
}
|
||||
func TestAuthenticated(t *testing.T) {
|
||||
t.Run("subscribe to a private single stream", func(t *testing.T) {
|
||||
c := &MockedClient{}
|
||||
|
||||
c.On("GetAuth").Return(Auth{UID: "UIDABC00001"})
|
||||
c.On("GetSubscriptions").Return([]string{"trades"}).Once()
|
||||
c.On("SubscribePrivate", "trades").Return()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":["trades"]}}`).Return()
|
||||
|
||||
h := setup(c, []string{
|
||||
"trades",
|
||||
})
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 1, len(h.PrivateTopics))
|
||||
|
||||
c.On("UnsubscribePrivate", "trades").Return().Once()
|
||||
c.On("GetSubscriptions").Return([]string{}).Once()
|
||||
c.On("Send", `{"success":{"message":"unsubscribed","streams":[]}}`).Return()
|
||||
|
||||
teardown(h, c, []string{"trades"})
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
})
|
||||
|
||||
t.Run("subscribe to multiple private streams", func(t *testing.T) {
|
||||
c := &MockedClient{}
|
||||
|
||||
c.On("GetSubscriptions").Return([]string{"trades", "orders"}).Once()
|
||||
c.On("GetAuth").Return(Auth{UID: "UIDABC00001"})
|
||||
c.On("SubscribePrivate", "trades").Return()
|
||||
c.On("SubscribePrivate", "orders").Return()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":["trades","orders"]}}`).Return()
|
||||
|
||||
h := setup(c, []string{"trades", "orders"})
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 1, len(h.PrivateTopics))
|
||||
|
||||
uTopics, ok := h.PrivateTopics["UIDABC00001"]
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, 2, len(uTopics))
|
||||
|
||||
c.On("UnsubscribePrivate", "trades").Return().Once()
|
||||
c.On("UnsubscribePrivate", "orders").Return().Once()
|
||||
c.On("GetSubscriptions").Return([]string{}).Once()
|
||||
c.On("Send", `{"success":{"message":"unsubscribed","streams":[]}}`).Return()
|
||||
|
||||
teardown(h, c, []string{"trades", "orders"})
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
|
||||
})
|
||||
|
||||
t.Run("subscribe to multiple private and public streams", func(t *testing.T) {
|
||||
c := &MockedClient{}
|
||||
|
||||
c.On("GetSubscriptions").Return([]string{"trades", "orders", "eurusd.updates"}).Once()
|
||||
c.On("GetAuth").Return(Auth{UID: "UIDABC00001"})
|
||||
c.On("SubscribePrivate", "trades").Return()
|
||||
c.On("SubscribePrivate", "orders").Return()
|
||||
c.On("SubscribePublic", "eurusd.updates").Return()
|
||||
c.On("Send", `{"success":{"message":"subscribed","streams":["trades","orders","eurusd.updates"]}}`).Return()
|
||||
|
||||
h := setup(c, []string{"trades", "orders", "eurusd.updates"})
|
||||
assert.Equal(t, 1, len(h.PublicTopics))
|
||||
assert.Equal(t, 1, len(h.PrivateTopics))
|
||||
|
||||
uTopics, ok := h.PrivateTopics["UIDABC00001"]
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, 2, len(uTopics))
|
||||
|
||||
c.On("UnsubscribePrivate", "trades").Return().Once()
|
||||
c.On("UnsubscribePrivate", "orders").Return().Once()
|
||||
c.On("UnsubscribePublic", "eurusd.updates").Return().Once()
|
||||
c.On("GetSubscriptions").Return([]string{}).Once()
|
||||
c.On("Send", `{"success":{"message":"unsubscribed","streams":[]}}`).Return()
|
||||
|
||||
teardown(h, c, []string{"trades", "orders", "eurusd.updates"})
|
||||
assert.Equal(t, 0, len(h.PublicTopics))
|
||||
assert.Equal(t, 0, len(h.PrivateTopics))
|
||||
})
|
||||
}
|
||||
|
||||
func TestIsIncremental(t *testing.T) {
|
||||
assert.True(t, isIncrementObject("public.eurusd.ob-inc"))
|
||||
assert.False(t, isIncrementObject("public.eurusd.ob-snap"))
|
||||
assert.False(t, isIncrementObject("public.eurusd.ob"))
|
||||
|
||||
assert.True(t, isSnapshotObject("public.eurusd.ob-snap"))
|
||||
assert.False(t, isSnapshotObject("public.eurusd.ob-inc"))
|
||||
assert.False(t, isSnapshotObject("public.eurusd.ob"))
|
||||
}
|
||||
|
||||
func TestGetTopic(t *testing.T) {
|
||||
assert.Equal(t, "abc.count", getTopic("public", "abc", "count"))
|
||||
assert.Equal(t, "count", getTopic("private", "abc", "count"))
|
||||
assert.Equal(t, "abc.count-inc", getTopic("public", "abc", "count-inc"))
|
||||
assert.Equal(t, "abc.count-inc", getTopic("public", "abc", "count-snap"))
|
||||
}
|
||||
|
||||
func TestHandleMessage(t *testing.T) {
|
||||
h := NewHub(nil)
|
||||
c := &MockedClient{}
|
||||
c.On("SubscribePublic", "abc.ticker").Return()
|
||||
c.On("Send", "{\"abc.ticker\":{\"some\":\"data\"}}").Return()
|
||||
|
||||
h.subscribePublic("abc.ticker", &Request{
|
||||
client: c,
|
||||
})
|
||||
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "ticker",
|
||||
Topic: "abc.ticker",
|
||||
Body: map[string]interface{}{
|
||||
"some": "data",
|
||||
},
|
||||
})
|
||||
|
||||
c.AssertExpectations(t)
|
||||
}
|
||||
|
||||
func TestIncrementalObjectStorage(t *testing.T) {
|
||||
h := NewHub(nil)
|
||||
|
||||
// Increments before the first snapshot must be ignored
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "count-inc",
|
||||
Topic: "abc.count-inc",
|
||||
Body: map[string]interface{}{
|
||||
"data": 1,
|
||||
"sequence": 11,
|
||||
},
|
||||
})
|
||||
|
||||
require.Equal(t, 0, len(h.IncrementalObjects))
|
||||
|
||||
// Initial snapshot
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "count-snap",
|
||||
Topic: "abc.count-inc",
|
||||
Body: map[string]interface{}{
|
||||
"data": []int{2, 3, 4},
|
||||
"sequence": 12,
|
||||
},
|
||||
})
|
||||
|
||||
require.Equal(t, 1, len(h.IncrementalObjects))
|
||||
|
||||
o, ok := h.IncrementalObjects["abc.count-inc"]
|
||||
require.True(t, ok)
|
||||
require.Equal(t, 0, len(o.Increments))
|
||||
require.Equal(t, `{"abc.count-snap":{"data":[2,3,4],"sequence":12}}`, o.Snapshot)
|
||||
|
||||
// First Increment
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "count-inc",
|
||||
Topic: "abc.count-inc",
|
||||
Body: map[string]interface{}{
|
||||
"data": 5,
|
||||
"sequence": 13,
|
||||
},
|
||||
})
|
||||
require.Equal(t, 1, len(h.IncrementalObjects))
|
||||
o, ok = h.IncrementalObjects["abc.count-inc"]
|
||||
require.True(t, ok)
|
||||
require.Equal(t, 1, len(o.Increments))
|
||||
require.Equal(t, `{"abc.count-snap":{"data":[2,3,4],"sequence":12}}`, o.Snapshot)
|
||||
require.Equal(t, `{"abc.count-inc":{"data":5,"sequence":13}}`, o.Increments[0])
|
||||
|
||||
// Second Increment
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "count-inc",
|
||||
Topic: "abc.count-inc",
|
||||
Body: map[string]interface{}{
|
||||
"data": 6,
|
||||
"sequence": 14,
|
||||
},
|
||||
})
|
||||
require.Equal(t, 1, len(h.IncrementalObjects))
|
||||
o, ok = h.IncrementalObjects["abc.count-inc"]
|
||||
require.True(t, ok)
|
||||
require.Equal(t, 2, len(o.Increments))
|
||||
require.Equal(t, `{"abc.count-snap":{"data":[2,3,4],"sequence":12}}`, o.Snapshot)
|
||||
require.Equal(t, `{"abc.count-inc":{"data":5,"sequence":13}}`, o.Increments[0])
|
||||
require.Equal(t, `{"abc.count-inc":{"data":6,"sequence":14}}`, o.Increments[1])
|
||||
|
||||
// Second snapshot
|
||||
h.routeMessage(&Event{
|
||||
Scope: "public",
|
||||
Stream: "abc",
|
||||
Type: "count-snap",
|
||||
Topic: "abc.count-inc",
|
||||
Body: map[string]interface{}{
|
||||
"data": []int{2, 3, 4, 5, 6},
|
||||
"sequence": 14,
|
||||
},
|
||||
})
|
||||
|
||||
require.Equal(t, 1, len(h.IncrementalObjects))
|
||||
o, ok = h.IncrementalObjects["abc.count-inc"]
|
||||
require.True(t, ok)
|
||||
require.Equal(t, 0, len(o.Increments))
|
||||
require.Equal(t, `{"abc.count-snap":{"data":[2,3,4,5,6],"sequence":14}}`, o.Snapshot)
|
||||
}
|
||||
Reference in New Issue
Block a user