From 45371608b161a823ab7c65bc6c2a9a0b1cb61318 Mon Sep 17 00:00:00 2001
From: Bartosz Prusinowski <bartosz@interactivethings.com>
Date: Thu, 14 Dec 2023 09:44:44 +0100
Subject: [PATCH] feat: Set up initial GraphQL performance tests

---
 .github/workflows/performance-tests.yml       | 17 +++++++++
 .../generate-github-action.js                 | 37 +++++++++++++++++++
 k6/performance-tests/graphql/metadata.js      | 19 +++-------
 package.json                                  |  3 +-
 4 files changed, 62 insertions(+), 14 deletions(-)
 create mode 100644 .github/workflows/performance-tests.yml
 create mode 100644 k6/performance-tests/generate-github-action.js

diff --git a/.github/workflows/performance-tests.yml b/.github/workflows/performance-tests.yml
new file mode 100644
index 000000000..81ae748ad
--- /dev/null
+++ b/.github/workflows/performance-tests.yml
@@ -0,0 +1,17 @@
+name: Performance tests
+
+on: workflow_dispatch
+
+jobs:
+  compile:
+    name: Compile site assets
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+      - name: Run the build process with Docker
+        uses: addnab/docker-run-action@v3
+        with:
+          image: grafana/k6:latest
+          options: -rm -e K6_PROMETHEUS_RW_USERNAME=${{ secrets.K6_PROMETHEUS_RW_USERNAME }} -e K6_PROMETHEUS_RW_PASSWORD=${{ secrets.K6_PROMETHEUS_RW_PASSWORD }} -e K6_PROMETHEUS_RW_SERVER_URL=${{ secrets.K6_PROMETHEUS_RW_SERVER_URL }}
+          run: k6 run -o experimental-prometheus-rw --tag testid=StateAccounts_Office/4/ --env ENV=test --env CUBE=StateAccounts_Office/4/ - <k6/performance-tests/graphql/metadata.js && k6 run -o experimental-prometheus-rw --tag testid=StateAccounts_Office/4/ --env ENV=int --env CUBE=StateAccounts_Office/4/ - <k6/performance-tests/graphql/metadata.js && k6 run -o experimental-prometheus-rw --tag testid=StateAccounts_Office/4/ --env ENV=prod --env CUBE=StateAccounts_Office/4/ - <k6/performance-tests/graphql/metadata.js
\ No newline at end of file
diff --git a/k6/performance-tests/generate-github-action.js b/k6/performance-tests/generate-github-action.js
new file mode 100644
index 000000000..5095b4802
--- /dev/null
+++ b/k6/performance-tests/generate-github-action.js
@@ -0,0 +1,37 @@
+const fs = require("fs");
+
+const envs = ["test", "int", "prod"];
+const queries = ["metadata"];
+const cubes = ["StateAccounts_Office/4/"];
+
+const generate = () => {
+  const file = `name: Performance tests
+
+on: workflow_dispatch
+
+jobs:
+  compile:
+    name: Compile site assets
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out the repo
+        uses: actions/checkout@v2
+      - name: Run the build process with Docker
+        uses: addnab/docker-run-action@v3
+        with:
+          image: grafana/k6:latest
+          options: -rm -e K6_PROMETHEUS_RW_USERNAME=\${{ secrets.K6_PROMETHEUS_RW_USERNAME }} -e K6_PROMETHEUS_RW_PASSWORD=\${{ secrets.K6_PROMETHEUS_RW_PASSWORD }} -e K6_PROMETHEUS_RW_SERVER_URL=\${{ secrets.K6_PROMETHEUS_RW_SERVER_URL }}
+          run: ${envs
+            .map((env) => {
+              return queries.map((query) => {
+                return cubes.map((cube) => {
+                  return `k6 run -o experimental-prometheus-rw --tag testid=${cube} --env ENV=${env} --env CUBE=${cube} - <k6/performance-tests/graphql/${query}.js`;
+                });
+              });
+            })
+            .join(" && ")}`;
+
+  fs.writeFileSync("./.github/workflows/performance-tests.yml", file);
+};
+
+generate();
diff --git a/k6/performance-tests/graphql/metadata.js b/k6/performance-tests/graphql/metadata.js
index 9b4f90e74..31a15c86c 100644
--- a/k6/performance-tests/graphql/metadata.js
+++ b/k6/performance-tests/graphql/metadata.js
@@ -24,32 +24,25 @@ const variables = {
   },
 };
 
-const env = __ENV.ENV || "prod";
-const enableCache = __ENV.ENABLE_GQL_SERVER_SIDE_CACHE === "true";
+const env = __ENV.ENV;
+const cube = __ENV.CUBE;
 
 /** @type {import("k6/options").Options} */
 export const options = {
   iterations: 1,
-  ext: {
-    loadimpact: {
-      name: `GraphQL - DataCubeMetadata (${env.toUpperCase()}, GQL ${
-        enableCache ? "cache" : "no-cache"
-      })`,
-    },
-  },
 };
 
 export default function Components() {
-  // Set tags for metrics
-  exec.vu.metrics.tags.env = "int";
-  exec.vu.metrics.tags.cube = "StateAccounts_Office/4/";
+  exec.vu.metrics.tags.env = env;
+  exec.vu.metrics.tags.cube = cube;
 
   http.post(
-    "https://int.visualize.admin.ch/api/graphql",
+    `https://${env === "prod" ? "" : `${env}.`}visualize.admin.ch/api/graphql`,
     JSON.stringify({ query, variables }),
     {
       headers: {
         "Content-Type": "application/json",
+        "x-visualize-cache-control": "no-cache",
       },
     }
   );
diff --git a/package.json b/package.json
index 55b6a1d79..933a8efea 100644
--- a/package.json
+++ b/package.json
@@ -31,7 +31,8 @@
     "e2e:dev": "E2E_BASE_URL=http://localhost:3000 yarn playwright test",
     "e2e": "percy exec -- playwright test",
     "cube": "NODE_ENV=development ts-node app/scripts/cube.ts",
-    "dev:circular-deps": "madge --warning --extensions js,jsx,ts,tsx -b ./app -c ./app/pages/ --ts-config ./app/tsconfig.json | bun scripts/circular-deps-analysis.ts -"
+    "dev:circular-deps": "madge --warning --extensions js,jsx,ts,tsx -b ./app -c ./app/pages/ --ts-config ./app/tsconfig.json | bun scripts/circular-deps-analysis.ts -",
+    "github:codegen": "node k6/performance-tests/generate-github-action.js"
   },
   "dependencies": {
     "@babel/runtime": "^7.11.2",