graph/pools_test.go
6856a6b1
 package graph
 
209deff9
 import (
 	"testing"
 
 	"github.com/docker/docker/pkg/reexec"
 )
 
 func init() {
 	reexec.Init()
 }
6856a6b1
 
 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`")
 	}
 }