149 lines
4.1 KiB
Go
149 lines
4.1 KiB
Go
package stream
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
time "time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSubscription(t *testing.T) {
|
|
eb := NewEventBuffer()
|
|
|
|
index := uint64(100)
|
|
|
|
startHead := eb.Head()
|
|
|
|
// Start with an event in the buffer
|
|
publishTestEvent(index, eb, "test")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// Create a subscription
|
|
req := &SubscribeRequest{
|
|
Topic: Topic_ServiceHealth,
|
|
Key: "test",
|
|
}
|
|
sub := NewSubscription(ctx, req, startHead)
|
|
|
|
// First call to sub.Next should return our published event immediately
|
|
start := time.Now()
|
|
got, err := sub.Next()
|
|
elapsed := time.Since(start)
|
|
require.NoError(t, err)
|
|
require.True(t, elapsed < 200*time.Millisecond,
|
|
"Event should have been delivered immediately, took %s", elapsed)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, index, got[0].Index)
|
|
|
|
// Schedule an event publish in a while
|
|
index++
|
|
start = time.Now()
|
|
time.AfterFunc(200*time.Millisecond, func() {
|
|
publishTestEvent(index, eb, "test")
|
|
})
|
|
|
|
// Next call should block until event is delivered
|
|
got, err = sub.Next()
|
|
elapsed = time.Since(start)
|
|
require.NoError(t, err)
|
|
require.True(t, elapsed > 200*time.Millisecond,
|
|
"Event should have been delivered after blocking 200ms, took %s", elapsed)
|
|
require.True(t, elapsed < 2*time.Second,
|
|
"Event should have been delivered after short time, took %s", elapsed)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, index, got[0].Index)
|
|
|
|
// Event with wrong key should not be delivered. Deliver a good message right
|
|
// so we don't have to block test thread forever or cancel func yet.
|
|
index++
|
|
publishTestEvent(index, eb, "nope")
|
|
index++
|
|
publishTestEvent(index, eb, "test")
|
|
|
|
start = time.Now()
|
|
got, err = sub.Next()
|
|
elapsed = time.Since(start)
|
|
require.NoError(t, err)
|
|
require.True(t, elapsed < 200*time.Millisecond,
|
|
"Event should have been delivered immediately, took %s", elapsed)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, index, got[0].Index)
|
|
require.Equal(t, "test", got[0].Key)
|
|
|
|
// Cancelling the subscription context should unblock Next
|
|
start = time.Now()
|
|
time.AfterFunc(200*time.Millisecond, func() {
|
|
cancel()
|
|
})
|
|
|
|
_, err = sub.Next()
|
|
elapsed = time.Since(start)
|
|
require.Error(t, err)
|
|
require.True(t, elapsed > 200*time.Millisecond,
|
|
"Event should have been delivered after blocking 200ms, took %s", elapsed)
|
|
require.True(t, elapsed < 2*time.Second,
|
|
"Event should have been delivered after short time, took %s", elapsed)
|
|
}
|
|
|
|
func TestSubscriptionCloseReload(t *testing.T) {
|
|
eb := NewEventBuffer()
|
|
|
|
index := uint64(100)
|
|
|
|
startHead := eb.Head()
|
|
|
|
// Start with an event in the buffer
|
|
publishTestEvent(index, eb, "test")
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// Create a subscription
|
|
req := &SubscribeRequest{
|
|
Topic: Topic_ServiceHealth,
|
|
Key: "test",
|
|
}
|
|
sub := NewSubscription(ctx, req, startHead)
|
|
|
|
// First call to sub.Next should return our published event immediately
|
|
start := time.Now()
|
|
got, err := sub.Next()
|
|
elapsed := time.Since(start)
|
|
require.NoError(t, err)
|
|
require.True(t, elapsed < 200*time.Millisecond,
|
|
"Event should have been delivered immediately, took %s", elapsed)
|
|
require.Len(t, got, 1)
|
|
require.Equal(t, index, got[0].Index)
|
|
|
|
// Schedule a CloseReload simulating the server deciding this subscroption
|
|
// needs to reset (e.g. on ACL perm change).
|
|
start = time.Now()
|
|
time.AfterFunc(200*time.Millisecond, func() {
|
|
sub.CloseReload()
|
|
})
|
|
|
|
_, err = sub.Next()
|
|
elapsed = time.Since(start)
|
|
require.Error(t, err)
|
|
require.Equal(t, ErrSubscriptionReload, err)
|
|
require.True(t, elapsed > 200*time.Millisecond,
|
|
"Reload should have happened after blocking 200ms, took %s", elapsed)
|
|
require.True(t, elapsed < 2*time.Second,
|
|
"Reload should have been delivered after short time, took %s", elapsed)
|
|
}
|
|
|
|
func publishTestEvent(index uint64, b *EventBuffer, key string) {
|
|
// Don't care about the event payload for now just the semantics of publishing
|
|
// something. This is not a valid stream in the end-to-end streaming protocol
|
|
// but enough to test subscription mechanics.
|
|
e := Event{
|
|
Index: index,
|
|
Topic: Topic_ServiceHealth,
|
|
Key: key,
|
|
}
|
|
b.Append([]Event{e})
|
|
}
|