queue_test.go

package PriorityQueue

import (
	"github.com/magiconair/properties/assert"
	"testing"
)

func TestQueuePush(t *testing.T) {
	item := struct{}{}
	q := Build()
	q.Push(item, 2)
	assert.Equal(t, q.Len(), 1)
}

func TestQueuePull(t *testing.T) {
	item := struct{}{}
	q := Build()
	q.Push(item, 2)
	gotItem, _ := q.Pull()
	assert.Equal(t, gotItem, item)
}

func TestQueuePullEmpty(t *testing.T) {
	q := Build()
	gotItem, err := q.Pull()
	assert.Equal(t, err.Error(), "empty")
	assert.Equal(t, gotItem, nil)
}

func TestQueueLen(t *testing.T) {
	q := Build()
	assert.Equal(t, q.Len(), 0)
	q.Push(struct{}{}, 1)
	assert.Equal(t, q.Len(), 1)
}

func TestPriority(t *testing.T) {
	item1 := 111
	item2 := 222
	q := Build()
	q.Push(item2, 2)
	q.Push(item1, 1)
	gotItem, _ := q.Pull()
	assert.Equal(t, gotItem, item1)
}

func TestPrioritize(t *testing.T) {
	ch1 := make(chan interface{})
	ch2 := make(chan interface{})
	item1 := 111
	item2 := 222

	outCh, _ := Prioritize(ch1, ch2)

	ch2 <- item2
	ch2 <- item2
	ch2 <- item2

	ch1 <- item1
	ch1 <- item1
	ch1 <- item1

	<-outCh

	got := <-outCh
	assert.Equal(t, got, item1)
}