Add k6 module (#8318)

Co-authored-by: Eddú Meléndez <[email protected]>

Author: javaducky

.github/ISSUE_TEMPLATE/bug_report.yaml

@@ -30,6 +30,7 @@ body:
         - HiveMQ
         - InfluxDB
         - K3S
+        - K6
         - Kafka
         - LocalStack
         - MariaDB

.github/ISSUE_TEMPLATE/enhancement.yaml

@@ -30,6 +30,7 @@ body:
         - HiveMQ
         - InfluxDB
         - K3S
+        - K6
         - Kafka
         - LocalStack
         - MariaDB

.github/ISSUE_TEMPLATE/feature.yaml

@@ -30,6 +30,7 @@ body:
         - HiveMQ
         - InfluxDB
         - K3S
+        - K6
         - Kafka
         - LocalStack
         - MariaDB

.github/dependabot.yml

@@ -148,6 +148,14 @@ updates:
     ignore:
       - dependency-name: "com.fasterxml.jackson.dataformat:jackson-dataformat-yaml"
         update-types: [ "version-update:semver-minor", "version-update:semver-patch" ]
+  - package-ecosystem: "gradle"
+    directory: "/modules/k6"
+    schedule:
+      interval: "weekly"
+    open-pull-requests-limit: 10
+    ignore:
+      - dependency-name: "com.fasterxml.jackson.dataformat:jackson-dataformat-yaml"
+        update-types: [ "version-update:semver-minor", "version-update:semver-patch" ]
   - package-ecosystem: "gradle"
     directory: "/modules/kafka"
     schedule:

.github/labeler.yml

@@ -87,6 +87,10 @@
   - changed-files:
     - any-glob-to-any-file:
       - modules/k3s/**/*
+"modules/k6":
+  - changed-files:
+    - any-glob-to-any-file:
+      - modules/k6/**/*
 "modules/kafka":
   - changed-files:
     - any-glob-to-any-file:

.github/settings.yml

@@ -160,6 +160,9 @@ labels:
   - name: modules/k3s
     color: '#006b75'
 
+  - name: modules/k6
+    color: '#006b75'
+
   - name: modules/kafka
     color: '#006b75'
 

docs/modules/k6.md

@@ -0,0 +1,39 @@
+# k6 Module
+
+!!! note 
+    This module is INCUBATING. 
+    While it is ready for use and operational in the current version of Testcontainers, it is possible that it may receive breaking changes in the future. 
+    See [our contributing guidelines](/contributing/#incubating-modules) for more information on our incubating modules policy.
+
+Testcontainers module for [k6](https://registry.hub.docker.com/r/grafana/k6).
+
+[k6](https://k6.io/) is an extensible reliability testing tool built for developer happiness.
+
+## Basic script execution
+
+Execute a simple k6 test script, `test.js`, with commandline options and injected script variable.
+
+Create a simple k6 test script to be executed as part of your tests:
+
+<!--codeinclude-->
+[Setup the container](../../modules/k6/src/test/java/org/testcontainers/k6/K6ContainerTests.java) inside_block:standard_k6
+[Content of `scripts/test.js`](../../modules/k6/src/test/resources/scripts/test.js) inside_block:access_script_vars
+<!--/codeinclude-->
+
+## Adding this module to your project dependencies
+
+Add the following dependency to your `pom.xml`/`build.gradle` file:
+
+=== "Gradle"
+    ```groovy
+    testImplementation "org.testcontainers:k6:{{latest_version}}"
+    ```
+=== "Maven"
+    ```xml
+    <dependency>
+        <groupId>org.testcontainers</groupId>
+        <artifactId>k6</artifactId>
+        <version>{{latest_version}}</version>
+        <scope>test</scope>
+    </dependency>
+    ```

mkdocs.yml

@@ -82,6 +82,7 @@ nav:
           - modules/gcloud.md
           - modules/hivemq.md
           - modules/k3s.md
+          - modules/k6.md
           - modules/kafka.md
           - modules/localstack.md
           - modules/milvus.md

modules/k6/build.gradle

@@ -0,0 +1,7 @@
+description = "Testcontainers :: k6"
+
+dependencies {
+    api project(':testcontainers')
+
+    testImplementation 'org.assertj:assertj-core:3.25.2'
+}

modules/k6/src/main/java/org/testcontainers/k6/K6Container.java

@@ -0,0 +1,88 @@
+package org.testcontainers.k6;
+
+import org.apache.commons.io.FilenameUtils;
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.utility.DockerImageName;
+import org.testcontainers.utility.MountableFile;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class K6Container extends GenericContainer<K6Container> {
+
+    /** Standard image for k6, as provided by Grafana. */
+    private static final DockerImageName K6_IMAGE = DockerImageName.parse("grafana/k6");
+
+    private String testScript;
+
+    private List<String> cmdOptions = new ArrayList<>();
+
+    private Map<String, String> scriptVars = new HashMap<>();
+
+    /**
+     * Creates a new container instance based upon the provided image name.
+     */
+    public K6Container(String dockerImageName) {
+        this(DockerImageName.parse(dockerImageName));
+    }
+
+    /**
+     * Creates a new container instance based upon the provided image.
+     */
+    public K6Container(DockerImageName dockerImageName) {
+        super(dockerImageName);
+        dockerImageName.assertCompatibleWith(K6_IMAGE);
+    }
+
+    /**
+     * Specifies the test script to be executed within the container.
+     * @param testScript file to be copied into the container
+     * @return the builder
+     */
+    public K6Container withTestScript(MountableFile testScript) {
+        this.testScript = "/home/k6/" + FilenameUtils.getName(testScript.getResolvedPath());
+        withCopyFileToContainer(testScript, this.testScript);
+        return self();
+    }
+
+    /**
+     * Specifies additional command line options to be provided to the k6 command.
+     * @param options command line options
+     * @return the builder
+     */
+    public K6Container withCmdOptions(String... options) {
+        cmdOptions.addAll(Arrays.asList(options));
+        return self();
+    }
+
+    /**
+     * Adds a key-value pair for access within test scripts as an environment variable.
+     * @param key   unique identifier for the variable
+     * @param value value of the variable
+     * @return the builder
+     */
+    public K6Container withScriptVar(String key, String value) {
+        scriptVars.put(key, value);
+        return self();
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    @Override
+    protected void configure() {
+        List<String> commandParts = new ArrayList<>();
+        commandParts.add("run");
+        commandParts.addAll(cmdOptions);
+        for (Map.Entry<String, String> entry : scriptVars.entrySet()) {
+            commandParts.add("--env");
+            commandParts.add(String.format("%s=%s", entry.getKey(), entry.getValue()));
+        }
+        commandParts.add(testScript);
+
+        setCommand(commandParts.toArray(new String[] {}));
+    }
+}