-
Notifications
You must be signed in to change notification settings - Fork 89
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'master' into bugfix/template-tech-impr
- Loading branch information
Showing
32 changed files
with
1,697 additions
and
1,551 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
150 changes: 150 additions & 0 deletions
150
...as-api/src/test/java/io/openbas/executors/caldera/service/CalderaExecutorServiceTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,150 @@ | ||
| package io.openbas.executors.caldera.service; | ||
|
|
||
| import static org.junit.jupiter.api.Assertions.assertEquals; | ||
| import static org.junit.jupiter.api.Assertions.assertTrue; | ||
| import static org.mockito.Mockito.verify; | ||
| import static org.mockito.Mockito.when; | ||
|
|
||
| import io.openbas.asset.EndpointService; | ||
| import io.openbas.database.model.Endpoint; | ||
| import io.openbas.database.model.Executor; | ||
| import io.openbas.executors.caldera.client.CalderaExecutorClient; | ||
| import io.openbas.executors.caldera.config.CalderaExecutorConfig; | ||
| import io.openbas.executors.caldera.model.Agent; | ||
| import io.openbas.integrations.ExecutorService; | ||
| import io.openbas.integrations.InjectorService; | ||
| import io.openbas.service.PlatformSettingsService; | ||
| import java.util.List; | ||
| import java.util.Optional; | ||
| import org.junit.jupiter.api.BeforeEach; | ||
| import org.junit.jupiter.api.Test; | ||
| import org.junit.jupiter.api.extension.ExtendWith; | ||
| import org.mockito.InjectMocks; | ||
| import org.mockito.Mock; | ||
| import org.mockito.junit.jupiter.MockitoExtension; | ||
|
|
||
| @ExtendWith(MockitoExtension.class) | ||
| public class CalderaExecutorServiceTest { | ||
| private static final String CALDERA_AGENT_HOSTNAME = "calderaHostname"; | ||
| private static final String CALDERA_AGENT_EXTERNAL_REF = "calderaExt"; | ||
| private static final String CALDERA_AGENT_IP = "10.10.10.10"; | ||
|
|
||
| private static final String CALDERA_EXECUTOR_TYPE = "openbas_caldera"; | ||
| private static final String CALDERA_EXECUTOR_NAME = "Caldera"; | ||
|
|
||
| private static final String DATE = "2024-12-11T12:45:30Z"; | ||
|
|
||
| @Mock private ExecutorService executorService; | ||
|
|
||
| @Mock private CalderaExecutorClient client; | ||
|
|
||
| @Mock private CalderaExecutorConfig config; | ||
|
|
||
| @Mock private CalderaExecutorContextService calderaExecutorContextService; | ||
|
|
||
| @Mock private EndpointService endpointService; | ||
|
|
||
| @Mock private InjectorService injectorService; | ||
|
|
||
| @Mock private PlatformSettingsService platformSettingsService; | ||
|
|
||
| @Mock private Executor executor; | ||
|
|
||
| @InjectMocks private CalderaExecutorService calderaExecutorService; | ||
|
|
||
| private Endpoint calderaEndpoint; | ||
| private Endpoint randomEndpoint; | ||
| private Agent calderaAgent; | ||
| private Agent randomAgent; | ||
| private Executor calderaExecutor; | ||
| private Executor randomExecutor; | ||
|
|
||
| @BeforeEach | ||
| void setUp() { | ||
| calderaAgent = new Agent(); | ||
| calderaAgent.setArchitecture("Arch"); | ||
| calderaAgent.setPaw(CALDERA_AGENT_EXTERNAL_REF); | ||
| calderaExecutor = new Executor(); | ||
| calderaExecutor.setName(CALDERA_EXECUTOR_NAME); | ||
| calderaExecutor.setType(CALDERA_EXECUTOR_TYPE); | ||
| randomExecutor = new Executor(); | ||
| randomExecutor.setName("NAME"); | ||
| randomExecutor.setType("TYPE"); | ||
| calderaExecutorService.setExecutor(calderaExecutor); | ||
|
|
||
| calderaAgent = | ||
| createAgent(CALDERA_AGENT_HOSTNAME, CALDERA_AGENT_IP, CALDERA_AGENT_EXTERNAL_REF); | ||
| randomAgent = createAgent("hostname", "1.1.1.1", "ref"); | ||
| calderaEndpoint = createEndpoint(calderaAgent, calderaExecutor); | ||
| randomEndpoint = createEndpoint(randomAgent, randomExecutor); | ||
|
|
||
| when(endpointService.findAssetsForInjectionByHostname(CALDERA_AGENT_HOSTNAME)) | ||
| .thenReturn(List.of(calderaEndpoint, randomEndpoint)); | ||
| } | ||
|
|
||
| private Endpoint createEndpoint(Agent agent, Executor executor) { | ||
| Endpoint endpoint = new Endpoint(); | ||
| endpoint.setExecutor(executor); | ||
| endpoint.setExternalReference(agent.getPaw()); | ||
| endpoint.setName(agent.getHost()); | ||
| endpoint.setDescription("Asset collected by Caldera executor context."); | ||
| endpoint.setIps(agent.getHost_ip_addrs()); | ||
| endpoint.setHostname(agent.getHost()); | ||
| endpoint.setPlatform(CalderaExecutorService.toPlatform("windows")); | ||
| endpoint.setArch(CalderaExecutorService.toArch("amd64")); | ||
| endpoint.setProcessName(agent.getExe_name()); | ||
| endpoint.setLastSeen(calderaExecutorService.toInstant(DATE)); | ||
| return endpoint; | ||
| } | ||
|
|
||
| private Agent createAgent(String hostname, String ip, String externalRef) { | ||
| Agent agent = new Agent(); | ||
| agent.setArchitecture("amd64"); | ||
| agent.setPaw(externalRef); | ||
| agent.setPlatform("windows"); | ||
| agent.setExe_name("exe"); | ||
| agent.setLast_seen(DATE); | ||
| agent.setHost_ip_addrs(new String[] {ip}); | ||
| agent.setHost(hostname); | ||
| return agent; | ||
| } | ||
|
|
||
| @Test | ||
| void test_run_WITH_one_endpoint() throws Exception { | ||
| when(client.agents()).thenReturn(List.of(calderaAgent)); | ||
| calderaExecutorService.run(); | ||
| verify(endpointService).updateEndpoint(calderaEndpoint); | ||
| } | ||
|
|
||
| @Test | ||
| void test_run_WITH_2_existing_endpoint_same_machine() throws Exception { | ||
| when(client.agents()).thenReturn(List.of(calderaAgent)); | ||
| randomEndpoint.setHostname(CALDERA_AGENT_HOSTNAME); | ||
| randomEndpoint.setIps(new String[] {CALDERA_AGENT_IP}); | ||
| calderaExecutorService.run(); | ||
| verify(endpointService).updateEndpoint(calderaEndpoint); | ||
| } | ||
|
|
||
| @Test | ||
| void test_findExistingEndpointForAnAgent_WITH_2_existing_endpoint_same_host() throws Exception { | ||
| Optional<Endpoint> result = calderaExecutorService.findExistingEndpointForAnAgent(calderaAgent); | ||
| assertEquals(calderaEndpoint, result.get()); | ||
| } | ||
|
|
||
| @Test | ||
| void test_findExistingEndpointForAnAgent_WITH_no_existing_endpoint() throws Exception { | ||
| when(endpointService.findAssetsForInjectionByHostname(CALDERA_AGENT_HOSTNAME)) | ||
| .thenReturn(List.of()); | ||
| Optional<Endpoint> result = calderaExecutorService.findExistingEndpointForAnAgent(calderaAgent); | ||
| assertTrue(result.isEmpty()); | ||
| } | ||
|
|
||
| @Test | ||
| void test_findExistingEndpointForAnAgent_WITH_1_enpdoint_with_null_executor() throws Exception { | ||
| randomEndpoint.setExecutor(null); | ||
| randomEndpoint.setHostname(CALDERA_AGENT_HOSTNAME); | ||
| randomEndpoint.setIps(new String[] {CALDERA_AGENT_IP}); | ||
| Optional<Endpoint> result = calderaExecutorService.findExistingEndpointForAnAgent(calderaAgent); | ||
| assertEquals(calderaEndpoint, result.get()); | ||
| } | ||
| } |
Oops, something went wrong.