Test charts

dff/stats/utils.py

@@ -109,6 +109,8 @@ def get_superset_session(args: Namespace, base_url: str = "http://localhost:8088
 
     :return: Authorized session - authorization headers tuple.
     """
+    username = args.username
+    password = args.password
     healthcheck_url = parse.urljoin(base_url, "/healthcheck")
     login_url = parse.urljoin(base_url, "/api/v1/security/login")
     csrf_url = parse.urljoin(base_url, "/api/v1/security/csrf_token/")
@@ -121,7 +123,7 @@ def get_superset_session(args: Namespace, base_url: str = "http://localhost:8088
     access_request = session.post(
         login_url,
         headers={"Content-Type": "application/json", "Accept": "*/*"},
-        data=json.dumps({"username": args.username, "password": args.password, "refresh": True, "provider": "db"}),
+        data=json.dumps({"username": username, "password": password, "refresh": True, "provider": "db"}),
     )
     access_token = access_request.json()["access_token"]
     # get csrf_token

dff/utils/docker/dockerfile_stats

@@ -1,6 +1,7 @@
 FROM apache/superset:2.1.0rc1
 USER root
 RUN cd /app && pip install .[clickhouse]
+COPY health_stats.sh /app/docker/
 COPY entrypoint_stats.sh /app/docker/
 COPY --chown=superset superset_config_docker.py /app/pythonpath/
 ENV SUPERSET_CONFIG_PATH /app/pythonpath/superset_config_docker.py

dff/utils/docker/entrypoint_stats.sh

@@ -2,7 +2,7 @@
 export SERVER_THREADS_AMOUNT=8
 set -m
 nohup /bin/bash /usr/bin/run-server.sh &
-sleep 5
+/bin/bash /app/docker/health_stats.sh
 superset fab create-admin --firstname superset --lastname admin --username $SUPERSET_USERNAME --email [email protected] --password $SUPERSET_PASSWORD
 superset db upgrade
 superset init

dff/utils/docker/health_stats.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+for itr in {1..10}
+do
+healthcheck=$(curl -X GET http://localhost:8088/health | grep "OK")
+healthcheck=$?
+if [ "$healthcheck" -ne 0 ] ; then
+echo "Healthcheck failed. sleeping for 5 secs"
+sleep 5
+echo 'Iteration' $itr
+if [ $itr == 10 ]; then
+echo 'Healthcheck suite unsuccessful.'
+fi
+else
+echo "Healthcheck suite succesful."
+break
+exit 0
+fi
+done