fix types
This commit is contained in:
+13
-14
@@ -2,7 +2,6 @@ package prism
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"git.wisehodl.dev/jay/go-honeybee"
|
|
||||||
"git.wisehodl.dev/jay/go-mana-component"
|
"git.wisehodl.dev/jay/go-mana-component"
|
||||||
"git.wisehodl.dev/jay/go-roots-ws"
|
"git.wisehodl.dev/jay/go-roots-ws"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
@@ -51,12 +50,12 @@ func TestEnvoy_IsConnected(t *testing.T) {
|
|||||||
ctx := component.MustNew(context.Background(), "prism", "test")
|
ctx := component.MustNew(context.Background(), "prism", "test")
|
||||||
mu := sync.RWMutex{}
|
mu := sync.RWMutex{}
|
||||||
url := "wss://test"
|
url := "wss://test"
|
||||||
events := make(chan honeybee.OutboundPoolEvent)
|
events := make(chan OutboundPoolEvent)
|
||||||
|
|
||||||
envoy := newEnvoy(ctx, url, nil, nil, events, nil, nil)
|
envoy := newEnvoy(ctx, url, nil, nil, events, nil, nil)
|
||||||
eventSub := envoy.SubscribeEvents()
|
eventSub := envoy.SubscribeEvents()
|
||||||
|
|
||||||
gotEvents := []honeybee.OutboundPoolEvent{}
|
gotEvents := []OutboundPoolEvent{}
|
||||||
go func() {
|
go func() {
|
||||||
for ev := range eventSub {
|
for ev := range eventSub {
|
||||||
mu.Lock()
|
mu.Lock()
|
||||||
@@ -65,8 +64,8 @@ func TestEnvoy_IsConnected(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
events <- honeybee.OutboundPoolEvent{
|
events <- OutboundPoolEvent{
|
||||||
ID: url, Kind: honeybee.OutboundEventConnected, At: time.Now()}
|
ID: url, Kind: EventConnected, At: time.Now()}
|
||||||
|
|
||||||
Eventually(t, func() bool {
|
Eventually(t, func() bool {
|
||||||
return envoy.IsConnected()
|
return envoy.IsConnected()
|
||||||
@@ -77,12 +76,12 @@ func TestEnvoy_Events(t *testing.T) {
|
|||||||
ctx := component.MustNew(context.Background(), "prism", "test")
|
ctx := component.MustNew(context.Background(), "prism", "test")
|
||||||
mu := sync.RWMutex{}
|
mu := sync.RWMutex{}
|
||||||
url := "wss://test"
|
url := "wss://test"
|
||||||
events := make(chan honeybee.OutboundPoolEvent)
|
events := make(chan OutboundPoolEvent)
|
||||||
|
|
||||||
envoy := newEnvoy(ctx, url, nil, nil, events, nil, nil)
|
envoy := newEnvoy(ctx, url, nil, nil, events, nil, nil)
|
||||||
eventSub := envoy.SubscribeEvents()
|
eventSub := envoy.SubscribeEvents()
|
||||||
|
|
||||||
gotEvents := []honeybee.OutboundPoolEvent{}
|
gotEvents := []OutboundPoolEvent{}
|
||||||
go func() {
|
go func() {
|
||||||
for ev := range eventSub {
|
for ev := range eventSub {
|
||||||
mu.Lock()
|
mu.Lock()
|
||||||
@@ -91,8 +90,8 @@ func TestEnvoy_Events(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
events <- honeybee.OutboundPoolEvent{
|
events <- OutboundPoolEvent{
|
||||||
ID: url, Kind: honeybee.OutboundEventConnected, At: time.Now()}
|
ID: url, Kind: EventConnected, At: time.Now()}
|
||||||
|
|
||||||
Eventually(t, func() bool {
|
Eventually(t, func() bool {
|
||||||
mu.RLock()
|
mu.RLock()
|
||||||
@@ -101,7 +100,7 @@ func TestEnvoy_Events(t *testing.T) {
|
|||||||
}, "should have gotten event")
|
}, "should have gotten event")
|
||||||
|
|
||||||
mu.RLock()
|
mu.RLock()
|
||||||
assert.Equal(t, honeybee.OutboundEventConnected, gotEvents[0].Kind)
|
assert.Equal(t, EventConnected, gotEvents[0].Kind)
|
||||||
mu.RUnlock()
|
mu.RUnlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -109,12 +108,12 @@ func TestEnvoy_Inbox(t *testing.T) {
|
|||||||
ctx := component.MustNew(context.Background(), "prism", "test")
|
ctx := component.MustNew(context.Background(), "prism", "test")
|
||||||
mu := sync.RWMutex{}
|
mu := sync.RWMutex{}
|
||||||
url := "wss://test"
|
url := "wss://test"
|
||||||
inbox := make(chan honeybee.InboxMessage)
|
inbox := make(chan InboxMessage)
|
||||||
|
|
||||||
envoy := newEnvoy(ctx, url, nil, nil, nil, inbox, nil)
|
envoy := newEnvoy(ctx, url, nil, nil, nil, inbox, nil)
|
||||||
inboxSub := envoy.SubscribeInbox([]string{"EVENT"})
|
inboxSub := envoy.SubscribeInbox([]string{"EVENT"})
|
||||||
|
|
||||||
gotInbox := []honeybee.InboxMessage{}
|
gotInbox := []InboxMessage{}
|
||||||
go func() {
|
go func() {
|
||||||
for ev := range inboxSub {
|
for ev := range inboxSub {
|
||||||
mu.Lock()
|
mu.Lock()
|
||||||
@@ -123,12 +122,12 @@ func TestEnvoy_Inbox(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
inbox <- honeybee.InboxMessage{
|
inbox <- InboxMessage{
|
||||||
ID: url,
|
ID: url,
|
||||||
Data: envelope.EncloseEvent([]byte("{}")),
|
Data: envelope.EncloseEvent([]byte("{}")),
|
||||||
ReceivedAt: time.Now(),
|
ReceivedAt: time.Now(),
|
||||||
}
|
}
|
||||||
inbox <- honeybee.InboxMessage{
|
inbox <- InboxMessage{
|
||||||
ID: url,
|
ID: url,
|
||||||
Data: envelope.EncloseOK("id", true, "ok"),
|
Data: envelope.EncloseOK("id", true, "ok"),
|
||||||
ReceivedAt: time.Now(),
|
ReceivedAt: time.Now(),
|
||||||
|
|||||||
Reference in New Issue
Block a user