Пример #1
0
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")
		}
	}
}
Пример #2
0
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()
	}
}
Пример #3
0
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()
}
Пример #4
0
func TestMessageQueue_EmptyQueueHasZeroLength(t *testing.T) {
	q := amq.NewQueue(queue.NewQueueHandler())

	if q.Len() != 0 {
		t.Errorf("Unexpected queue length")
	}
	q.Close()
}
Пример #5
0
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()
}
Пример #6
0
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()
}
Пример #7
0
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)
	}
}
Пример #8
0
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")
	}
}
Пример #9
0
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")
	}
}
Пример #10
0
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)
	}
}