Add tests for the Migration

Signed-off-by: Goutham Veeramachaneni <cs14btech11014@iith.ac.in>

Makefile

@@ -30,6 +30,7 @@ container:
 	docker build -t $(REPO):$(TAG) .
 
 e2e-test:
+	go test -timeout 20m -v ./test/migration/ $(TEST_RUN_ARGS) --kubeconfig "$(HOME)/.kube/config" --operator-image=$(REPO):$(TAG) --namespace=$(NAMESPACE) --cluster-ip=$(CLUSTER_IP)
 	go test -timeout 20m -v ./test/e2e/ $(TEST_RUN_ARGS) --kubeconfig "$(HOME)/.kube/config" --operator-image=$(REPO):$(TAG) --namespace=$(NAMESPACE) --cluster-ip=$(CLUSTER_IP)
 
 e2e-status:

test/e2e/framework/framework.go

@@ -78,7 +78,7 @@ func New(ns, kubeconfig, opImage, ip string) (*Framework, error) {
 		DefaultTimeout: time.Minute,
 	}
 
-	err = f.setup(opImage)
+	err = f.Setup(opImage)
 	if err != nil {
 		return nil, err
 	}
@@ -86,7 +86,7 @@ func New(ns, kubeconfig, opImage, ip string) (*Framework, error) {
 	return f, nil
 }
 
-func (f *Framework) setup(opImage string) error {
+func (f *Framework) Setup(opImage string) error {
 	if err := f.setupPrometheusOperator(opImage); err != nil {
 		return err
 	}

test/migration/migration_test.go

@@ -0,0 +1,202 @@
+package migration
+
+import (
+	"flag"
+	"log"
+	"os"
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/coreos/prometheus-operator/pkg/client/monitoring/v1alpha1"
+	"github.com/coreos/prometheus-operator/pkg/k8sutil"
+	operatorFramework "github.com/coreos/prometheus-operator/test/e2e/framework"
+	apiextensionsclient "k8s.io/apiextensions-apiserver/pkg/client/clientset/clientset"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/client-go/kubernetes"
+	"k8s.io/client-go/tools/clientcmd"
+)
+
+var framework *operatorFramework.Framework
+var kubeconfig, opImage, ns, ip *string
+
+func TestMain(m *testing.M) {
+	kubeconfig = flag.String("kubeconfig", "", "kube config path, e.g. $HOME/.kube/config")
+	opImage = flag.String("operator-image", "", "operator image, e.g. quay.io/coreos/prometheus-operator")
+	ns = flag.String("namespace", "prometheus-operator-e2e-tests", "e2e test namespace")
+	ip = flag.String("cluster-ip", "", "ip of the kubernetes cluster to use for external requests")
+	flag.Parse()
+
+	os.Exit(m.Run())
+}
+
+func TestMigration(t *testing.T) {
+	// Create all the clients required.
+	config, err := clientcmd.BuildConfigFromFlags("", *kubeconfig)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	kclient, err := kubernetes.NewForConfig(config)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	mv, err := k8sutil.GetMinorVersion(kclient.Discovery())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if mv < 7 {
+		t.Skip("lower than 1.7 version")
+		return
+	}
+
+	extClient, err := apiextensionsclient.NewForConfig(config)
+	if err != nil {
+		t.Fatal(err)
+	}
+	monClient, err := v1alpha1.NewForConfig(config)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	tprClient := kclient.ExtensionsV1beta1().ThirdPartyResources()
+	crdClient := extClient.ApiextensionsV1beta1().CustomResourceDefinitions()
+
+	if framework, err = operatorFramework.New(
+		*ns,
+		*kubeconfig,
+		"quay.io/coreos/prometheus-operator:v0.10.2",
+		*ip,
+	); err != nil {
+		log.Printf("failed to setup framework: %v\n", err)
+		t.Fatal(err)
+	}
+
+	ctx := framework.NewTestCtx(t)
+	defer ctx.Cleanup(t)
+	ns2 := ctx.CreateNamespace(t, framework.KubeClient)
+
+	// Launch the objects.
+	name := "test"
+	group := "servicediscovery-test"
+
+	prometheusTPR := framework.MakeBasicPrometheus(ns2, name, name, 1)
+	prometheusTPR.Namespace = ns2
+	if err := framework.CreatePrometheusAndWaitUntilReady(ns2, prometheusTPR); err != nil {
+		t.Fatal(err)
+	}
+
+	s := framework.MakeBasicServiceMonitor(group)
+	if _, err := framework.MonClient.ServiceMonitors(ns2).Create(s); err != nil {
+		t.Fatal("Creating ServiceMonitor failed: ", err)
+	}
+
+	if err := framework.CreateAlertmanagerAndWaitUntilReady(ns2, framework.MakeBasicAlertmanager(name, 3)); err != nil {
+		t.Fatal(err)
+	}
+
+	obj, err := monClient.Prometheuses(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	proms := obj.(*v1alpha1.PrometheusList)
+
+	// Get the objects.
+	obj, err = monClient.Alertmanagers(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	ams := obj.(*v1alpha1.AlertmanagerList)
+
+	obj, err = monClient.ServiceMonitors(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	sms := obj.(*v1alpha1.ServiceMonitorList)
+
+	// Delete and launch new operator.
+	if err := framework.
+		KubeClient.
+		Extensions().
+		Deployments(framework.Namespace.Name).
+		Delete("prometheus-operator", nil); err != nil {
+		t.Fatal(err)
+	}
+
+	// TODO: Wait until terminated.
+	time.Sleep(15 * time.Second)
+
+	if err := framework.Setup(*opImage); err != nil {
+		t.Fatal(err)
+	}
+
+	// Check if TPRs are deleted.
+	tprList, err := tprClient.List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(tprList.Items) != 0 {
+		t.Fatalf("expected 0 TPRs got %d", len(tprList.Items))
+	}
+
+	// Check if CRDs are created.
+	crdList, err := crdClient.List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(crdList.Items) != 3 {
+		t.Fatalf("expected 3 TPRs got %d", len(crdList.Items))
+	}
+
+	// Compare old and new objects.
+	obj, err = monClient.Prometheuses(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	promsNew := obj.(*v1alpha1.PrometheusList)
+	if len(promsNew.Items) != len(proms.Items) {
+		t.Fatal("expected %d proms, got %d", len(proms.Items), len(promsNew.Items))
+	}
+
+	for i, prom := range proms.Items {
+		if !reflect.DeepEqual(prom.Spec, promsNew.Items[i].Spec) {
+			t.Fatal("yolo")
+		}
+	}
+
+	obj, err = monClient.Alertmanagers(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	amsNew := obj.(*v1alpha1.AlertmanagerList)
+	if len(amsNew.Items) != len(ams.Items) {
+		t.Fatal("expected %d ams, got %d", len(ams.Items), len(amsNew.Items))
+	}
+
+	for i, am := range ams.Items {
+		if !reflect.DeepEqual(am.Spec, amsNew.Items[i].Spec) {
+			t.Fatal("yolo")
+		}
+	}
+
+	obj, err = monClient.ServiceMonitors(ns2).List(metav1.ListOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	smsNew := obj.(*v1alpha1.ServiceMonitorList)
+	if len(smsNew.Items) != len(sms.Items) {
+		t.Fatal("expected %d ams, got %d", len(sms.Items), len(smsNew.Items))
+	}
+
+	for i, sm := range sms.Items {
+		if !reflect.DeepEqual(sm.Spec, smsNew.Items[i].Spec) {
+			t.Fatal("yolo")
+		}
+	}
+
+	if err := framework.Teardown(); err != nil {
+		t.Fatal(err)
+	}
+}