Load testing

Author: scottgerring

apps/load-test-service/scripts/continuous.js

@@ -121,16 +121,18 @@ function testPassImageApi(passApiUrl, imageApiUrl) {
 
 function testPassSummaryApi(baseUrl) {
   // Either use normal or slow endpoint
-  if (Math.random() < 0.9) {
-    // 90% of the time, use normal endpoint
+  const endpointSelector = Math.random();
+  
+  if (endpointSelector < 0.8) {
+    // 80% of the time, use normal endpoint
     const summaryRes = http.get(`${baseUrl}/pass-summary`, {
       tags: { endpoint: 'get_pass_summary', mode: 'continuous' }
     });
 
     check(summaryRes, {
       'status is 200': (r) => r.status === 200
     });
-  } else {
+  } else if (endpointSelector < 0.9) {
     // 10% of the time, use slow endpoint
     const slowRes = http.get(`${baseUrl}/pass-summary/slow`, {
       tags: { endpoint: 'get_pass_summary_slow', mode: 'continuous' }
@@ -139,6 +141,15 @@ function testPassSummaryApi(baseUrl) {
     check(slowRes, {
       'slow endpoint status is 200': (r) => r.status === 200
     });
+  } else {
+    // 10% of the time, use breaky endpoint
+    const breakyRes = http.get(`${baseUrl}/pass-summary/breaky`, {
+      tags: { endpoint: 'get_pass_summary_breaky', mode: 'continuous' }
+    });
+    
+    check(breakyRes, {
+      'breaky endpoint status is 200': (r) => r.status === 200
+    });
   }
 }
 

apps/load-test-service/scripts/main.js

@@ -1,18 +1,12 @@
 #!/bin/bash
 # Load test service start script
-# Provides easy switching between different test modes
 
 # Set defaults
-SCRIPT_TYPE=${SCRIPT_TYPE:-main}
 DURATION=${DURATION:-30s}
 
 # Parse arguments
 while [[ $# -gt 0 ]]; do
   case $1 in
-    --type)
-      SCRIPT_TYPE="$2"
-      shift 2
-      ;;
     --duration)
       DURATION="$2"
       shift 2
@@ -24,28 +18,9 @@ while [[ $# -gt 0 ]]; do
   esac
 done
 
-# Map script type to actual k6 script
-case $SCRIPT_TYPE in
-  main|default)
-    K6_SCRIPT="/app/scripts/main.js"
-    echo "Running standard test sequence (main.js)"
-    ;;
-  continuous|background)
-    K6_SCRIPT="/app/scripts/continuous.js"
-    echo "Running continuous background load test (continuous.js)"
-    ;;
-  *)
-    # If a direct path is provided, use that
-    if [[ -f "$SCRIPT_TYPE" ]]; then
-      K6_SCRIPT="$SCRIPT_TYPE"
-      echo "Running custom script: $SCRIPT_TYPE"
-    else
-      echo "Invalid script type: $SCRIPT_TYPE"
-      echo "Valid options: main, continuous, or a path to a custom script"
-      exit 1
-    fi
-    ;;
-esac
+# Set the script path
+K6_SCRIPT="/app/scripts/continuous.js"
+echo "Running continuous background load test (continuous.js)"
 
 # Override the duration if specified
 if [[ "$DURATION" != "30s" ]]; then

apps/load-test-service/scripts/start.sh

@@ -0,0 +1,15 @@
+#!/bin/bash
+
+
+export DD_TRACE_AGENT_PORT=8136
+quarkus build -DskipTests
+java -javaagent:/Users/scott.gerring/dd-java-agent.jar \
+  -Ddd.profiling.enabled=true \
+  -XX:FlightRecorderOptions=stackdepth=256 \
+  -Ddd.logs.injection=true \
+  -Ddd.service=my-app \
+  -Ddd.env=staging \
+  -Ddd.version=1.0 \
+  -Ddd.trace.integration.vertx.enabled=false \
+  -Ddd.trace.executors=org.jboss.threads.EnhancedQueueExecutor \
+  -jar target/quarkus-app/quarkus-run.jar

apps/pass-summary-api/run-with-agent.sh

@@ -9,12 +9,14 @@
 import jakarta.ws.rs.Path;
 import jakarta.ws.rs.Produces;
 import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
 import org.eclipse.microprofile.rest.client.inject.RestClient;
 import org.sdlcdemo.passapi.client.Pass;
 import org.sdlcdemo.passapi.client.PassApiService;
 
 import java.util.HashMap;
 import java.util.Map;
+import java.util.Random;
 import java.util.Set;
 
 /**
@@ -68,4 +70,30 @@ public Map<String, Object> passSummarySlow() {
 
         return response;
     }
+
+    /**
+     * This endpoint periodically fails with a 500 to simulate unreliable services.
+     * When it succeeds, it returns pass summary data like the main /pass-summary endpoint.
+     */
+    @GET
+    @Path("/breaky")
+    @Produces(MediaType.APPLICATION_JSON)
+    public Response breakyEndpoint() {
+        Random r = new Random();
+        if (r.nextInt(3) == 0) { // ~33% chance of failure
+            throw new RuntimeException("Something has gone terribly amiss");
+        }
+
+        Set<Pass> allPasses = passApiService.all();
+        int totalAscent = allPasses.stream()
+                .mapToInt(Pass::ascent)
+                .sum();
+
+        Map<String, Object> response = new HashMap<>();
+        response.put("pass_count", allPasses.size());
+        response.put("total_ascent", totalAscent);
+        response.put("status", "ok");
+
+        return Response.ok(response).build();
+    }
 }

apps/pass-summary-api/src/main/java/org/sdlcdemo/PassSummaryResource.java

@@ -1,2 +1 @@
-quarkus.rest-client.pass-api-client.url=http://localhost:8081/
-
+quarkus.rest-client.pass-api-client.url=http://pass-api:8080/

apps/pass-summary-api/src/main/resources/application-dev.properties

@@ -1,2 +1 @@
-quarkus.rest-client.pass-api-client.url=http://pass-api:8080/
-
+quarkus.rest-client.pass-api-client.url=http://pass-api:8080/