package com.linkedin.metadata.kafka;

import com.linkedin.entity.client.RestliEntityClient;
import com.linkedin.gms.factory.auth.SystemAuthenticationFactory;
import com.linkedin.metadata.dao.producer.KafkaHealthChecker;
import com.linkedin.metadata.entity.EntityService;
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.TimeseriesAspectService;
import com.linkedin.parseq.retry.backoff.ExponentialBackoff;
import com.linkedin.restli.client.Client;
import io.ebean.EbeanServer;
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;
import org.springframework.context.annotation.Primary;

import java.net.URI;

@TestConfiguration
@Import(value = {SystemAuthenticationFactory.class})
public class MceConsumerApplicationTestConfiguration {

    @Autowired
    private TestRestTemplate restTemplate;

    @MockBean
    public KafkaHealthChecker kafkaHealthChecker;

    @MockBean
    public EntityService entityService;

    @Bean("restliEntityClient")
    @Primary
    public RestliEntityClient restliEntityClient() {
        String selfUri = restTemplate.getRootUri();
        final Client restClient = DefaultRestliClientFactory.getRestLiClient(URI.create(selfUri), null);
        return new RestliEntityClient(restClient, new ExponentialBackoff(1), 1);
    }

    @MockBean
    public EbeanServer ebeanServer;

    @MockBean
    protected TimeseriesAspectService timeseriesAspectService;

    @MockBean
    protected EntityRegistry entityRegistry;

    @MockBean
    protected ConfigEntityRegistry configEntityRegistry;

    @MockBean
    protected SiblingGraphService siblingGraphService;
}