-
Notifications
You must be signed in to change notification settings - Fork 88
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'refs/heads/master' into release/1.10.0
- Loading branch information
Showing
5 changed files
with
218 additions
and
62 deletions.
There are no files selected for viewing
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
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()); | ||
| } | ||
| } |
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