[backend] Add enabled property to injectInput (#2043)

OpenBAS-Platform · Dec 23, 2024 · b6df3e4 · b6df3e4
1 parent 851370a
commit b6df3e4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/openbas-api/src/main/java/io/openbas/rest/inject/form/InjectInput.java b/openbas-api/src/main/java/io/openbas/rest/inject/form/InjectInput.java
@@ -56,6 +56,9 @@ public class InjectInput {
   @JsonProperty("inject_tags")
   private List<String> tagIds = new ArrayList<>();
 
+  @JsonProperty("inject_enabled")
+  private boolean enabled = true;
+
   public Inject toInject(@NotNull final InjectorContract injectorContract) {
     Inject inject = new Inject();
     inject.setTitle(getTitle());
@@ -66,6 +69,7 @@ public Inject toInject(@NotNull final InjectorContract injectorContract) {
     inject.setAllTeams(isAllTeams());
     inject.setCountry(getCountry());
     inject.setCity(getCity());
+    inject.setEnabled(isEnabled());
     return inject;
   }
 }
diff --git a/openbas-api/src/test/java/io/openbas/octi/OpenCTIApiTest.java b/openbas-api/src/test/java/io/openbas/octi/OpenCTIApiTest.java
@@ -164,7 +164,7 @@ public void testCreateScenario_ValidInput() throws Exception {
   @Test
   @DisplayName(
       "Test to validate existence of 'Add Inject to Scenario' endpoint and validate InjectInput")
-  public void testCreateInjectForScenario_ValidInput() throws Exception {
+  public void testCreateInjectForScenario_ValidInput_Additional_param() throws Exception {
     // -- PREPARE --
     String jsonInput =
         "{"
@@ -174,6 +174,7 @@ public void testCreateInjectForScenario_ValidInput() throws Exception {
             + "\"inject_content\": null, "
             + "\"inject_depends_duration\": 100, "
             + "\"inject_tags\": [\"Tag1\"], "
+            + "\"inject_enabled\": false, "
             + "\"inject_additional_param\": \"additional_param\""
             + "}";