This commit is contained in:
ErickSkrauch
2019-04-15 01:32:22 +03:00
parent d2d6d07fa6
commit 879a33344b
4 changed files with 137 additions and 138 deletions

View File

@@ -1,47 +0,0 @@
package queue
import (
"testing"
testify "github.com/stretchr/testify/assert"
)
func TestEnqueue(t *testing.T) {
assert := testify.New(t)
s := createQueue()
s.Enqueue(&Job{Username: "username1"})
s.Enqueue(&Job{Username: "username2"})
s.Enqueue(&Job{Username: "username3"})
assert.Equal(3, s.Size())
}
func TestDequeueN(t *testing.T) {
assert := testify.New(t)
s := createQueue()
s.Enqueue(&Job{Username: "username1"})
s.Enqueue(&Job{Username: "username2"})
s.Enqueue(&Job{Username: "username3"})
s.Enqueue(&Job{Username: "username4"})
items := s.Dequeue(2)
assert.Len(items, 2)
assert.Equal("username1", items[0].Username)
assert.Equal("username2", items[1].Username)
assert.Equal(2, s.Size())
items = s.Dequeue(40)
assert.Len(items, 2)
assert.Equal("username3", items[0].Username)
assert.Equal("username4", items[1].Username)
assert.True(s.IsEmpty())
}
func createQueue() JobsQueue {
s := JobsQueue{}
s.New()
return s
}