X-Git-Url: https://gerrit.akraino.org/r/gitweb?p=ta%2Fcloudtaf.git;a=blobdiff_plain;f=testcases%2FHPA_check%2FHPA_check.py;fp=testcases%2FHPA_check%2FHPA_check.py;h=d6ab1e3276d85d7402ef90186138b5314d968c6e;hp=0000000000000000000000000000000000000000;hb=af5eb3ff36b92ab1d9c156ffa0391eadc73eb6ba;hpb=025a45508d009db84c34076fb4a668f712628d6d

diff --git a/testcases/HPA_check/HPA_check.py b/testcases/HPA_check/HPA_check.py
new file mode 100644
index 0000000..d6ab1e3
--- /dev/null
+++ b/testcases/HPA_check/HPA_check.py
@@ -0,0 +1,70 @@
+import sys
+import os
+import time
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '../libraries/common'))
+from datetime import datetime  # noqa
+from datetime import timedelta  # noqa
+from robot.libraries.BuiltIn import BuiltIn  # noqa
+from robot.api import logger  # noqa
+import common_utils  # noqa
+from decorators_for_robot_functionalities import *  # noqa
+from test_constants import *  # noqa
+
+
+execute = BuiltIn().get_library_instance('execute_command')
+stack_infos = BuiltIn().get_library_instance('stack_infos')
+
+
+def HPA_check():
+    steps = ['step1_check_initial_replica_count',
+             'step2_check_scale_out',
+             'step3_check_scale_in']
+    BuiltIn().run_keyword("HPA_check.setup")
+    common_utils.keyword_runner(steps)
+
+
+def setup():
+    common_utils.helm_install(chart_name="default/php-apache", release_name="crf01",
+                              values="registry_url={reg_url}".format(reg_url=reg))
+    common_utils.check_kubernetes_object(kube_object=php_apache_pod,
+                                         tester_function=common_utils.test_kubernetes_object_available,
+                                         additional_filter="Running",
+                                         timeout=90)
+    flags = ["--horizontal-pod-autoscaler-downscale-stabilization=10s", "--horizontal-pod-autoscaler-sync-period=10s"]
+    common_utils.modify_static_pod_config(common_utils.add_flag_to_command, "cm.yml", flags)
+    common_utils.helm_install(chart_name="default/load-generator-for-apache", release_name="load")
+    common_utils.check_kubernetes_object(kube_object=load_generator_for_apache,
+                                         tester_function=common_utils.test_kubernetes_object_available,
+                                         additional_filter="Running",
+                                         timeout=60)
+
+
+def step1_check_initial_replica_count():
+    time.sleep(5)
+    replica_count = int(
+        execute.execute_unix_command("kubectl get hpa | grep php-apache-hpa | awk '{print $6}'"))
+    if replica_count == 1:
+        logger.info("number of php apache pod is 1")
+    else:
+        raise Exception("Expected initial replica count is not correct: expected: 1, got: " + str(replica_count))
+
+
+def step2_check_scale_out():
+    check_scaling(expected_replicas="2", timeout=360)
+
+
+def step3_check_scale_in():
+    check_scaling(expected_replicas="1", timeout=480)
+
+
+def check_scaling(expected_replicas, timeout):
+    wait_until = datetime.now() + timedelta(seconds=timeout)
+    actual_replicas = execute.execute_unix_command("kubectl get hpa | grep php-apache-hpa | awk '{print $6}'")
+    while actual_replicas != expected_replicas:
+        time.sleep(5)
+        actual_replicas = execute.execute_unix_command("kubectl get hpa | grep php-apache-hpa | awk '{print $6}'")
+        if actual_replicas == expected_replicas:
+            logger.info("number of php apache pod is " + expected_replicas + ", scale out was successful")
+        elif wait_until < datetime.now():
+            raise Exception("Scaling did not happen in " + str(timeout) + " seconds, expected replica count is " +
+                            expected_replicas + ", got " + actual_replicas)