DiscoveryClientConfigurationTest.java 3.02 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
package de.codecentric.boot.admin.config;

import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;

import org.junit.After;
import org.junit.Test;
import org.springframework.boot.autoconfigure.PropertyPlaceholderAutoConfiguration;
import org.springframework.boot.autoconfigure.web.ServerPropertiesAutoConfiguration;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.cloud.client.discovery.noop.NoopDiscoveryClientAutoConfiguration;
import org.springframework.cloud.commons.util.UtilAutoConfiguration;
import org.springframework.cloud.netflix.eureka.EurekaClientAutoConfiguration;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;

import de.codecentric.boot.admin.discovery.DefaultServiceInstanceConverter;
import de.codecentric.boot.admin.discovery.EurekaServiceInstanceConverter;
import de.codecentric.boot.admin.discovery.ServiceInstanceConverter;
import de.codecentric.boot.admin.model.Application;

public class DiscoveryClientConfigurationTest {

	private AnnotationConfigWebApplicationContext context;

	@After
	public void close() {
		if (this.context != null) {
			this.context.close();
		}
	}

	@Test
	public void defaultServiceInstanceConverter() {
		load(NoopDiscoveryClientAutoConfiguration.class);
		assertThat(context.getBean(ServiceInstanceConverter.class),
				is(instanceOf(DefaultServiceInstanceConverter.class)));
	}

	@Test
	public void eurekaServiceInstanceConverter() {
		load(UtilAutoConfiguration.class, EurekaClientAutoConfiguration.class);
		assertThat(context.getBean(ServiceInstanceConverter.class),
				is(instanceOf(EurekaServiceInstanceConverter.class)));
	}

	@Test
	public void customServiceInstanceConverter() {
		load(NoopDiscoveryClientAutoConfiguration.class,
				TestCustomServiceInstanceConverterConfig.class);
		assertThat(context.getBean(ServiceInstanceConverter.class),
				is(instanceOf(CustomServiceInstanceConverter.class)));
	}


	@Configuration
	static class TestCustomServiceInstanceConverterConfig {
		@Bean
		public CustomServiceInstanceConverter converter() {
			return new CustomServiceInstanceConverter();
		}
	}

	static class CustomServiceInstanceConverter implements ServiceInstanceConverter {
		@Override
		public Application convert(ServiceInstance instance) {
			return null;
		}
	}

	private void load(Class<?>... configs) {
		AnnotationConfigWebApplicationContext applicationContext = new AnnotationConfigWebApplicationContext();
		for (Class<?> config : configs) {
			applicationContext.register(config);
		}

		applicationContext.register(PropertyPlaceholderAutoConfiguration.class);
		applicationContext.register(ServerPropertiesAutoConfiguration.class);
		applicationContext.register(AdminServerWebConfiguration.class);
		applicationContext.register(DiscoveryClientConfiguration.class);

		applicationContext.refresh();
		this.context = applicationContext;
	}
}