yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-08-05 07:47:56 +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.ConfigEntityRegistry;
import com.linkedin.metadata.models.registry.EntityRegistry; import com.linkedin.metadata.models.registry.EntityRegistry;
import com.linkedin.metadata.restli.DefaultRestliClientFactory; 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.parseq.retry.backoff.ExponentialBackoff;
import com.linkedin.restli.client.Client; import com.linkedin.restli.client.Client;
import io.ebean.EbeanServer; import io.ebean.EbeanServer;
import org.mockito.Mockito;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.TestConfiguration; 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.boot.test.web.client.TestRestTemplate;
import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Import; import org.springframework.context.annotation.Import;
@ -29,17 +29,11 @@ public class MceConsumerApplicationTestConfiguration {
@Autowired @Autowired
private TestRestTemplate restTemplate; private TestRestTemplate restTemplate;
@Bean("kafkaHealthChecker") @MockBean
@Primary public KafkaHealthChecker kafkaHealthChecker;
public KafkaHealthChecker kafkaHealthChecker() {
return Mockito.mock(KafkaHealthChecker.class);
}
@Bean("entityService") @MockBean
@Primary public EntityService entityService;
public EntityService entityService() {
return Mockito.mock(EntityService.class);
}
@Bean("restliEntityClient") @Bean("restliEntityClient")
@Primary @Primary
@ -49,32 +43,18 @@ public class MceConsumerApplicationTestConfiguration {
return new RestliEntityClient(restClient, new ExponentialBackoff(1), 1); return new RestliEntityClient(restClient, new ExponentialBackoff(1), 1);
} }
@Bean @MockBean
@Primary public EbeanServer ebeanServer;
public EbeanServer ebeanServer() {
return Mockito.mock(EbeanServer.class);
}
@Bean(name = "elasticSearchTimeseriesAspectService") @MockBean
@Primary protected TimeseriesAspectService timeseriesAspectService;
protected ElasticSearchTimeseriesAspectService elasticSearchTimeseriesAspectService() {
return Mockito.mock(ElasticSearchTimeseriesAspectService.class);
}
@Bean("entityRegistry") @MockBean
@Primary protected EntityRegistry entityRegistry;
protected EntityRegistry entityRegistry() {
return Mockito.mock(EntityRegistry.class);
}
@Bean("configEntityRegistry") @MockBean
protected ConfigEntityRegistry configEntityRegistry() { protected ConfigEntityRegistry configEntityRegistry;
return Mockito.mock(ConfigEntityRegistry.class);
}
@Bean("siblingGraphService") @MockBean
@Primary protected SiblingGraphService siblingGraphService;
protected SiblingGraphService siblingGraphService() {
return Mockito.mock(SiblingGraphService.class);
}
} }