42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
![]() |
package graph
|
||
|
|
||
|
import "testing"
|
||
|
|
||
|
func TestPools(t *testing.T) {
|
||
|
s := &TagStore{
|
||
|
pullingPool: make(map[string]chan struct{}),
|
||
|
pushingPool: make(map[string]chan struct{}),
|
||
|
}
|
||
|
|
||
|
if _, err := s.poolAdd("pull", "test1"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if _, err := s.poolAdd("pull", "test2"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if _, err := s.poolAdd("push", "test1"); err == nil || err.Error() != "pull test1 is already in progress" {
|
||
|
t.Fatalf("Expected `pull test1 is already in progress`")
|
||
|
}
|
||
|
if _, err := s.poolAdd("pull", "test1"); err == nil || err.Error() != "pull test1 is already in progress" {
|
||
|
t.Fatalf("Expected `pull test1 is already in progress`")
|
||
|
}
|
||
|
if _, err := s.poolAdd("wait", "test3"); err == nil || err.Error() != "Unknown pool type" {
|
||
|
t.Fatalf("Expected `Unknown pool type`")
|
||
|
}
|
||
|
if err := s.poolRemove("pull", "test2"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := s.poolRemove("pull", "test2"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := s.poolRemove("pull", "test1"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := s.poolRemove("push", "test1"); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := s.poolRemove("wait", "test3"); err == nil || err.Error() != "Unknown pool type" {
|
||
|
t.Fatalf("Expected `Unknown pool type`")
|
||
|
}
|
||
|
}
|