From e3011f8336c257dc9295eb1ddb9d6ba77ddb8d75 Mon Sep 17 00:00:00 2001
From: Lokesh Pant <7022317+KraTos5589@users.noreply.github.com>
Date: Tue, 26 Sep 2023 08:33:15 +0000
Subject: [PATCH] Adding Support for Depending on sourcesets other than main
 and test

---
 src/main/groovy/io/gatling/gradle/GatlingPlugin.groovy         | 3 +++
 .../groovy/io/gatling/gradle/GatlingPluginExtension.groovy     | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/src/main/groovy/io/gatling/gradle/GatlingPlugin.groovy b/src/main/groovy/io/gatling/gradle/GatlingPlugin.groovy
index e718a1d..5ae4d48 100644
--- a/src/main/groovy/io/gatling/gradle/GatlingPlugin.groovy
+++ b/src/main/groovy/io/gatling/gradle/GatlingPlugin.groovy
@@ -218,6 +218,9 @@ final class GatlingPlugin implements Plugin<Project> {
                 if (evaluatedExt.includeTestOutput) {
                     gatlingImplementation evaluatedProject.sourceSets.test.output
                 }
+                evaluatedExt.includeSourceSetsOutput.forEach { sourceSet ->
+                    gatlingImplementation sourceSets.named(sourceSet).get().output
+                }
             }
         }
     }
diff --git a/src/main/groovy/io/gatling/gradle/GatlingPluginExtension.groovy b/src/main/groovy/io/gatling/gradle/GatlingPluginExtension.groovy
index a6f8429..4519959 100644
--- a/src/main/groovy/io/gatling/gradle/GatlingPluginExtension.groovy
+++ b/src/main/groovy/io/gatling/gradle/GatlingPluginExtension.groovy
@@ -332,6 +332,8 @@ class GatlingPluginExtension {
     Boolean includeMainOutput = true
     Boolean includeTestOutput = true
 
+    Set<String> includeSourceSetsOutput = new HashSet<>();
+
     String logLevel
 
     LogHttp logHttp