Create OLM upgrade e2e scenario using codeflare SDK

Author: Srihari1192

test/e2e/olm_upgrade_test.go

@@ -42,9 +42,16 @@ func TestMNISTRayClusterUp(t *testing.T) {
 	test.T().Parallel()
 	if os.Getenv("RUN_OLM_TESTS") == "true" {
 		// Create a namespace
-		namespace := CreateTestNamespaceWithName(test, namespaceName)
+		namespace = CreateTestNamespaceWithName(test, namespaceName)
 		test.T().Logf("Created namespace %s successfully", namespace.Name)
 
+		// Delete namespace only if test failed
+		defer func() {
+			if t.Failed() {
+				DeleteTestNamespace(test, namespaceName)
+			}
+		}()
+
 		// Test configuration
 		config := &corev1.ConfigMap{
 			TypeMeta: metav1.TypeMeta{
@@ -212,6 +219,9 @@ func TestMnistJobSubmit(t *testing.T) {
 	test.T().Parallel()
 	if os.Getenv("RUN_OLM_TESTS") == "true" {
 
+		//delete the namespace after test complete
+		defer DeleteTestNamespace(test, namespaceName)
+
 		// Test configuration
 		config := &corev1.ConfigMap{
 			TypeMeta: metav1.TypeMeta{
@@ -315,6 +325,7 @@ func TestMnistJobSubmit(t *testing.T) {
 		// Assert the job has completed successfully
 		test.Expect(GetJob(test, job.Namespace, job.Name)).
 			To(WithTransform(ConditionStatus(batchv1.JobComplete), Equal(corev1.ConditionTrue)))
+
 	} else {
 		test.T().Skip("Skipping OLM upgarde test because RUN_OLM_TESTS is not set")
 	}

test/support/namespace.go

@@ -75,3 +75,12 @@ func CreateTestNamespaceWithName(t Test, namespaceName string, options ...Option
 
 	return namespace
 }
+
+func DeleteTestNamespace(t Test, namespaceName string) {
+	t.T().Helper()
+	propagationPolicy := metav1.DeletePropagationBackground
+	err := t.Client().Core().CoreV1().Namespaces().Delete(t.Ctx(), namespaceName, metav1.DeleteOptions{
+		PropagationPolicy: &propagationPolicy,
+	})
+	t.Expect(err).NotTo(gomega.HaveOccurred())
+}