Local update test

ytsaurus · Apr 15, 2024 · 069eeb6 · 069eeb6
1 parent 305b07a
commit 069eeb6
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/controllers/ytsaurus_local_test.go b/controllers/ytsaurus_local_test.go
@@ -114,7 +114,7 @@ func TestYtsaurusUpdateStatelessComponent(t *testing.T) {
 		"cluster state is updating",
 		func(obj client.Object) bool {
 			state := obj.(*ytv1.Ytsaurus).Status.State
-			return state == ytv1.ClusterStateUpdating
+			return state == ytv1.ClusterStateRunning
 		},
 	)
 
@@ -124,6 +124,17 @@ func TestYtsaurusUpdateStatelessComponent(t *testing.T) {
 	ytsaurusResource.Spec.EnableFullUpdate = false
 	testutil.UpdateObject(h, &ytv1.Ytsaurus{}, &ytsaurusResource)
 
+	t.Log("[ Wait for YTsaurus Updating status ]")
+	testutil.FetchAndCheckEventually(
+		h,
+		ytsaurusName,
+		&ytv1.Ytsaurus{},
+		"cluster state is running",
+		func(obj client.Object) bool {
+			state := obj.(*ytv1.Ytsaurus).Status.State
+			return state == ytv1.ClusterStateRunning
+		},
+	)
 	t.Log("[ Wait for YTsaurus Running status ]")
 	testutil.FetchAndCheckEventually(
 		h,