41 lines
833 B
Go
41 lines
833 B
Go
package stream
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestWithoutOutputs(t *testing.T) {
|
|
stream := New()
|
|
defer stream.Close()
|
|
stream.Broadcast <- []byte("hello world")
|
|
}
|
|
|
|
func TestWithOneOutput(t *testing.T) {
|
|
stream := New()
|
|
defer stream.Close()
|
|
|
|
// Register one output
|
|
output := make(chan []byte, 64)
|
|
stream.Register(output, false)
|
|
|
|
// Try to pass one message
|
|
stream.Broadcast <- []byte("hello world")
|
|
msg := <-output
|
|
if string(msg) != "hello world" {
|
|
t.Errorf("Message has wrong content: %s != hello world", msg)
|
|
}
|
|
|
|
// Check client count
|
|
if count := stream.Count(); count != 1 {
|
|
t.Errorf("Client counter returned %d, expected 1", count)
|
|
}
|
|
|
|
// Unregister
|
|
stream.Unregister(output, false)
|
|
|
|
// Check client count
|
|
if count := stream.Count(); count != 0 {
|
|
t.Errorf("Client counter returned %d, expected 0", count)
|
|
}
|
|
}
|