ConnectionDetailsFactoriesTests Class — spring-boot Architecture

Source Code

core/spring-boot-autoconfigure/src/test/java/org/springframework/boot/autoconfigure/service/connection/ConnectionDetailsFactoriesTests.java lines 41–181

class ConnectionDetailsFactoriesTests {

	private final MockSpringFactoriesLoader loader = new MockSpringFactoriesLoader();

	@Test
	void getRequiredConnectionDetailsWhenNoFactoryForSourceThrowsException() {
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		assertThatExceptionOfType(ConnectionDetailsFactoryNotFoundException.class)
			.isThrownBy(() -> factories.getConnectionDetails("source", true));
	}

	@Test
	void getOptionalConnectionDetailsWhenNoFactoryForSourceThrowsException() {
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		assertThat(factories.getConnectionDetails("source", false)).isEmpty();
	}

	@Test
	void getConnectionDetailsWhenSourceHasOneMatchReturnsSingleResult() {
		this.loader.addInstance(ConnectionDetailsFactory.class, new TestConnectionDetailsFactory());
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		Map<Class<?>, ConnectionDetails> connectionDetails = factories.getConnectionDetails("source", false);
		assertThat(connectionDetails).hasSize(1);
		assertThat(connectionDetails.get(TestConnectionDetails.class)).isInstanceOf(TestConnectionDetailsImpl.class);
	}

	@Test
	void getRequiredConnectionDetailsWhenSourceHasNoMatchTheowsException() {
		this.loader.addInstance(ConnectionDetailsFactory.class, new NullResultTestConnectionDetailsFactory());
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		assertThatExceptionOfType(ConnectionDetailsNotFoundException.class)
			.isThrownBy(() -> factories.getConnectionDetails("source", true));
	}

	@Test
	void getOptionalConnectionDetailsWhenSourceHasNoMatchReturnsEmptyMap() {
		this.loader.addInstance(ConnectionDetailsFactory.class, new NullResultTestConnectionDetailsFactory());
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		Map<Class<?>, ConnectionDetails> connectionDetails = factories.getConnectionDetails("source", false);
		assertThat(connectionDetails).isEmpty();
	}

	@Test
	void getConnectionDetailsWhenSourceHasMultipleMatchesReturnsMultipleResults() {
		this.loader.addInstance(ConnectionDetailsFactory.class, new TestConnectionDetailsFactory(),
				new OtherConnectionDetailsFactory());
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		Map<Class<?>, ConnectionDetails> connectionDetails = factories.getConnectionDetails("source", false);
		assertThat(connectionDetails).hasSize(2);
	}

	@Test
	void getConnectionDetailsWhenDuplicatesThrowsException() {
		this.loader.addInstance(ConnectionDetailsFactory.class, new TestConnectionDetailsFactory(),
				new TestConnectionDetailsFactory());
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		assertThatIllegalStateException().isThrownBy(() -> factories.getConnectionDetails("source", false))
			.withMessage("Duplicate connection details supplied for " + TestConnectionDetails.class.getName());
	}

	@Test
	void getRegistrationsReturnsOrderedDelegates() {
		TestConnectionDetailsFactory orderOne = new TestConnectionDetailsFactory(1);
		TestConnectionDetailsFactory orderTwo = new TestConnectionDetailsFactory(2);
		TestConnectionDetailsFactory orderThree = new TestConnectionDetailsFactory(3);
		this.loader.addInstance(ConnectionDetailsFactory.class, orderOne, orderThree, orderTwo);
		ConnectionDetailsFactories factories = new ConnectionDetailsFactories(this.loader);
		List<Registration<String, ?>> registrations = factories.getRegistrations("source", false);
		assertThat(registrations.get(0).factory()).isEqualTo(orderOne);
		assertThat(registrations.get(1).factory()).isEqualTo(orderTwo);
		assertThat(registrations.get(2).factory()).isEqualTo(orderThree);
	}

	@Test
	void factoryLoadFailureDoesNotPreventOtherFactoriesFromLoading() {
		this.loader.add(ConnectionDetailsFactory.class.getName(), "com.example.NonExistentConnectionDetailsFactory");
		assertThatNoException().isThrownBy(() -> new ConnectionDetailsFactories(this.loader));
	}

	private static final class TestConnectionDetailsFactory
			implements ConnectionDetailsFactory<String, TestConnectionDetails>, Ordered {

		private final int order;

		private TestConnectionDetailsFactory() {
			this(0);
		}

		private TestConnectionDetailsFactory(int order) {
			this.order = order;
		}

		@Override
		public TestConnectionDetails getConnectionDetails(String source) {
			return new TestConnectionDetailsImpl();
		}

		@Override
		public int getOrder() {
			return this.order;
		}

	}

	private static final class NullResultTestConnectionDetailsFactory
			implements ConnectionDetailsFactory<String, TestConnectionDetails> {

		@Override
		public @Nullable TestConnectionDetails getConnectionDetails(String source) {
			return null;
		}

	}

	private static final class OtherConnectionDetailsFactory
			implements ConnectionDetailsFactory<String, OtherConnectionDetails> {

		@Override
		public OtherConnectionDetails getConnectionDetails(String source) {
			return new OtherConnectionDetailsImpl();
		}

	}

	private interface TestConnectionDetails extends ConnectionDetails {

	}

	private static final class TestConnectionDetailsImpl implements TestConnectionDetails {

	}

	private interface OtherConnectionDetails extends ConnectionDetails {

	}

	private static final class OtherConnectionDetailsImpl implements OtherConnectionDetails {

	}

}