| package libnetwork |
| |
| import ( |
| "fmt" |
| "os" |
| "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 := os.CreateTemp("", "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") |
| } |
| } |