add unit test

sofastack · Sep 24, 2023 · 1278f01 · 1278f01
1 parent 96de3f9
commit 1278f01
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/module-controller/internal/controller/module_controller_suit_test.go b/module-controller/internal/controller/module_controller_suit_test.go
@@ -78,12 +78,14 @@ var _ = Describe("Module Controller", func() {
 
 	Context("delete module deployment with ip by deleting module", func() {
 		moduleReplicaSetName := "test-modulereplicaset"
+		updateModuleUrl := "https://module-test-url"
 		It("should be deleted and recreate a new one", func() {
 			module.Labels[label.ModuleReplicasetLabel] = moduleReplicaSetName
 			module.Labels[label.ModuleNameLabel] = "test-module"
 			Expect(k8sClient.Update(context.TODO(), &module)).Should(Succeed())
 			Expect(k8sClient.Delete(context.TODO(), &module)).Should(Succeed())
 			moduleReplicaSet := prepareModuleReplicaSet(namespace, moduleReplicaSetName)
+			moduleReplicaSet.Spec.Template.Spec.Module.Url = updateModuleUrl
 			Expect(k8sClient.Create(context.TODO(), &moduleReplicaSet)).Should(Succeed())
 			key := types.NamespacedName{
 				Name:      moduleName,
@@ -97,7 +99,7 @@ var _ = Describe("Module Controller", func() {
 					err = k8sClient.List(context.TODO(), modules, &client.ListOptions{Namespace: module.Namespace, LabelSelector: selector})
 					if err == nil && len(modules.Items) > 0 {
 						module = modules.Items[0]
-						return true
+						return module.Spec.Module.Url == updateModuleUrl
 					}
 				}
 				return false