107 lines
2.1 KiB
Go
107 lines
2.1 KiB
Go
package network
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestCreate(t *testing.T) {
|
|
var (
|
|
h = newIntegrationHarness(t)
|
|
network = h.createNetwork(t, "primus", nil)
|
|
)
|
|
|
|
gotCreationParams, err := LoadCreationParams(network.stateDir)
|
|
if err != nil {
|
|
t.Fatalf("calling LoadCreationParams: %v", err)
|
|
} else if network.creationParams != gotCreationParams {
|
|
t.Fatalf(
|
|
"expected CreationParams %+v, got %+v",
|
|
network.creationParams,
|
|
gotCreationParams,
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestLoad(t *testing.T) {
|
|
var (
|
|
h = newIntegrationHarness(t)
|
|
network = h.createNetwork(t, "primus", &createNetworkOpts{
|
|
manualShutdown: true,
|
|
})
|
|
)
|
|
|
|
t.Log("Shutting down network")
|
|
if err := network.Shutdown(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
t.Log("Calling Load")
|
|
loadedNetwork, err := Load(
|
|
h.ctx,
|
|
h.logger.WithNamespace("loadedNetwork"),
|
|
network.networkConfig,
|
|
getEnvBinDirPath(),
|
|
network.stateDir,
|
|
h.mkDir(t, "runtime"),
|
|
network.opts,
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("Load failed: %v", err)
|
|
}
|
|
t.Cleanup(func() {
|
|
t.Log("Shutting down loadedNetwork")
|
|
if err := loadedNetwork.Shutdown(); err != nil {
|
|
t.Logf("Shutting down loadedNetwork failed: %v", err)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestJoin(t *testing.T) {
|
|
var (
|
|
h = newIntegrationHarness(t)
|
|
primus = h.createNetwork(t, "primus", nil)
|
|
secondus = h.joinNetwork(t, primus, "secondus", nil)
|
|
)
|
|
|
|
primusHosts, err := primus.GetHosts(h.ctx)
|
|
if err != nil {
|
|
t.Fatalf("getting hosts from primus: %v", err)
|
|
}
|
|
|
|
secondusHosts, err := secondus.GetHosts(h.ctx)
|
|
if err != nil {
|
|
t.Fatalf("getting hosts from secondus: %v", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(primusHosts, secondusHosts) {
|
|
t.Fatalf(
|
|
"expected primusHosts %+v to equal secondusHosts %+v",
|
|
primusHosts,
|
|
secondusHosts,
|
|
)
|
|
}
|
|
}
|
|
|
|
func TestNetwork_GetConfig(t *testing.T) {
|
|
var (
|
|
h = newIntegrationHarness(t)
|
|
network = h.createNetwork(t, "primus", nil)
|
|
)
|
|
|
|
config, err := network.GetConfig(h.ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(config, network.networkConfig) {
|
|
t.Fatalf(
|
|
"Config doesn't match the one used to create the Network\n"+
|
|
"exp: %+v\n"+
|
|
"got: %+v",
|
|
config,
|
|
network.networkConfig,
|
|
)
|
|
}
|
|
}
|