addressing comments

pull/6836/head
Sharif Elgamal 2020-03-04 14:09:15 -08:00
parent 8a732aeb8d
commit a3e000d850
3 changed files with 27 additions and 33 deletions

View File

@ -38,7 +38,7 @@ type CreateParams struct {
Name string // used for container name and hostname Name string // used for container name and hostname
Image string // container image to use to create the node. Image string // container image to use to create the node.
ClusterLabel string // label the clusters we create using minikube so we can clean up ClusterLabel string // label the clusters we create using minikube so we can clean up
NodeLabel string //l abel the nodes so we can clean up by node name NodeLabel string // label the nodes so we can clean up by node name
Role string // currently only role supported is control-plane Role string // currently only role supported is control-plane
Mounts []Mount // volume mounts Mounts []Mount // volume mounts
APIServerPort int // kubernetes api server port APIServerPort int // kubernetes api server port

View File

@ -214,7 +214,7 @@ func SetLibvirtURI(v string) {
} }
//MachineName returns the name of the machine given the cluster and node names // MachineName returns the name of the machine, as seen by the hypervisor given the cluster and node names
func MachineName(cc config.ClusterConfig, n config.Node) string { func MachineName(cc config.ClusterConfig, n config.Node) string {
// For single node cluster, default to back to old naming // For single node cluster, default to back to old naming
if len(cc.Nodes) == 1 || n.ControlPlane { if len(cc.Nodes) == 1 || n.ControlPlane {

View File

@ -60,17 +60,13 @@ func RegisterMockDriver(t *testing.T) {
} }
} }
var defaultNodeConfig = config.Node{
Name: viper.GetString("profile"),
}
var defaultClusterConfig = config.ClusterConfig{ var defaultClusterConfig = config.ClusterConfig{
Name: viper.GetString("profile"), Name: viper.GetString("profile"),
Driver: driver.Mock, Driver: driver.Mock,
MinikubeISO: constants.DefaultISOURL, MinikubeISO: constants.DefaultISOURL,
Downloader: MockDownloader{}, Downloader: MockDownloader{},
DockerEnv: []string{"MOCK_MAKE_IT_PROVISION=true"}, DockerEnv: []string{"MOCK_MAKE_IT_PROVISION=true"},
Nodes: []config.Node{defaultNodeConfig}, Nodes: []config.Node{config.Node{Name: "minikube"}},
} }
func TestCreateHost(t *testing.T) { func TestCreateHost(t *testing.T) {
@ -82,7 +78,7 @@ func TestCreateHost(t *testing.T) {
t.Fatal("Machine already exists.") t.Fatal("Machine already exists.")
} }
_, err := createHost(api, defaultClusterConfig, defaultNodeConfig) _, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatalf("Error creating host: %v", err) t.Fatalf("Error creating host: %v", err)
} }
@ -120,7 +116,7 @@ func TestStartHostExists(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
// Create an initial host. // Create an initial host.
ih, err := createHost(api, defaultClusterConfig, defaultNodeConfig) ih, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatalf("Error creating host: %v", err) t.Fatalf("Error creating host: %v", err)
} }
@ -137,8 +133,8 @@ func TestStartHostExists(t *testing.T) {
mc := defaultClusterConfig mc := defaultClusterConfig
mc.Name = ih.Name mc.Name = ih.Name
n := defaultNodeConfig n := config.Node{Name: ih.Name}
n.Name = ih.Name
// This should pass without calling Create because the host exists already. // This should pass without calling Create because the host exists already.
h, err := StartHost(api, mc, n) h, err := StartHost(api, mc, n)
if err != nil { if err != nil {
@ -160,7 +156,7 @@ func TestStartHostErrMachineNotExist(t *testing.T) {
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
// Create an incomplete host with machine does not exist error(i.e. User Interrupt Cancel) // Create an incomplete host with machine does not exist error(i.e. User Interrupt Cancel)
api.NotExistError = true api.NotExistError = true
h, err := createHost(api, defaultClusterConfig, defaultNodeConfig) h, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatalf("Error creating host: %v", err) t.Fatalf("Error creating host: %v", err)
} }
@ -171,8 +167,7 @@ func TestStartHostErrMachineNotExist(t *testing.T) {
mc := defaultClusterConfig mc := defaultClusterConfig
mc.Name = h.Name mc.Name = h.Name
n := defaultNodeConfig n := config.Node{Name: h.Name}
n.Name = h.Name
// This should pass with creating host, while machine does not exist. // This should pass with creating host, while machine does not exist.
h, err = StartHost(api, mc, n) h, err = StartHost(api, mc, n)
@ -206,7 +201,7 @@ func TestStartStoppedHost(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
// Create an initial host. // Create an initial host.
h, err := createHost(api, defaultClusterConfig, defaultNodeConfig) h, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatalf("Error creating host: %v", err) t.Fatalf("Error creating host: %v", err)
} }
@ -218,8 +213,7 @@ func TestStartStoppedHost(t *testing.T) {
provision.SetDetector(md) provision.SetDetector(md)
mc := defaultClusterConfig mc := defaultClusterConfig
mc.Name = h.Name mc.Name = h.Name
n := defaultNodeConfig n := config.Node{Name: h.Name}
n.Name = h.Name
h, err = StartHost(api, mc, n) h, err = StartHost(api, mc, n)
if err != nil { if err != nil {
t.Fatal("Error starting host.") t.Fatal("Error starting host.")
@ -248,7 +242,7 @@ func TestStartHost(t *testing.T) {
md := &tests.MockDetector{Provisioner: &tests.MockProvisioner{}} md := &tests.MockDetector{Provisioner: &tests.MockProvisioner{}}
provision.SetDetector(md) provision.SetDetector(md)
h, err := StartHost(api, defaultClusterConfig, defaultNodeConfig) h, err := StartHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatal("Error starting host.") t.Fatal("Error starting host.")
} }
@ -283,7 +277,7 @@ func TestStartHostConfig(t *testing.T) {
Downloader: MockDownloader{}, Downloader: MockDownloader{},
} }
h, err := StartHost(api, cfg, defaultNodeConfig) h, err := StartHost(api, cfg, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Fatal("Error starting host.") t.Fatal("Error starting host.")
} }
@ -313,14 +307,14 @@ func TestStopHostError(t *testing.T) {
func TestStopHost(t *testing.T) { func TestStopHost(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
h, err := createHost(api, defaultClusterConfig, defaultNodeConfig) h, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
cc := defaultClusterConfig cc := defaultClusterConfig
cc.Name = viper.GetString("profile") cc.Name = viper.GetString("profile")
m := driver.MachineName(cc, defaultNodeConfig) m := driver.MachineName(cc, config.Node{Name: "minikube"})
if err := StopHost(api, m); err != nil { if err := StopHost(api, m); err != nil {
t.Fatalf("Unexpected error stopping machine: %v", err) t.Fatalf("Unexpected error stopping machine: %v", err)
} }
@ -332,14 +326,14 @@ func TestStopHost(t *testing.T) {
func TestDeleteHost(t *testing.T) { func TestDeleteHost(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
if _, err := createHost(api, defaultClusterConfig, defaultNodeConfig); err != nil { if _, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"}); err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
cc := defaultClusterConfig cc := defaultClusterConfig
cc.Name = viper.GetString("profile") cc.Name = viper.GetString("profile")
if err := DeleteHost(api, driver.MachineName(cc, defaultNodeConfig)); err != nil { if err := DeleteHost(api, driver.MachineName(cc, config.Node{Name: "minikube"})); err != nil {
t.Fatalf("Unexpected error deleting host: %v", err) t.Fatalf("Unexpected error deleting host: %v", err)
} }
} }
@ -347,7 +341,7 @@ func TestDeleteHost(t *testing.T) {
func TestDeleteHostErrorDeletingVM(t *testing.T) { func TestDeleteHostErrorDeletingVM(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
h, err := createHost(api, defaultClusterConfig, defaultNodeConfig) h, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
@ -355,7 +349,7 @@ func TestDeleteHostErrorDeletingVM(t *testing.T) {
d := &tests.MockDriver{RemoveError: true, T: t} d := &tests.MockDriver{RemoveError: true, T: t}
h.Driver = d h.Driver = d
if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, defaultNodeConfig)); err == nil { if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, config.Node{Name: "minikube"})); err == nil {
t.Fatal("Expected error deleting host.") t.Fatal("Expected error deleting host.")
} }
} }
@ -364,11 +358,11 @@ func TestDeleteHostErrorDeletingFiles(t *testing.T) {
RegisterMockDriver(t) RegisterMockDriver(t)
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
api.RemoveError = true api.RemoveError = true
if _, err := createHost(api, defaultClusterConfig, defaultNodeConfig); err != nil { if _, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"}); err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, defaultNodeConfig)); err == nil { if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, config.Node{Name: "minikube"})); err == nil {
t.Fatal("Expected error deleting host.") t.Fatal("Expected error deleting host.")
} }
} }
@ -378,12 +372,12 @@ func TestDeleteHostErrMachineNotExist(t *testing.T) {
api := tests.NewMockAPI(t) api := tests.NewMockAPI(t)
// Create an incomplete host with machine does not exist error(i.e. User Interrupt Cancel) // Create an incomplete host with machine does not exist error(i.e. User Interrupt Cancel)
api.NotExistError = true api.NotExistError = true
_, err := createHost(api, defaultClusterConfig, defaultNodeConfig) _, err := createHost(api, defaultClusterConfig, config.Node{Name: "minikube"})
if err != nil { if err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, defaultNodeConfig)); err == nil { if err := DeleteHost(api, driver.MachineName(defaultClusterConfig, config.Node{Name: "minikube"})); err == nil {
t.Fatal("Expected error deleting host.") t.Fatal("Expected error deleting host.")
} }
} }
@ -395,7 +389,7 @@ func TestGetHostStatus(t *testing.T) {
cc := defaultClusterConfig cc := defaultClusterConfig
cc.Name = viper.GetString("profile") cc.Name = viper.GetString("profile")
m := driver.MachineName(cc, defaultNodeConfig) m := driver.MachineName(cc, config.Node{Name: "minikube"})
checkState := func(expected string, machineName string) { checkState := func(expected string, machineName string) {
s, err := GetHostStatus(api, machineName) s, err := GetHostStatus(api, machineName)
@ -409,13 +403,13 @@ func TestGetHostStatus(t *testing.T) {
checkState(state.None.String(), m) checkState(state.None.String(), m)
if _, err := createHost(api, cc, defaultNodeConfig); err != nil { if _, err := createHost(api, cc, config.Node{Name: "minikube"}); err != nil {
t.Errorf("createHost failed: %v", err) t.Errorf("createHost failed: %v", err)
} }
cc.Name = viper.GetString("profile") cc.Name = viper.GetString("profile")
m = driver.MachineName(cc, defaultNodeConfig) m = driver.MachineName(cc, config.Node{Name: "minikube"})
checkState(state.Running.String(), m) checkState(state.Running.String(), m)
@ -449,7 +443,7 @@ func TestCreateSSHShell(t *testing.T) {
cc.Name = viper.GetString("profile") cc.Name = viper.GetString("profile")
cliArgs := []string{"exit"} cliArgs := []string{"exit"}
if err := CreateSSHShell(api, cc, defaultNodeConfig, cliArgs); err != nil { if err := CreateSSHShell(api, cc, config.Node{Name: "minikube"}, cliArgs); err != nil {
t.Fatalf("Error running ssh command: %v", err) t.Fatalf("Error running ssh command: %v", err)
} }