Merge pull request #387 from runcom/add-some-e2e

test/e2e: add MC smoke test

Author: openshift-merge-robot

test/e2e/mcd_test.go

@@ -1,23 +1,29 @@
 package e2e_test
 
 import (
-	"testing"
+	"fmt"
 	"strings"
+	"testing"
+	"time"
 
+	ignv2_2types "github.com/coreos/ignition/config/v2_2/types"
 	"k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/labels"
+	"k8s.io/apimachinery/pkg/util/uuid"
+	"k8s.io/apimachinery/pkg/util/wait"
 
 	"github.com/openshift/machine-config-operator/cmd/common"
+	mcv1 "github.com/openshift/machine-config-operator/pkg/apis/machineconfiguration.openshift.io/v1"
 )
 
 // Test case for https://github.com/openshift/machine-config-operator/issues/358
 func TestMCDToken(t *testing.T) {
 	cb, err := common.NewClientBuilder("")
-	if err != nil{
+	if err != nil {
 		t.Errorf("%#v", err)
 	}
-	k := cb.KubeClientOrDie("sanity-test")
+	k := cb.KubeClientOrDie("mcd-token-test")
 
 	listOptions := metav1.ListOptions{
 		LabelSelector: labels.SelectorFromSet(labels.Set{"k8s-app": "machine-config-daemon"}).String(),
@@ -40,3 +46,103 @@ func TestMCDToken(t *testing.T) {
 		}
 	}
 }
+
+func mcLabelForWorkers() map[string]string {
+	mcLabels := make(map[string]string)
+	mcLabels["machineconfiguration.openshift.io/role"] = "worker"
+	return mcLabels
+}
+
+func createMCFile(path, content string, mode int) ignv2_2types.File {
+	return ignv2_2types.File{
+		FileEmbedded1: ignv2_2types.FileEmbedded1{
+			Contents: ignv2_2types.FileContents{
+				Source: content,
+			},
+			Mode: &mode,
+		},
+		Node: ignv2_2types.Node{
+			Filesystem: "root",
+			Path:       path,
+		},
+	}
+}
+
+func TestMCDeployed(t *testing.T) {
+	cb, err := common.NewClientBuilder("")
+	if err != nil {
+		t.Errorf("%#v", err)
+	}
+	mcClient := cb.MachineConfigClientOrDie("mc-file-add")
+	k := cb.KubeClientOrDie("mc-file-add")
+
+	// create a dummy MC
+	mcName := fmt.Sprintf("00-0add-a-file-%s", uuid.NewUUID())
+	mcadd := &mcv1.MachineConfig{}
+	mcadd.ObjectMeta = metav1.ObjectMeta{
+		Name:   mcName,
+		Labels: mcLabelForWorkers(),
+	}
+	mcadd.Spec = mcv1.MachineConfigSpec{
+		Config: ignv2_2types.Config{
+			Ignition: ignv2_2types.Ignition{
+				Version: "2.2.0",
+			},
+			Storage: ignv2_2types.Storage{
+				Files: []ignv2_2types.File{
+					createMCFile("/etc/mytestconf", "data:,test", 420),
+				},
+			},
+		},
+	}
+
+	// create the dummy MC now
+	_, err = mcClient.MachineconfigurationV1().MachineConfigs().Create(mcadd)
+	if err != nil {
+		t.Errorf("failed to create machine config %v", err)
+	}
+
+	// grab the latest worker- MC
+	var newMCName string
+	err = wait.Poll(2*time.Second, 5*time.Minute, func() (bool, error) {
+		mcp, err := mcClient.MachineconfigurationV1().MachineConfigPools().Get("worker", metav1.GetOptions{})
+		if err != nil {
+			return false, err
+		}
+		for _, mc := range mcp.Status.Configuration.Source {
+			if mc.Name == mcName {
+				newMCName = mcp.Status.Configuration.Name
+				return true, nil
+			}
+		}
+		return false, nil
+	})
+
+	listOptions := metav1.ListOptions{
+		LabelSelector: labels.SelectorFromSet(labels.Set{"k8s-app": "machine-config-daemon"}).String(),
+	}
+
+	err = wait.Poll(3*time.Second, 5*time.Minute, func() (bool, error) {
+		mcdList, err := k.CoreV1().Pods("openshift-machine-config-operator").List(listOptions)
+		if err != nil {
+			return false, err
+		}
+
+		for _, pod := range mcdList.Items {
+			res, err := k.CoreV1().Pods(pod.Namespace).GetLogs(pod.Name, &v1.PodLogOptions{}).DoRaw()
+			if err != nil {
+				// do not error out, we may be rebooting, that's why we list at every iteration
+				return false, nil
+			}
+			for _, line := range strings.Split(string(res), "\n") {
+				if strings.Contains(line, "completed update for config "+newMCName) {
+					return true, nil
+				}
+			}
+		}
+		return false, nil
+	})
+	if err != nil {
+		t.Errorf("machine config didn't result in file being on any worker: %v", err)
+	}
+}