mirror of
https://github.com/moby/moby.git
synced 2026-08-13 17:06:44 +00:00
6a15f40803
Now that this function is only ever called from contexts where a *testing.T value is available, several improvements can be made to it. Refactor it to be a test helper function so that callers do not need to check and fail the test themselves. Leverage t.TempDir() so that the temporary file is cleaned up automatically. Change it to return a single config.Option to get better ergonomics at call sites. Signed-off-by: Cory Snider <csnider@mirantis.com>
88 lines
2.8 KiB
Go
88 lines
2.8 KiB
Go
package libnetwork
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"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)
|
|
}
|
|
defer ctrl.Stop()
|
|
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)
|
|
}
|
|
defer ctrl.Stop()
|
|
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(t *testing.T) config.Option {
|
|
t.Helper()
|
|
tmp := filepath.Join(t.TempDir(), "bolt.db")
|
|
if err := os.WriteFile(tmp, nil, 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return func(c *config.Config) {
|
|
config.OptionLocalKVProvider("boltdb")(c)
|
|
config.OptionLocalKVProviderURL(tmp)(c)
|
|
config.OptionLocalKVProviderConfig(&store.Config{Bucket: "testBackend"})(c)
|
|
}
|
|
}
|
|
|
|
func TestMultipleControllersWithSameStore(t *testing.T) {
|
|
cfgOptions := OptionBoltdbWithRandomDBFile(t)
|
|
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
|
|
ctrl2, err := New(cfgOptions)
|
|
if err != nil {
|
|
t.Fatalf("Local store must support concurrent controllers")
|
|
}
|
|
ctrl2.Stop()
|
|
}
|