mirror of
https://github.com/compute-blade-community/compute-blade-agent.git
synced 2026-04-21 17:45:43 +02:00
631ddfedd4
* chore: update repository references from uptime-industries to compute-blade-community chore: update repository references from uptime-industries to compute-blade-community for consistency and clarity across all files fix: update links in CHANGELOG.md and README.md to point to the new repository location for accurate documentation fix: update Dockerfile and systemd service file to reflect the new repository URL for proper source tracking refactor: change import paths in Go files to use the new repository name for correct package referencing * chore: Add CODEOWNERS * feat: add auto-labeling --------- Co-authored-by: Cedric Kienzler <cedric@specht-labs.de>
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package events_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/compute-blade-community/compute-blade-agent/pkg/events"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestEventBusManySubscribers(t *testing.T) {
|
|
eb := events.New()
|
|
|
|
// Create a channel and subscribe to a topic without a filter
|
|
sub0 := eb.Subscribe("topic0", 2, events.MatchAll)
|
|
assert.Equal(t, cap(sub0.C()), 2)
|
|
assert.Equal(t, len(sub0.C()), 0)
|
|
defer sub0.Unsubscribe()
|
|
|
|
// Create a channel and subscribe to a topic with a filter
|
|
sub1 := eb.Subscribe("topic0", 2, func(msg any) bool {
|
|
return msg.(int) > 5
|
|
})
|
|
assert.Equal(t, cap(sub1.C()), 2)
|
|
assert.Equal(t, len(sub1.C()), 0)
|
|
defer sub1.Unsubscribe()
|
|
|
|
// Create a channel and subscribe to another topic
|
|
sub2 := eb.Subscribe("topic1", 1, events.MatchAll)
|
|
assert.Equal(t, cap(sub2.C()), 1)
|
|
assert.Equal(t, len(sub2.C()), 0)
|
|
defer sub2.Unsubscribe()
|
|
|
|
sub3 := eb.Subscribe("topic1", 0, events.MatchAll)
|
|
assert.Equal(t, cap(sub3.C()), 0)
|
|
assert.Equal(t, len(sub3.C()), 0)
|
|
defer sub3.Unsubscribe()
|
|
|
|
// Publish some messages
|
|
eb.Publish("topic0", 10)
|
|
eb.Publish("topic0", 4)
|
|
eb.Publish("topic1", "Hello, World!")
|
|
|
|
// Assert received messages
|
|
assert.Equal(t, len(sub0.C()), 2)
|
|
assert.Equal(t, <-sub0.C(), 10)
|
|
assert.Equal(t, <-sub0.C(), 4)
|
|
|
|
assert.Equal(t, len(sub1.C()), 1)
|
|
assert.Equal(t, <-sub1.C(), 10)
|
|
|
|
assert.Equal(t, len(sub2.C()), 1)
|
|
assert.Equal(t, <-sub2.C(), "Hello, World!")
|
|
|
|
// sub3 has no buffer, so it should be empty as there's been no consumer at time of publishing
|
|
assert.Equal(t, len(sub3.C()), 0)
|
|
}
|
|
|
|
func TestUnsubscribe(t *testing.T) {
|
|
eb := events.New()
|
|
|
|
// Create a channel and subscribe to a topic
|
|
sub := eb.Subscribe("topic", 2, events.MatchAll)
|
|
|
|
// Unsubscribe from the topic
|
|
sub.Unsubscribe()
|
|
|
|
// Try to publish a message after unsubscribing
|
|
eb.Publish("topic", "This message should not be received")
|
|
|
|
// Assert that the channel is closed
|
|
_, ok := <-sub.C()
|
|
assert.False(t, ok, "Unsubscribed channel should be closed")
|
|
|
|
}
|