// Test push and pop of queue using channel for syncronous queueing
func TestChanQueue(t *testing.T) {
q := queue.NewChanQueue(2)
q.Push("hello")
q.Push("world")
v, ok := q.Pop()
if !ok {
t.Error("Pop failed!")
}
if v.(string) != "hello" {
t.Errorf("Expected : hello, Result : %s", v.(string))
}
if q.Len() != 1 {
t.Errorf("Expected queue size : 1, Result : %d", q.Len())
}
v, ok = q.Pop()
if !ok {
t.Error("Pop failed!")
}
if v.(string) != "world" {
t.Errorf("Expected : world, Result : %s", v.(string))
}
if q.Len() != 0 {
t.Errorf("Expected queue size : 0, Result : %d", q.Len())
}
}
func BenchmarkChanQueue(b *testing.B) {
q := queue.NewChanQueue(b.N)
for i := 0; i < b.N; i++ {
q.Push(i)
}
for i := 0; i < b.N; i++ {
q.Pop()
}
}
// Test push and pop of channel queue using goroutine
func TestChanQueueSync(t *testing.T) {
runtime.GOMAXPROCS(runtime.NumCPU())
q := queue.NewChanQueue(10)
wg := new(sync.WaitGroup)
for i := 0; i < 1000; i++ {
wg.Add(1)
go func(id int) {
for j := (id * 100) + 0; j < (id*100)+100; j++ {
q.Push(j)
}
wg.Done()
}(i)
}
for i := 0; i < 1000; i++ {
wg.Add(1)
go func(id int) {
for j := 0; j < 100; j++ {
for {
_, ok := q.Pop()
if ok {
//fmt.Printf("%d : %v\n", id, v)
break
}
}
}
wg.Done()
}(i)
}
wg.Wait()
if q.Len() != 0 {
t.Error("Queue size is : ", q.Len())
}
}
