package libnetwork import ( "fmt" "io/ioutil" "testing" "github.com/docker/docker/libnetwork/config" "github.com/docker/docker/libnetwork/datastore" "github.com/docker/docker/libnetwork/netlabel" "github.com/docker/docker/libnetwork/options" "github.com/docker/libkv/store" ) 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)) 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).getStore(datastore.LocalScope).KVStore() if exists, err := store.Exists(datastore.Key(datastore.NetworkKeyPrefix, nw.ID())); !exists || err != nil { t.Fatalf("Network key should have been created.") } if exists, err := store.Exists(datastore.Key([]string{datastore.EndpointKeyPrefix, nw.ID(), ep.ID()}...)); !exists || err != nil { t.Fatalf("Endpoint key should 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) } } // 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"} cfgOptions = append(cfgOptions, config.OptionLocalKVProviderConfig(sCfg)) return cfgOptions, nil } func TestMultipleControllersWithSameStore(t *testing.T) { cfgOptions, err := OptionBoltdbWithRandomDBFile() if err != nil { t.Fatalf("Error getting random boltdb configs %v", err) } ctrl1, err := New(cfgOptions...) if err != nil { t.Fatalf("Error new controller: %v", err) } defer ctrl1.Stop() // Use the same boltdb file without closing the previous controller _, err = New(cfgOptions...) if err != nil { t.Fatalf("Local store must support concurrent controllers") } }