yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-08-03 06:47:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(mce-consumer): mockbeans (#6878)

Browse Source
This commit is contained in:
david-leifker 2022-12-28 16:18:10 -06:00 committed by GitHub
parent 1503ef38f9
commit b5275acab8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
metadata-jobs/mce-consumer-job/src/test/java/com/linkedin/metadata/kafka/MceConsumerApplicationTestConfiguration.java
View File

@ -8,13 +8,13 @@ import com.linkedin.metadata.graph.SiblingGraphService;
import com.linkedin.metadata.models.registry.ConfigEntityRegistry;
import com.linkedin.metadata.models.registry.EntityRegistry;
import com.linkedin.metadata.restli.DefaultRestliClientFactory;
import com.linkedin.metadata.timeseries.elastic.ElasticSearchTimeseriesAspectService;
import com.linkedin.metadata.timeseries.TimeseriesAspectService;
import com.linkedin.parseq.retry.backoff.ExponentialBackoff;
import com.linkedin.restli.client.Client;
import io.ebean.EbeanServer;
import org.mockito.Mockito;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.TestConfiguration;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.boot.test.web.client.TestRestTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Import;
@ -29,17 +29,11 @@ public class MceConsumerApplicationTestConfiguration {
@Autowired
private TestRestTemplate restTemplate;
@Bean("kafkaHealthChecker")
@Primary
public KafkaHealthChecker kafkaHealthChecker() {
return Mockito.mock(KafkaHealthChecker.class);
}
@MockBean
public KafkaHealthChecker kafkaHealthChecker;
@Bean("entityService")
@Primary
public EntityService entityService() {
return Mockito.mock(EntityService.class);
}
@MockBean
public EntityService entityService;
@Bean("restliEntityClient")
@Primary
@ -49,32 +43,18 @@ public class MceConsumerApplicationTestConfiguration {
return new RestliEntityClient(restClient, new ExponentialBackoff(1), 1);
}
@Bean
@Primary
public EbeanServer ebeanServer() {
return Mockito.mock(EbeanServer.class);
}
@MockBean
public EbeanServer ebeanServer;
@Bean(name = "elasticSearchTimeseriesAspectService")
@Primary
protected ElasticSearchTimeseriesAspectService elasticSearchTimeseriesAspectService() {
return Mockito.mock(ElasticSearchTimeseriesAspectService.class);
}
@MockBean
protected TimeseriesAspectService timeseriesAspectService;
@Bean("entityRegistry")
@Primary
protected EntityRegistry entityRegistry() {
return Mockito.mock(EntityRegistry.class);
}
@MockBean
protected EntityRegistry entityRegistry;
@Bean("configEntityRegistry")
protected ConfigEntityRegistry configEntityRegistry() {
return Mockito.mock(ConfigEntityRegistry.class);
}
@MockBean
protected ConfigEntityRegistry configEntityRegistry;
@Bean("siblingGraphService")
@Primary
protected SiblingGraphService siblingGraphService() {
return Mockito.mock(SiblingGraphService.class);
}
@MockBean
protected SiblingGraphService siblingGraphService;
}