Move scale function.

This commit is contained in:
Jan Mußler 2020-10-21 18:12:33 +02:00
parent b422cf9d71
commit e40abdb249
1 changed files with 3 additions and 8 deletions

View File

@ -107,19 +107,14 @@ class K8s:
return False return False
return True 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 = { body = {
"spec": { "spec": {
"numberOfInstances": number_of_instances "numberOfInstances": number_of_instances
} }
} }
_ = self.api.custom_objects_api.patch_namespaced_custom_object( self.api.custom_objects_api.patch_namespaced_custom_object(
"acid.zalan.do", "v1", namespace, "postgresqls", "acid-minimal-cluster", body) "acid.zalan.do", "v1", namespace, "postgresqls", name, 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)
def wait_for_running_pods(self, labels, number, namespace=''): def wait_for_running_pods(self, labels, number, namespace=''):
while self.count_pods_with_label(labels) != number: while self.count_pods_with_label(labels) != number: