diff --git a/hooks/persistence-defectdojo/hook/src/main/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategy.java b/hooks/persistence-defectdojo/hook/src/main/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategy.java index 3d7ef0debf..b11778e331 100644 --- a/hooks/persistence-defectdojo/hook/src/main/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategy.java +++ b/hooks/persistence-defectdojo/hook/src/main/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategy.java @@ -72,19 +72,14 @@ public void init(Config defectDojoConfig, PersistenceProviderConfig persistenceP @Override public List run(Scan scan, ScanFile scanResultFile) throws Exception { - Long userId = null; - if (this.config.getUserId() != null) { - LOG.debug("Using configured User Id"); - userId = this.config.getUserId(); - } else { - LOG.debug("Getting DefectDojo User Id via user profile API"); - List userProfiles = userProfileService.search(); - if (userProfiles.isEmpty()) { + LOG.debug("Getting DefectDojo User Id via user profile API"); + Long userId = null; + List userProfiles = userProfileService.search(); + if (userProfiles.isEmpty()) { throw new DefectDojoPersistenceException("UserProfileService did return empty list. Expected current user to be in list"); - } else { + } else { userId = userProfiles.get(0).getUser().getId(); - } } LOG.info("Running with DefectDojo User Id: {}", userId); diff --git a/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/mapping/DefectDojoFindingToSecureCodeBoxMapperTest.java b/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/mapping/DefectDojoFindingToSecureCodeBoxMapperTest.java index 17dece33e7..6a017acdde 100644 --- a/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/mapping/DefectDojoFindingToSecureCodeBoxMapperTest.java +++ b/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/mapping/DefectDojoFindingToSecureCodeBoxMapperTest.java @@ -18,12 +18,12 @@ import org.mockito.junit.jupiter.MockitoExtension; import java.time.Instant; -import java.time.LocalDateTime; import java.time.OffsetDateTime; import java.time.ZoneId; import java.util.List; -import static org.junit.jupiter.api.Assertions.*; +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.mockito.Mockito.when; @ExtendWith(MockitoExtension.class) @@ -37,11 +37,13 @@ class DefectDojoFindingToSecureCodeBoxMapperTest { @Mock FindingService findingService; + @Mock + Config config; + Finding exampleFinding; @BeforeEach - public void setup(){ - var config = new Config("http://example.defectdojo.com", "placeholder", "placeholder", 1000); + public void setup() { this.mapper = new DefectDojoFindingToSecureCodeBoxMapper(config, endpointService, findingService); this.exampleFinding = Finding.builder() @@ -59,7 +61,7 @@ public void setup(){ } @Test - public void shouldMapBasicFindings(){ + public void shouldMapBasicFindings() { // Typical ZAP Finding in DefectDojo var ddFinding = exampleFinding; @@ -94,7 +96,7 @@ public void shouldMapBasicFindings(){ } @Test - public void shouldIncludeOriginalDuplicateFindingInAttributes(){ + public void shouldIncludeOriginalDuplicateFindingInAttributes() { // Typical ZAP Finding in DefectDojo var ddFinding = exampleFinding; @@ -134,7 +136,7 @@ public void shouldIncludeOriginalDuplicateFindingInAttributes(){ } @Test - public void shouldNotBeStuckInARecursiveLoop(){ + public void shouldNotBeStuckInARecursiveLoop() { // Typical ZAP Finding in DefectDojo var ddFinding = exampleFinding; @@ -159,9 +161,7 @@ public void shouldNotBeStuckInARecursiveLoop(){ when(findingService.get(7L)).thenReturn(originalFinding); - var exception = Assertions.assertThrows(RuntimeException.class, () -> { - this.mapper.fromDefectDojoFinding(ddFinding); - }); + var exception = Assertions.assertThrows(RuntimeException.class, () -> this.mapper.fromDefectDojoFinding(ddFinding)); assertEquals( "Duplicate finding does not point to the actual original finding, as the original finding (id: 7) is also a duplicate. This should never happen.", diff --git a/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategyTest.java b/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategyTest.java index 9492bc1091..87ce520cb6 100644 --- a/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategyTest.java +++ b/hooks/persistence-defectdojo/hook/src/test/java/io/securecodebox/persistence/strategies/VersionedEngagementsStrategyTest.java @@ -24,9 +24,7 @@ import java.util.ArrayList; import java.util.List; -import java.util.Optional; -import static org.mockito.Mockito.any; import static org.mockito.Mockito.when; @ExtendWith(MockitoExtension.class) @@ -52,11 +50,14 @@ public class VersionedEngagementsStrategyTest { @Mock ImportScanService importScanService; + @Mock + Config config; + Scan scan; @BeforeEach - public void setup() throws Exception { - versionedEngagementsStrategy.config = new Config("https://defectdojo.example.com", "", "foobar", 1000); + public void setup() { + versionedEngagementsStrategy.config = config; versionedEngagementsStrategy.persistenceProviderConfig = new PersistenceProviderConfig(new String[]{"http://example.com","http://example.com"}); scan = new Scan(); @@ -74,7 +75,7 @@ public void setup() throws Exception { @Test @DisplayName("Fails when Configured User can not be looked up in the DefectDojo API") void requiresUserToBeFound() throws Exception { - when(userProfileService.search()).thenReturn(new ArrayList()); + when(userProfileService.search()).thenReturn(new ArrayList<>()); Assertions.assertThrows(DefectDojoPersistenceException.class, () -> { versionedEngagementsStrategy.run(scan, new ScanFile("nmap.xml",""));