func TestMessageQueue_SubscibersList(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
defer q.Close()
subs := map[amq.MessageConsumer]struct{}{
new(countingConsumer): struct{}{},
new(countingConsumer): struct{}{},
new(countingConsumer): struct{}{},
}
for c := range subs {
err := q.Subscribe(c)
if err != nil {
t.Error("Unexpected error", err)
}
}
list := q.Subscriptions()
if len(list) != 3 {
t.Error("Unexpected consumer count, expedted:", 4, "got:", len(list))
}
for _, c := range list {
if _, found := subs[c]; !found {
t.Error("Consumer not found in reference list")
}
}
}
func TestMessageQueue_QueueBalancesMessagesBetweenConsumers(t *testing.T) {
var wg sync.WaitGroup
wg.Add(1)
q := amq.NewQueue(queue.NewQueueHandler())
c1 := new(countingConsumer)
c2 := new(countingConsumer)
c3 := new(countingConsumer)
c4 := new(countingConsumer)
q.Subscribe(c1)
q.Subscribe(c2)
q.Subscribe(c3)
q.Subscribe(c4)
go func() {
for i := 0; i < 1000; i++ {
q.Consume(testMsg{})
}
q.Close()
wg.Done()
}()
wg.Wait()
for _, consumer := range []*countingConsumer{c1, c2, c3, c4} {
consumer.mu.RLock()
if consumer.callsCount != 250 {
t.Errorf("Consumer has unexpected call count %d expected %d", consumer.callsCount, 250)
}
consumer.mu.RUnlock()
}
}
func TestMessageQueue_QueueCloseFlushesMessages(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
c := new(countingConsumer)
err := q.Subscribe(c)
if err != nil {
t.Error("Unexpected error")
}
var wg sync.WaitGroup
wg.Add(1)
go func() {
for i := 0; i < 100; i++ {
q.Consume(testMsg{})
}
q.Close()
wg.Done()
}()
wg.Wait()
c.mu.RLock()
if c.callsCount != 100 {
t.Errorf("Queue did not flush on close, got: %d messages", c.callsCount)
}
c.mu.RUnlock()
}
func TestMessageQueue_EmptyQueueHasZeroLength(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
if q.Len() != 0 {
t.Errorf("Unexpected queue length")
}
q.Close()
}
func TestMessageQueue_ForceCloseOnEmptyQueue(t *testing.T) {
// Just to get coverage, there is no way to distinguish close from force close on empty queue
q := amq.NewQueue(queue.NewQueueHandler())
if q.Len() != 0 {
t.Errorf("Unexpected queue length")
}
q.ForceClose()
}
func TestMessageQueue_QueueWithoutSubscribersAcummulateMessages(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
for i := 0; i < 100; i++ {
q.Consume(testMsg{})
}
if q.Len() != 100 {
t.Errorf("Queue has unexpected size %d != %d", q.Len(), 100)
}
q.Close()
}
func TestMessageQueue_UnsubscribeWithEmptySubscriptionsListReturnsError(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
defer q.Close()
c := new(countingConsumer)
err := q.Unsubscribe(c)
if err != amq.ErrConsumerNotFound {
t.Error("Invalid error returned", err)
}
}
func TestMessageQueue_SubscribeConsumer(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
defer q.Close()
c := new(countingConsumer)
err := q.Subscribe(c)
if err != nil {
t.Error("Unexpected error")
}
}
func TestMessageQueue_EmptySubscibersList(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
defer q.Close()
list := q.Subscriptions()
if len(list) != 0 {
t.Error("Unexpected consumer in list")
}
if list != nil {
t.Error("Unexpected type")
}
}
func TestMessageQueue_SubscribeConsumerTwiceReturnsError(t *testing.T) {
q := amq.NewQueue(queue.NewQueueHandler())
defer q.Close()
c := new(countingConsumer)
err := q.Subscribe(c)
if err != nil {
t.Error("Unexpected error")
}
err = q.Subscribe(c)
if err != amq.ErrConsumerAlreadySubscribed {
t.Error("Invalid error returned", err)
}
}