more unit tests

services/etcd_test.go

@@ -0,0 +1,59 @@
+package services
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/rancher/rke/hosts"
+	"github.com/rancher/types/apis/management.cattle.io/v3"
+)
+
+const (
+	TestInitEtcdClusterString = "etcd-etcd1=http://1.1.1.1:2380,etcd-etcd2=http://2.2.2.2:2380"
+	TestEtcdImage             = "etcd/etcdImage:latest"
+	TestEtcdNamePrefix        = "--name=etcd-"
+	TestEtcdVolumeBind        = "/var/lib/etcd:/etcd-data"
+	TestEtcdExtraArgs         = "--foo=bar"
+)
+
+func TestEtcdConfig(t *testing.T) {
+	etcdHosts := []*hosts.Host{
+		&hosts.Host{
+			RKEConfigNode: v3.RKEConfigNode{
+				Address:          "1.1.1.1",
+				InternalAddress:  "1.1.1.1",
+				Role:             []string{"etcd"},
+				HostnameOverride: "etcd1",
+			},
+			DClient: nil,
+		},
+		&hosts.Host{
+			RKEConfigNode: v3.RKEConfigNode{
+				Address:          "2.2.2.2",
+				InternalAddress:  "2.2.2.2",
+				Role:             []string{"etcd"},
+				HostnameOverride: "etcd2",
+			},
+			DClient: nil,
+		},
+	}
+
+	etcdService := v3.ETCDService{}
+	etcdService.Image = TestEtcdImage
+	etcdService.ExtraArgs = map[string]string{"foo": "bar"}
+	// Test init cluster string
+	initCluster := getEtcdInitialCluster(etcdHosts)
+	assertEqual(t, initCluster, TestInitEtcdClusterString, "")
+
+	for _, host := range etcdHosts {
+		imageCfg, hostCfg := buildEtcdConfig(host, etcdService, TestInitEtcdClusterString)
+		assertEqual(t, isStringInSlice(TestEtcdNamePrefix+host.HostnameOverride, imageCfg.Cmd), true,
+			fmt.Sprintf("Failed to find [%s] in Etcd command", TestEtcdNamePrefix+host.HostnameOverride))
+		assertEqual(t, TestEtcdImage, imageCfg.Image,
+			fmt.Sprintf("Failed to verify [%s] as Etcd Image", TestEtcdImage))
+		assertEqual(t, isStringInSlice(TestEtcdVolumeBind, hostCfg.Binds), true,
+			fmt.Sprintf("Failed to find [%s] in volume binds of Etcd Service", TestEtcdVolumeBind))
+		assertEqual(t, isStringInSlice(TestEtcdExtraArgs, imageCfg.Entrypoint), true,
+			fmt.Sprintf("Failed to find [%s] in extra args of Etcd Service", TestEtcdExtraArgs))
+	}
+}