delayqueue/priority_queue_test.go at main · MoeYang/delayqueue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package goqueue

import (
	"testing"
)

func TestLen(t *testing.T) {
	q := NewPriorityQueue(3)
	if q.Len() != 0 {
		t.Error("TestLen_len1 <> 0")
	}
	q.Offer(struct{}{}, 1)
	if q.Len() != 1 {
		t.Error("TestLen_len2 <> 1")
	}
}

func TestOffer(t *testing.T) {
	q := NewPriorityQueue(0)
	ele := q.Offer(struct{}{}, 100)
	if ele == nil || ele.Index != 0 {
		t.Error("TestOffer_err1 ele.Index != 0")
	}
	ele = q.Offer(struct{}{}, 2)
	if ele == nil || ele.Index != 0 {
		t.Error("TestOffer_err2 ele.Index != 0", ele)
	}
}

func TestPeek(t *testing.T) {
	q := NewPriorityQueue(3)
	q.Offer(1, 100)
	ele := q.Peek()
	if ele == nil || ele.Value.(int) != 1 {
		t.Error("TestPeek_err1  ele.Value != 1")
	}
	q.Offer(2, 10)
	ele = q.Peek()
	if ele == nil || ele.Value.(int) != 2 {
		t.Error("TestPeek_err2 ele.Value != 2")
	}
}

func TestPeekAndShift(t *testing.T) {
	q := NewPriorityQueue(3)
	ele, wait := q.PeekAndShift(1)
	if ele != nil || wait != 0 {
		t.Errorf("TestPeekAndShift_err1 ele=%+v wait=%d", ele, wait)
	}
	q.Offer(1, 100)
	ele, wait = q.PeekAndShift(1)
	if ele != nil || wait != 99 {
		t.Errorf("TestPeekAndShift_err2 ele=%+v wait=%d", ele, wait)
	}
	ele, wait = q.PeekAndShift(111)
	if ele == nil || wait != 0 {
		t.Errorf("TestPeekAndShift_err3 ele=%+v wait=%d", ele, wait)
	}
}

func TestPeekAndRemove(t *testing.T) {
	q := NewPriorityQueue(3)
	ele := q.PeekAndRemove()
	if ele != nil {
		t.Errorf("TestPeekAndRemove_err1 ele=%+v", ele)
	}
	q.Offer(1, 100)
	q.Offer(2, 99)
	ele = q.PeekAndRemove()
	if ele == nil || ele.Value.(int) != 2 {
		t.Errorf("TestPeekAndRemove_err2 ele=%+v", ele)
	}
	ele = q.PeekAndRemove()
	if ele == nil || ele.Value.(int) != 1 {
		t.Errorf("TestPeekAndRemove_err3 ele=%+v", ele)
	}
	ele = q.PeekAndRemove()
	if ele != nil {
		t.Errorf("TestPeekAndRemove_err4 ele=%+v", ele)
	}
}