workerpool/taskqueue_test.go at master · tvntsr/workerpool · 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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
package pool

import (
	"testing"
	"time"
)

func Test_WorkerTaskQueueAddAndRemoveTask(t *testing.T) {

	q := NewWorkerTaskQueue()

	if len(q.elements) != 0 {
		t.Fatal("Wrong elements count on new queue")
	}

	wt0, _ := createWorkerTask(0)
	wt1, _ := createWorkerTask(1)

	r := q.AddTask(wt0)
	if !r {
		t.Fatal("Expecting true in the response")
	}

	if len(q.elements) != 1 {
		t.Fatal("Wrong elements count on queue")
	}

	r = q.AddTask(wt1)
	if !r {
		t.Fatal("Expecting true in the response for second worker")
	}
	if len(q.elements) != 2 {
		t.Fatal("Wrong elements count on queue")
	}

	r = q.AddTask(wt0)
	if r {
		t.Fatal("Expecting false when adding self second time")
	}

	if len(q.elements) != 2 {
		t.Fatal("Wrong elements count on queue")
	}

	r = q.RemoveTask(wt0)
	if !r {
		t.Fatal("Expecting true when removing task")
	}
	if len(q.elements) != 1 {
		t.Fatal("Wrong elements count on queue")
	}

	r = q.RemoveTask(wt0)
	if r {
		t.Fatal("Expecting false when removing self second time")
	}
	if len(q.elements) != 1 {
		t.Fatal("Wrong elements count on queue")
	}

	r = q.RemoveTask(wt1)
	if !r {
		t.Fatal("Expecting true when removing self")
	}
	if len(q.elements) != 0 {
		t.Fatal("Wrong elements count on queue")
	}

}

func Test_TaimedWaitTaskFinishedTimeout(t *testing.T) {
	q := NewWorkerTaskQueue()
	wt0, _ := createWorkerTask(0)

	_ = q.AddTask(wt0)

	w, err := q.TaimedWaitTaskFinished(100 * time.Millisecond)
	if w != nil {
		t.Fatal("Wrong element returned")
	}
	if err == nil {
		t.Fatal("Expecting error")
	}
	if err.Error() != "timeout" {
		t.Fatal("Wrong error returned")
	}
}

func Test_TaimedWaitTaskFinished(t *testing.T) {
	q := NewWorkerTaskQueue()
	wt0, _ := createWorkerTask(0)
	wt1, _ := createWorkerTask(1)

	_ = q.AddTask(wt0)
	_ = q.AddTask(wt1)

	go func() {
		time.Sleep(1 * time.Second)
		wt0.setStatus(Finished, "", nil)
	}()

	w, err := q.TaimedWaitTaskFinished(3 * time.Second)
	if w == nil {
		t.Fatal("Wrong element returned")
	}
	if err != nil {
		t.Fatalf("Error returned, %v", err)
	}
	if w.id != wt0.id {
		t.Fatal("Wrong item returned")
	}
	if len(q.elements) != 1 {
		t.Fatal("Wrong elements count on queue")
	}

}

func Test_WaitTaskFinished(t *testing.T) {
	q := NewWorkerTaskQueue()
	wt0, _ := createWorkerTask(0)
	wt1, _ := createWorkerTask(1)

	_ = q.AddTask(wt0)
	_ = q.AddTask(wt1)

	go func() {
		time.Sleep(1 * time.Second)
		wt0.setStatus(Finished, "", nil)
	}()

	w, err := q.WaitTaskFinished()
	if w == nil {
		t.Fatal("Wrong element returned")
	}
	if err != nil {
		t.Fatalf("Error returned, %v", err)
	}
	if w.id != wt0.id {
		t.Fatal("Wrong item returned")
	}
	if len(q.elements) != 1 {
		t.Fatal("Wrong elements count on queue")
	}
}