11package io .openaev .rest .inject_expectation ;
22
33import static io .openaev .expectation .ExpectationPropertiesConfig .DEFAULT_TECHNICAL_EXPECTATION_EXPIRATION_TIME ;
4- import static io .openaev .integration .impl .injectors .openaev .OpenaevInjectorIntegration .OPENAEV_INJECTOR_ID ;
5- import static io .openaev .integration .impl .injectors .openaev .OpenaevInjectorIntegration .OPENAEV_INJECTOR_NAME ;
64import static io .openaev .utils .fixtures .ExpectationFixture .createDetectionExpectations ;
75import static io .openaev .utils .fixtures .ExpectationFixture .createPreventionExpectations ;
86import static java .util .Collections .emptyList ;
1715import io .openaev .service .InjectExpectationService ;
1816import io .openaev .utils .fixtures .*;
1917import io .openaev .utils .fixtures .composers .*;
20- import jakarta .persistence .EntityManager ;
2118import java .util .List ;
2219import java .util .Map ;
2320import org .junit .jupiter .api .*;
3027class InjectExpectationServiceTest extends IntegrationTest {
3128
3229 private static final String INJECTION_NAME = "AMSI Bypass - AMSI InitFailed" ;
33- private static final String INJECTOR_TYPE = "openaev_implant" ;
3430
3531 // Saved entities for test setup
3632 @ Autowired private InjectComposer injectComposer ;
3733 @ Autowired private InjectExpectationComposer injectExpectationComposer ;
3834 @ Autowired private AgentComposer agentComposer ;
3935 @ Autowired private EndpointComposer endpointComposer ;
40- @ Autowired private EntityManager entityManager ;
36+ @ Autowired private InjectorFixture injectorFixture ;
4137
4238 @ Autowired private InjectExpectationRepository injectExpectationRepository ;
4339 @ Autowired private InjectorContractRepository injectorContractRepository ;
@@ -58,14 +54,13 @@ class InjectExpectationServiceTest extends IntegrationTest {
5854 void beforeAll () throws JsonProcessingException {
5955 InjectorContract injectorContract =
6056 InjectorContractFixture .createInjectorContract (Map .of ("en" , INJECTION_NAME ));
61- savedInjector =
62- injectorRepository .save (
63- InjectorFixture .createInjector (
64- OPENAEV_INJECTOR_ID , OPENAEV_INJECTOR_NAME , INJECTOR_TYPE ));
57+
58+ savedInjector = injectorFixture .getWellKnownOaevImplantInjector ();
6559 injectorContract .addInjector (savedInjector );
6660
6761 savedInjectorContract = injectorContractRepository .save (injectorContract );
6862 savedInjector .getContracts ().add (savedInjectorContract );
63+ savedInjector .setNewEntity (false );
6964 injectorRepository .save (savedInjector );
7065 savedAsset = assetRepository .save (AssetFixture .createDefaultAsset ("asset name" ));
7166 collectorComposer .forCollector (CollectorFixture .createDefaultCollector ("FAKE" )).persist ();
0 commit comments