package libnetwork import ( "fmt" "io/ioutil" "os" "testing" "time" "github.com/docker/libkv/store" "github.com/docker/libnetwork/config" "github.com/docker/libnetwork/datastore" "github.com/docker/libnetwork/netlabel" "github.com/docker/libnetwork/options" ) func TestZooKeeperBackend(t *testing.T) { c, err := testNewController(t, "zk", "127.0.0.1:2181") if err != nil { t.Fatal(err) } c.Stop() } func testNewController(t *testing.T, provider, url string) (NetworkController, error) { cfgOptions, err := OptionBoltdbWithRandomDBFile() if err != nil { return nil, err } cfgOptions = append(cfgOptions, config.OptionKVProvider(provider)) cfgOptions = append(cfgOptions, config.OptionKVProviderURL(url)) return New(cfgOptions...) } func TestBoltdbBackend(t *testing.T) { defer os.Remove(defaultLocalStoreConfig.Client.Address) testLocalBackend(t, "", "", nil) defer os.Remove("/tmp/boltdb.db") config := &store.Config{Bucket: "testBackend", ConnectionTimeout: 3 * time.Second} testLocalBackend(t, "boltdb", "/tmp/boltdb.db", config) } func testLocalBackend(t *testing.T, provider, url string, storeConfig *store.Config) { cfgOptions := []config.Option{} cfgOptions = append(cfgOptions, config.OptionLocalKVProvider(provider)) cfgOptions = append(cfgOptions, config.OptionLocalKVProviderURL(url)) cfgOptions = append(cfgOptions, config.OptionLocalKVProviderConfig(storeConfig)) driverOptions := options.Generic{} genericOption := make(map[string]interface{}) genericOption[netlabel.GenericData] = driverOptions cfgOptions = append(cfgOptions, config.OptionDriverConfig("host", genericOption)) fmt.Printf("URL : %s\n", url) ctrl, err := New(cfgOptions...) if err != nil { t.Fatalf("Error new controller: %v", err) } nw, err := ctrl.NewNetwork("host", "host") if err != nil { t.Fatalf("Error creating default \"host\" network: %v", err) } ep, err := nw.CreateEndpoint("newendpoint", []EndpointOption{}...) if err != nil { t.Fatalf("Error creating endpoint: %v", err) } store := ctrl.(*controller).localStore.KVStore() if exists, err := store.Exists(datastore.Key(datastore.NetworkKeyPrefix, string(nw.ID()))); !exists || err != nil { t.Fatalf("Network key should have been created.") } if exists, err := store.Exists(datastore.Key([]string{datastore.EndpointKeyPrefix, string(nw.ID()), string(ep.ID())}...)); exists || err != nil { t.Fatalf("Endpoint key shouldn't have been created.") } store.Close() // test restore of local store ctrl, err = New(cfgOptions...) if err != nil { t.Fatalf("Error creating controller: %v", err) } if _, err = ctrl.NetworkByID(nw.ID()); err != nil { t.Fatalf("Error getting network %v", err) } } func TestNoPersist(t *testing.T) { cfgOptions, err := OptionBoltdbWithRandomDBFile() if err != nil { t.Fatalf("Error creating random boltdb file : %v", err) } ctrl, err := New(cfgOptions...) if err != nil { t.Fatalf("Error new controller: %v", err) } nw, err := ctrl.NewNetwork("host", "host", NetworkOptionPersist(false)) if err != nil { t.Fatalf("Error creating default \"host\" network: %v", err) } ep, err := nw.CreateEndpoint("newendpoint", []EndpointOption{}...) if err != nil { t.Fatalf("Error creating endpoint: %v", err) } store := ctrl.(*controller).localStore.KVStore() if exists, _ := store.Exists(datastore.Key(datastore.NetworkKeyPrefix, string(nw.ID()))); exists { t.Fatalf("Network with persist=false should not be stored in KV Store") } if exists, _ := store.Exists(datastore.Key([]string{datastore.EndpointKeyPrefix, string(nw.ID()), string(ep.ID())}...)); exists { t.Fatalf("Endpoint in Network with persist=false should not be stored in KV Store") } store.Close() } // OptionBoltdbWithRandomDBFile function returns a random dir for local store backend func OptionBoltdbWithRandomDBFile() ([]config.Option, error) { tmp, err := ioutil.TempFile("", "libnetwork-") if err != nil { return nil, fmt.Errorf("Error creating temp file: %v", err) } if err := tmp.Close(); err != nil { return nil, fmt.Errorf("Error closing temp file: %v", err) } cfgOptions := []config.Option{} cfgOptions = append(cfgOptions, config.OptionLocalKVProvider("boltdb")) cfgOptions = append(cfgOptions, config.OptionLocalKVProviderURL(tmp.Name())) sCfg := &store.Config{Bucket: "testBackend", ConnectionTimeout: 3 * time.Second} cfgOptions = append(cfgOptions, config.OptionLocalKVProviderConfig(sCfg)) return cfgOptions, nil } func TestLocalStoreLockTimeout(t *testing.T) { cfgOptions, err := OptionBoltdbWithRandomDBFile() if err != nil { t.Fatalf("Error getting random boltdb configs %v", err) } ctrl, err := New(cfgOptions...) if err != nil { t.Fatalf("Error new controller: %v", err) } defer ctrl.Stop() // Use the same boltdb file without closing the previous controller _, err = New(cfgOptions...) if err == nil { t.Fatalf("Multiple boldtdb connection must fail") } }