feat: e2e test for cloud.stream being set

Author: yorugac

e2e/cloud-stream/manifests/configmap.yaml

@@ -0,0 +1,57 @@
+apiVersion: v1
+kind: ConfigMap
+metadata:
+  name: test
+  namespace: k6-tests
+data:
+  test.js: |
+    import http from "k6/http";
+    import { check, sleep } from "k6";
+    import { Trend, Counter } from "k6/metrics";
+
+    const BASE_URL = __ENV.BASE_URL || 'http://localhost:3333';
+
+    export const options = {
+      stages: [
+        { duration: '5s', target: 5 },
+        { duration: '10s', target: 5 },
+        { duration: '5s', target: 0 },
+      ],
+      cloud: {
+        projectID: $PROJECT_ID,
+        name: 'k6-operator-e2e-cloud-stream'
+      }
+    };
+
+    const pizzas = new Counter('quickpizza_number_of_pizzas');
+    const ingredients = new Trend('quickpizza_ingredients');
+
+    export function setup() {
+      console.log("Starting folks!")
+    }
+
+    export default function () {
+      let restrictions = {
+        maxCaloriesPerSlice: 500,
+        mustBeVegetarian: false,
+        excludedIngredients: ["pepperoni"],
+        excludedTools: ["knife"],
+        maxNumberOfToppings: 6,
+        minNumberOfToppings: 2
+      }
+      let res = http.post(`/api/pizza`, JSON.stringify(restrictions), {
+        headers: {
+          'Content-Type': 'application/json',
+          'X-User-ID': 23423,
+        },
+      });
+      check(res, { "status is 200": (res) => res.status === 200 });
+      console.log(`${res.json().pizza.name} (${res.json().pizza.ingredients.length} ingredients)`);
+      pizzas.add(1);
+      ingredients.add(res.json().pizza.ingredients.length);
+      sleep(1);
+    }
+
+    export function teardown(){
+      console.log("That's all folks!")
+    }

e2e/cloud-stream/manifests/kustomization.yaml

@@ -0,0 +1,7 @@
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+- ../../latest/
+- configmap.yaml
+- namespace.yaml
+- secret.yaml

e2e/cloud-stream/manifests/namespace.yaml

@@ -0,0 +1,4 @@
+apiVersion: v1
+kind: Namespace
+metadata:
+  name: k6-tests

e2e/cloud-stream/manifests/secret.yaml

@@ -0,0 +1,9 @@
+apiVersion: v1
+kind: Secret
+metadata:
+  name: cloud-token
+  namespace: k6-operator-system
+  labels:
+    k6cloud: token
+data:
+  token: $CLOUD_TOKEN

e2e/cloud-stream/test.js

@@ -0,0 +1,76 @@
+import { Environment } from 'k6/x/environment';
+import { sleep, fail } from 'k6';
+
+export const options = {
+  setupTimeout: '60s',
+};
+
+const PARENT = "./"
+
+const env = new Environment({
+  name: "cloud-stream",
+  implementation: "vcluster",
+  initFolder: PARENT + "manifests",
+})
+
+export function setup() {
+  console.log("init returns", env.init());
+  sleep(0.5);
+}
+
+export default function () {
+  let err = env.apply(PARENT + "testrun.yaml");
+  console.log("apply testrun returns", err);
+
+  err = env.wait({
+    kind: "TestRun",
+    name: "k6-sample",
+    namespace: "k6-tests",
+    status_key: "stage",
+    status_value: "started",
+  }, {
+    timeout: "1m",
+    interval: "10s",
+  });
+
+  if (err != null) {
+    fail("wait for started returns " + err);
+  }
+
+  let allPods = env.getN("pods", {
+    "namespace": "k6-tests",
+    "app": "k6",
+    "k6_cr": "k6-sample",
+  });
+
+  let runnerPods = env.getN("pods", {
+    "namespace": "k6-tests",
+    "app": "k6",
+    "k6_cr": "k6-sample",
+    "runner": "true",
+  });
+
+  // there should be N runner pods + initializer + starter
+  if (runnerPods != 1 || allPods != runnerPods + 2) {
+    fail("wrong number of pods:" + runnerPods + "/" + allPods);
+  }
+
+  err = env.wait({
+    kind: "TestRun",
+    name: "k6-sample",
+    namespace: "k6-tests",
+    status_key: "stage",
+    status_value: "finished",
+  }, {
+    timeout: "5m",
+    interval: "10s",
+  });
+
+  if (err != null) {
+    fail("wait for finished returns " + err);
+  }
+}
+
+export function teardown() {
+  console.log("delete returns", env.delete());
+}

e2e/cloud-stream/testrun.yaml

@@ -0,0 +1,18 @@
+---
+apiVersion: k6.io/v1alpha1
+kind: TestRun
+metadata:
+  name: k6-sample
+  namespace: k6-tests
+spec:
+  parallelism: 1
+  script:
+    configMap:
+      name: "test"
+      file: "test.js"
+  cloud:
+    stream: true
+  runner:
+    env:
+      - name: BASE_URL
+        value: https://pizza.grafana.fun

e2e/tests.yaml

@@ -99,6 +99,13 @@ tests:
       - "pkg/cloud/"
     requires_env: [CLOUD_TOKEN, PROJECT_ID]
 
+  - name: cloud-stream
+    source_patterns:
+      - "pkg/cloud/"
+      - "pkg/resources/jobs/runner.go"
+      - "internal/controller/k6_initialize.go"
+    requires_env: [CLOUD_TOKEN, PROJECT_ID]
+
   - name: initializer-disabled
     source_patterns:
       - "internal/controller/k6_initialize.go"