From e40abdb249dee99da2e9b76963ecbd29b952c608 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20Mu=C3=9Fler?= Date: Wed, 21 Oct 2020 18:12:33 +0200 Subject: [PATCH] Move scale function. --- e2e/tests/k8s_api.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/e2e/tests/k8s_api.py b/e2e/tests/k8s_api.py index ac38b6438..2db1c86b3 100644 --- a/e2e/tests/k8s_api.py +++ b/e2e/tests/k8s_api.py @@ -107,19 +107,14 @@ class K8s: return False return True - def wait_for_pg_to_scale(self, number_of_instances, namespace='default'): - + def scale_cluster(self, number_of_instances, name="acid-minimal-cluster", namespace="default"): body = { "spec": { "numberOfInstances": number_of_instances } } - _ = self.api.custom_objects_api.patch_namespaced_custom_object( - "acid.zalan.do", "v1", namespace, "postgresqls", "acid-minimal-cluster", body) - - labels = 'application=spilo,cluster-name=acid-minimal-cluster' - while self.count_pods_with_label(labels) != number_of_instances: - time.sleep(self.RETRY_TIMEOUT_SEC) + self.api.custom_objects_api.patch_namespaced_custom_object( + "acid.zalan.do", "v1", namespace, "postgresqls", name, body) def wait_for_running_pods(self, labels, number, namespace=''): while self.count_pods_with_label(labels) != number: