loadgenerator follow redirects

Author: do-joe

helm/templates/loadgenerator.yaml

@@ -74,7 +74,7 @@ spec:
           RETRY_INTERVAL=10
           for i in $(seq 1 $MAX_RETRIES); do
             echo "Attempt $i: Pinging frontend: ${FRONTEND_ADDR}..."
-            STATUSCODE=$(wget --server-response ${FRONTEND_ADDR} 2>&1 | awk '/^  HTTP/{print $2}')
+            STATUSCODE=$(wget --server-response http://${FRONTEND_ADDR} 2>&1 | awk '/^  HTTP/{print $2}')
             if [ $STATUSCODE -eq 200 ]; then
                 echo "Frontend is reachable."
                 exit 0

src/loadgenerator/Dockerfile

@@ -42,5 +42,6 @@ COPY locustfile.py .
 # enable gevent support in debugger
 ENV GEVENT_SUPPORT=True
 
-ENTRYPOINT ["/bin/sh", "-c", "locust --host=\"$FRONTEND_ADDR\" --headless -u \"${USERS}\" -r \"${RATE}\""]
+# Will automatically follow redirects in case HTTP -> HTTPS redirect in place
+ENTRYPOINT ["/bin/sh", "-c", "locust --host=\"http://$FRONTEND_ADDR\" --headless -u \"${USERS}\" -r \"${RATE}\""]
 

src/loadgenerator/locustfile.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 
 import random
-from locust import FastHttpUser, TaskSet, between
+from locust import HttpUser, TaskSet, between
 from faker import Faker
 import datetime
 fake = Faker()
@@ -87,6 +87,6 @@ def on_start(self):
         viewCart: 3,
         checkout: 1}
 
-class WebsiteUser(FastHttpUser):
+class WebsiteUser(HttpUser):
     tasks = [UserBehavior]
     wait_time = between(1, 10)