Skip to content

Commit bb53bda

Browse files
committed
wip
1 parent 5f8e48e commit bb53bda

File tree

1 file changed

+26
-50
lines changed

1 file changed

+26
-50
lines changed

e2e/tests/smoke_test.go

Lines changed: 26 additions & 50 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,19 @@ func checkPodsRunningAndReady(ctx context.Context, podLabelKey, podLabelValue st
146146
}, test.Timeout, test.Interval).Should(BeTrue())
147147
}
148148

149+
func deleteStorageSafely(ctx context.Context, storageSample *v1alpha1.Storage) {
150+
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
151+
152+
Eventually(func() bool {
153+
fetched := v1alpha1.Storage{}
154+
err := k8sClient.Get(ctx, types.NamespacedName{
155+
Name: storageSample.Name,
156+
Namespace: testobjects.YdbNamespace,
157+
}, &fetched)
158+
return apierrors.IsNotFound(err)
159+
}, Timeout, Interval).Should(BeTrue())
160+
}
161+
149162
func bringYdbCliToPod(podName, podNamespace string) {
150163
Eventually(func(g Gomega) error {
151164
args := []string{
@@ -305,9 +318,7 @@ var _ = Describe("Operator smoke test", func() {
305318
It("Check webhook defaulter", func() {
306319
emptyStorageDefaultFields(storageSample)
307320
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
308-
defer func() {
309-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
310-
}()
321+
defer deleteStorageSafely(ctx, storageSample)
311322

312323
emptyDatabaseDefaultFields(databaseSample)
313324
Expect(k8sClient.Create(ctx, databaseSample)).Should(Succeed())
@@ -330,17 +341,13 @@ var _ = Describe("Operator smoke test", func() {
330341
},
331342
}
332343
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
333-
defer func() {
334-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
335-
}()
344+
defer deleteStorageSafely(ctx, storageSample)
336345
})
337346

338347
It("general smoke pipeline, create storage + database", func() {
339348
By("issuing create commands...")
340349
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
341-
defer func() {
342-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
343-
}()
350+
defer deleteStorageSafely(ctx, storageSample)
344351
Expect(k8sClient.Create(ctx, databaseSample)).Should(Succeed())
345352
defer func() {
346353
Expect(k8sClient.Delete(ctx, databaseSample)).Should(Succeed())
@@ -381,9 +388,7 @@ var _ = Describe("Operator smoke test", func() {
381388
It("pause and un-pause Storage, should destroy and bring up Pods", func() {
382389
By("issuing create commands...")
383390
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
384-
defer func() {
385-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
386-
}()
391+
defer deleteStorageSafely(ctx, storageSample)
387392

388393
By("waiting until Storage is ready...")
389394
waitUntilStorageReady(ctx, storageSample.Name, testobjects.YdbNamespace)
@@ -431,9 +436,7 @@ var _ = Describe("Operator smoke test", func() {
431436
It("freeze + delete StatefulSet + un-freeze Storage", func() {
432437
By("issuing create commands...")
433438
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
434-
defer func() {
435-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
436-
}()
439+
defer deleteStorageSafely(ctx, storageSample)
437440

438441
By("waiting until Storage is ready...")
439442
waitUntilStorageReady(ctx, storageSample.Name, testobjects.YdbNamespace)
@@ -503,20 +506,6 @@ var _ = Describe("Operator smoke test", func() {
503506

504507
By("checking that all the storage pods are running and ready...")
505508
checkPodsRunningAndReady(ctx, "ydb-cluster", "kind-storage", storageSample.Spec.Nodes)
506-
507-
/*
508-
// This test suite attempts to create a database on uninitialised storage
509-
510-
By("database can be healthily created after Frozen storage...")
511-
Expect(k8sClient.Create(ctx, databaseSample)).Should(Succeed())
512-
defer func() {
513-
Expect(k8sClient.Delete(ctx, databaseSample)).Should(Succeed())
514-
}()
515-
By("waiting until database is ready...")
516-
waitUntilDatabaseReady(ctx, databaseSample.Name, testobjects.YdbNamespace)
517-
By("checking that all the database pods are running and ready...")
518-
checkPodsRunningAndReady(ctx, "ydb-cluster", "kind-database", databaseSample.Spec.Nodes)
519-
*/
520509
})
521510

522511
It("create storage and database with nodeSets", func() {
@@ -538,9 +527,7 @@ var _ = Describe("Operator smoke test", func() {
538527
})
539528
}
540529
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
541-
defer func() {
542-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
543-
}()
530+
defer deleteStorageSafely(ctx, storageSample)
544531
Expect(k8sClient.Create(ctx, databaseSample)).Should(Succeed())
545532
defer func() {
546533
Expect(k8sClient.Delete(ctx, databaseSample)).Should(Succeed())
@@ -619,9 +606,7 @@ var _ = Describe("Operator smoke test", func() {
619606
},
620607
}
621608
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
622-
defer func() {
623-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
624-
}()
609+
defer deleteStorageSafely(ctx, storageSample)
625610

626611
By("waiting until Storage is ready...")
627612
waitUntilStorageReady(ctx, storageSample.Name, testobjects.YdbNamespace)
@@ -632,9 +617,7 @@ var _ = Describe("Operator smoke test", func() {
632617

633618
It("storage.State goes Pending -> Preparing -> Initializing -> Provisioning -> Ready", func() {
634619
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
635-
defer func() {
636-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
637-
}()
620+
defer deleteStorageSafely(ctx, storageSample)
638621

639622
By("waiting until Storage is ready...")
640623
waitUntilStorageReady(ctx, storageSample.Name, testobjects.YdbNamespace)
@@ -688,9 +671,7 @@ var _ = Describe("Operator smoke test", func() {
688671
Key: "ca.crt",
689672
}
690673
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
691-
defer func() {
692-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
693-
}()
674+
defer deleteStorageSafely(ctx, storageSample)
694675

695676
By("waiting until Storage is ready...")
696677
waitUntilStorageReady(ctx, storageSample.Name, testobjects.YdbNamespace)
@@ -742,6 +723,7 @@ var _ = Describe("Operator smoke test", func() {
742723
It("Check that Storage deleted after Database...", func() {
743724
By("create storage...")
744725
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
726+
defer deleteStorageSafely(ctx, storageSample)
745727

746728
By("create database...")
747729
Expect(k8sClient.Create(ctx, databaseSample)).Should(Succeed())
@@ -803,9 +785,7 @@ var _ = Describe("Operator smoke test", func() {
803785
storageSample = testobjects.DefaultStorage(filepath.Join(".", "data", "storage-mirror-3-dc-dynconfig.yaml"))
804786

805787
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
806-
defer func() {
807-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
808-
}()
788+
defer deleteStorageSafely(ctx, storageSample)
809789

810790
storage := v1alpha1.Storage{}
811791
By("waiting until StorageInitialized condition is true...")
@@ -919,9 +899,7 @@ var _ = Describe("Operator smoke test", func() {
919899
},
920900
}
921901
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
922-
defer func() {
923-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
924-
}()
902+
defer deleteStorageSafely(ctx, storageSample)
925903

926904
By("create database...")
927905
databaseSample.Spec.Nodes = 1
@@ -959,9 +937,7 @@ var _ = Describe("Operator smoke test", func() {
959937
It("Check encryption for Database", func() {
960938
By("create storage...")
961939
Expect(k8sClient.Create(ctx, storageSample)).Should(Succeed())
962-
defer func() {
963-
Expect(k8sClient.Delete(ctx, storageSample)).Should(Succeed())
964-
}()
940+
defer deleteStorageSafely(ctx, storageSample)
965941
By("create database...")
966942
databaseSample.Spec.Encryption = &v1alpha1.EncryptionConfig{
967943
Enabled: true,

0 commit comments

Comments
 (0)