View Javadoc
1   /*
2    *    Copyright 2009-2023 the original author or authors.
3    *
4    *    Licensed under the Apache License, Version 2.0 (the "License");
5    *    you may not use this file except in compliance with the License.
6    *    You may obtain a copy of the License at
7    *
8    *       https://www.apache.org/licenses/LICENSE-2.0
9    *
10   *    Unless required by applicable law or agreed to in writing, software
11   *    distributed under the License is distributed on an "AS IS" BASIS,
12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *    See the License for the specific language governing permissions and
14   *    limitations under the License.
15   */
16  package org.apache.ibatis.testcontainers;
17  
18  import javax.sql.DataSource;
19  
20  import org.apache.ibatis.datasource.unpooled.UnpooledDataSource;
21  import org.testcontainers.containers.PostgreSQLContainer;
22  
23  public class PgContainer {
24  
25    private static final String DB_NAME = "mybatis_test";
26    private static final String USERNAME = "u";
27    private static final String PASSWORD = "p";
28    private static final String DRIVER = "org.postgresql.Driver";
29  
30    private static final PostgreSQLContainer<?> INSTANCE = initContainer();
31  
32    private static PostgreSQLContainer<?> initContainer() {
33      @SuppressWarnings("resource")
34      PostgreSQLContainer<?> container = new PostgreSQLContainer<>().withDatabaseName(DB_NAME).withUsername(USERNAME)
35          .withPassword(PASSWORD);
36      container.start();
37      return container;
38    }
39  
40    public static DataSource getUnpooledDataSource() {
41      return new UnpooledDataSource(PgContainer.DRIVER, INSTANCE.getJdbcUrl(), PgContainer.USERNAME,
42          PgContainer.PASSWORD);
43    }
44  
45    private PgContainer() {
46    }
47  }