/* * Copyright 2019 the original author or authors. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package io.zonky.test.db.provider; import io.zonky.test.db.AutoConfigureEmbeddedDatabase; import io.zonky.test.db.config.PostgreSQLContainerCustomizer; import org.junit.Test; import org.junit.runner.RunWith; import org.postgresql.ds.PGSimpleDataSource; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.TestPropertySource; import org.springframework.test.context.junit4.SpringRunner; import javax.sql.DataSource; import java.sql.SQLException; import static io.zonky.test.db.AutoConfigureEmbeddedDatabase.DatabaseProvider.DOCKER; import static org.assertj.core.api.Assertions.assertThat; @RunWith(SpringRunner.class) @AutoConfigureEmbeddedDatabase(beanName = "dataSource", provider = DOCKER) @TestPropertySource(properties = { "zonky.test.database.postgres.docker.image=postgres:9.6-alpine", "zonky.test.database.postgres.docker.tmpfs.enabled=true" }) @ContextConfiguration public class DockerProviderWithConfigurationIntegrationTest { @Configuration static class Config { @Bean public PostgreSQLContainerCustomizer postgresContainerCustomizer() { return container -> container.withPassword("docker-postgres"); } } @Autowired private DataSource dataSource; @Test public void testDataSource() throws SQLException { assertThat(dataSource.unwrap(PGSimpleDataSource.class).getPassword()).isEqualTo("docker-postgres"); JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource); String version = jdbcTemplate.queryForObject("show server_version", String.class); assertThat(version).startsWith("9.6."); } }