test(mae-consumer): test for injection of pe-consumer (#10755)

datahub-project · Jun 22, 2024 · 710e605 · 710e605
1 parent 79ad582
commit 710e605
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/metadata-jobs/mae-consumer-job/build.gradle b/metadata-jobs/mae-consumer-job/build.gradle
@@ -30,6 +30,7 @@ dependencies {
     implementation externalDependency.logbackClassic
 
     testImplementation project(':metadata-dao-impl:kafka-producer')
+    testImplementation project(':metadata-jobs:pe-consumer')
     testImplementation externalDependency.springBootTest
     testImplementation externalDependency.mockito
     testImplementation externalDependency.testng

diff --git a/...ae-consumer-job/src/test/java/com/linkedin/metadata/kafka/MaeConsumerApplicationTest.java b/...ae-consumer-job/src/test/java/com/linkedin/metadata/kafka/MaeConsumerApplicationTest.java
@@ -2,6 +2,7 @@
 
 import static org.testng.AssertJUnit.*;
 
+import com.datahub.event.PlatformEventProcessor;
 import com.linkedin.metadata.entity.EntityService;
 import com.linkedin.metadata.search.EntitySearchService;
 import com.linkedin.metadata.service.FormService;
@@ -14,7 +15,8 @@
 
 @ActiveProfiles("test")
 @SpringBootTest(
-    classes = {MaeConsumerApplication.class, MaeConsumerApplicationTestConfiguration.class})
+    classes = {MaeConsumerApplication.class, MaeConsumerApplicationTestConfiguration.class},
+    properties = "PE_CONSUMER_ENABLED=true")
 public class MaeConsumerApplicationTest extends AbstractTestNGSpringContextTests {
 
   @Autowired private EntityService<?> mockEntityService;
@@ -25,10 +27,13 @@ public class MaeConsumerApplicationTest extends AbstractTestNGSpringContextTests
 
   @Autowired private EntitySearchService entitySearchService;
 
+  @Autowired private PlatformEventProcessor platformEventProcessor;
+
   @Test
   public void testMaeConsumerAutoWiring() {
     assertNotNull(mockEntityService);
     assertNotNull(kafkaHealthIndicator);
     assertNotNull(formService);
+    assertNotNull(platformEventProcessor);
   }
 }