testcontainers-java

1
package org.testcontainers.containers;
2

3
import org.junit.Test;
4
import org.testcontainers.db.AbstractContainerDatabaseTest;
5
import org.testcontainers.utility.DockerImageName;
6

7
import java.sql.ResultSet;
8
import java.sql.SQLException;
9

10
import static org.assertj.core.api.Assertions.assertThat;
11

12
public class CompatibleImageTest extends AbstractContainerDatabaseTest {
13

14
    @Test
15
    public void pgvector() throws SQLException {
16
        try (
17
            // pgvectorContainer {
18
            PostgreSQLContainer<?> pgvector = new PostgreSQLContainer<>("pgvector/pgvector:pg16")
19
            // }
20
        ) {
21
            pgvector.start();
22

23
            ResultSet resultSet = performQuery(pgvector, "SELECT 1");
24
            int resultSetInt = resultSet.getInt(1);
25
            assertThat(resultSetInt).as("A basic SELECT query succeeds").isEqualTo(1);
26
        }
27
    }
28

29
    @Test
30
    public void postgis() throws SQLException {
31
        try (
32
            // postgisContainer {
33
            PostgreSQLContainer<?> postgis = new PostgreSQLContainer<>(
34
                DockerImageName.parse("postgis/postgis:16-3.4-alpine").asCompatibleSubstituteFor("postgres")
35
            )
36
            // }
37
        ) {
38
            postgis.start();
39

40
            ResultSet resultSet = performQuery(postgis, "SELECT 1");
41
            int resultSetInt = resultSet.getInt(1);
42
            assertThat(resultSetInt).as("A basic SELECT query succeeds").isEqualTo(1);
43
        }
44
    }
45

46
    @Test
47
    public void timescaledb() throws SQLException {
48
        try (
49
            // timescaledbContainer {
50
            PostgreSQLContainer<?> timescaledb = new PostgreSQLContainer<>(
51
                DockerImageName.parse("timescale/timescaledb:2.14.2-pg16").asCompatibleSubstituteFor("postgres")
52
            )
53
            // }
54
        ) {
55
            timescaledb.start();
56

57
            ResultSet resultSet = performQuery(timescaledb, "SELECT 1");
58
            int resultSetInt = resultSet.getInt(1);
59
            assertThat(resultSetInt).as("A basic SELECT query succeeds").isEqualTo(1);
60
        }
61
    }
62
}
63