Add tests for database and distributed architecture

Signed-off-by: Dmitry Tantsur <[email protected]>

Author: dtantsur

test/suite_test.go

@@ -29,6 +29,7 @@ import (
 	. "github.com/onsi/gomega"
 
 	corev1 "k8s.io/api/core/v1"
+	k8serrors "k8s.io/apimachinery/pkg/api/errors"
 	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
@@ -82,6 +83,42 @@ var _ = BeforeSuite(func() {
 
 })
 
+func WaitForIronic(name types.NamespacedName) *metal3api.Ironic {
+	GinkgoHelper()
+
+	ironic := &metal3api.Ironic{}
+
+	By("waiting for Ironic deployment")
+
+	var cond *metav1.Condition
+	for attempt := 0; attempt < 10; attempt += 1 {
+		err := k8sClient.Get(ctx, name, ironic)
+		Expect(err).NotTo(HaveOccurred())
+
+		cond = meta.FindStatusCondition(ironic.Status.Conditions, string(metal3api.IronicStatusAvailable))
+		if cond != nil && cond.Status == metav1.ConditionTrue {
+			break
+		}
+
+		time.Sleep(5 * time.Second)
+	}
+
+	if cond == nil || cond.Status != metav1.ConditionTrue {
+		Fail("Ironic deployment never succeded")
+	}
+
+	return ironic
+}
+
+func VerifyIronic(ironic *metal3api.Ironic) {
+	GinkgoHelper()
+
+	By("checking the service")
+
+	_, err := clientset.CoreV1().Services(ironic.Namespace).Get(ctx, ironic.Name, metav1.GetOptions{})
+	Expect(err).NotTo(HaveOccurred())
+}
+
 var _ = Describe("Ironic object tests", func() {
 	var namespace string
 
@@ -93,6 +130,16 @@ var _ = Describe("Ironic object tests", func() {
 		Expect(err).NotTo(HaveOccurred())
 		DeferCleanup(func() {
 			_ = clientset.CoreV1().Namespaces().Delete(ctx, namespace, metav1.DeleteOptions{})
+			for attempt := 0; attempt < 10; attempt += 1 {
+				_, err := clientset.CoreV1().Namespaces().Get(ctx, namespace, metav1.GetOptions{})
+				if err != nil && k8serrors.IsNotFound(err) {
+					break
+				} else {
+					Expect(err).NotTo(HaveOccurred())
+				}
+
+				time.Sleep(5 * time.Second)
+			}
 		})
 	})
 
@@ -102,33 +149,53 @@ var _ = Describe("Ironic object tests", func() {
 			Namespace: namespace,
 		}
 
-		ironic := metal3api.Ironic{
+		ironic := &metal3api.Ironic{
 			ObjectMeta: metav1.ObjectMeta{
 				Name:      name.Name,
 				Namespace: name.Namespace,
 			},
 		}
-		err := k8sClient.Create(ctx, &ironic)
+		err := k8sClient.Create(ctx, ironic)
 		Expect(err).NotTo(HaveOccurred())
 
-		var cond *metav1.Condition
-		for attempt := 0; attempt < 10; attempt += 1 {
-			err = k8sClient.Get(ctx, name, &ironic)
-			Expect(err).NotTo(HaveOccurred())
+		ironic = WaitForIronic(name)
+		VerifyIronic(ironic)
+	})
 
-			cond = meta.FindStatusCondition(ironic.Status.Conditions, string(metal3api.IronicStatusAvailable))
-			if cond != nil && cond.Status == metav1.ConditionTrue {
-				break
-			}
+	It("creates distributed Ironic", func() {
+		name := types.NamespacedName{
+			Name:      "test-ironic",
+			Namespace: namespace,
+		}
 
-			time.Sleep(5 * time.Second)
+		ironicDb := &metal3api.IronicDatabase{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      fmt.Sprintf("%s-db", name.Name),
+				Namespace: name.Namespace,
+			},
 		}
 
-		if cond == nil || cond.Status != metav1.ConditionTrue {
-			Fail("Ironic deployment never succeded")
+		err := k8sClient.Create(ctx, ironicDb)
+		Expect(err).NotTo(HaveOccurred())
+
+		ironic := &metal3api.Ironic{
+			ObjectMeta: metav1.ObjectMeta{
+				Name:      name.Name,
+				Namespace: name.Namespace,
+			},
+			Spec: metal3api.IronicSpec{
+				DatabaseRef: corev1.LocalObjectReference{
+					Name: ironicDb.Name,
+				},
+				Distributed: true,
+			},
 		}
-	})
+		err = k8sClient.Create(ctx, ironic)
+		Expect(err).NotTo(HaveOccurred())
 
+		ironic = WaitForIronic(name)
+		VerifyIronic(ironic)
+	})
 })
 
 var _ = AfterSuite(func() {