mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 17:47:03 +00:00
102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package extension_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/inbucket/inbucket/pkg/extension"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Simple smoke test without using AsyncTestListener.
|
|
func TestAsyncBrokerEmitCallsOneListener(t *testing.T) {
|
|
broker := &extension.AsyncEventBroker[string]{}
|
|
|
|
// Setup listener.
|
|
events := make(chan string, 1)
|
|
listener := func(s string) {
|
|
events <- s
|
|
}
|
|
broker.AddListener("x", listener)
|
|
|
|
want := "bacon"
|
|
broker.Emit(&want)
|
|
|
|
var got string
|
|
select {
|
|
case event := <-events:
|
|
got = event
|
|
|
|
case <-time.After(time.Second * 2):
|
|
t.Fatal("Timeout waiting for event")
|
|
}
|
|
|
|
if got != want {
|
|
t.Errorf("Emit got %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestAsyncBrokerEmitCallsMultipleListeners(t *testing.T) {
|
|
broker := &extension.AsyncEventBroker[string]{}
|
|
|
|
// Setup listeners.
|
|
first := broker.AsyncTestListener("first", 1)
|
|
second := broker.AsyncTestListener("second", 1)
|
|
|
|
want := "hi"
|
|
broker.Emit(&want)
|
|
|
|
first_got, err := first()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, *first_got)
|
|
|
|
second_got, err := second()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, *second_got)
|
|
}
|
|
|
|
func TestAsyncBrokerAddingDuplicateNameReplacesPrevious(t *testing.T) {
|
|
broker := &extension.AsyncEventBroker[string]{}
|
|
|
|
// Setup listeners.
|
|
first := broker.AsyncTestListener("dup", 1)
|
|
second := broker.AsyncTestListener("dup", 1)
|
|
|
|
want := "hi"
|
|
broker.Emit(&want)
|
|
|
|
first_got, err := first()
|
|
require.Error(t, err)
|
|
assert.Nil(t, first_got)
|
|
|
|
second_got, err := second()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, *second_got)
|
|
}
|
|
|
|
func TestAsyncBrokerRemovingListenerSuccessful(t *testing.T) {
|
|
broker := &extension.AsyncEventBroker[string]{}
|
|
|
|
// Setup listeners.
|
|
first := broker.AsyncTestListener("1", 1)
|
|
second := broker.AsyncTestListener("2", 1)
|
|
broker.RemoveListener("1")
|
|
|
|
want := "hi"
|
|
broker.Emit(&want)
|
|
|
|
first_got, err := first()
|
|
require.Error(t, err)
|
|
assert.Nil(t, first_got)
|
|
|
|
second_got, err := second()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, *second_got)
|
|
}
|
|
|
|
func TestAsyncBrokerRemovingMissingListener(t *testing.T) {
|
|
broker := &extension.AsyncEventBroker[string]{}
|
|
broker.RemoveListener("doesn't crash")
|
|
}
|