e2e: fix dbops-restart-with-failed-pod

Author: teoincontatto

stackgres-k8s/e2e/spec/dbops-restart-with-failed-pod

@@ -30,6 +30,8 @@ check_cluster_start_not_fail_if_cluster_failed() {
   set_restarted_pods
   kubectl patch sgcluster -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" --type json \
     -p '[{"op":"add","path":"/spec/pods/scheduling","value":{"nodeSelector":{"test":"test"}}}]'
+  wait_until eval 'kubectl get sts -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" -o json \
+    | jq .spec.nodeSelector.test | grep -q test'
 
   cat << EOF | kubectl create -f -
 apiVersion: stackgres.io/v1
@@ -49,6 +51,8 @@ EOF
 
   wait_until kubectl patch sgcluster -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" --type json \
     -p '[{"op":"remove","path":"/spec/pods/scheduling"}]'
+  wait_until eval 'kubectl get sts -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" -o json \
+    | jq .spec.nodeSelector | grep -q null'
 
   cat << EOF | kubectl create -f -
 apiVersion: stackgres.io/v1
@@ -77,6 +81,8 @@ check_cluster_start_not_fail_if_pod_failed() {
   set_restarted_pods dbops-restart-with-failed-pod-1 dbops-restart-with-failed-pod-0
   kubectl patch sgcluster -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" --type json \
     -p '[{"op":"add","path":"/spec/pods/scheduling","value":{"nodeSelector":{"test":"test"}}}]'
+  wait_until eval 'kubectl get sts -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" -o json \
+    | jq .spec.nodeSelector.test | grep -q test'
 
   cat << EOF | kubectl create -f -
 apiVersion: stackgres.io/v1
@@ -96,6 +102,8 @@ EOF
 
   wait_until kubectl patch sgcluster -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" --type json \
     -p '[{"op":"remove","path":"/spec/pods/scheduling"}]'
+  wait_until eval 'kubectl get sts -n "$CLUSTER_NAMESPACE" "$CLUSTER_NAME" -o json \
+    | jq .spec.nodeSelector | grep -q null'
 
   cat << EOF | kubectl create -f -
 apiVersion: stackgres.io/v1