package example;

import io.github.m4gshm.testcontainers.PostgresqlPod;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.testcontainers.containers.JdbcDatabaseContainer;
import org.testcontainers.containers.PostgreSQLContainer;

import java.sql.SQLException;

import static java.sql.DriverManager.getConnection;

public class JdbcTest {

    private final static JdbcDatabaseContainer<?> postgres = useKubernetes()
            ? new PostgresqlPod().withDeletePodOnStop(true).withDeletePodOnError(true)
            : new PostgreSQLContainer<>();

    private static boolean useKubernetes() {
        return "kuber".equals(System.getProperty("testcontainers-engine", "kuber"));
    }

    @BeforeAll
    static void beforeAll() {
        postgres.start();
    }

    @AfterAll
    static void afterAll() {
        postgres.stop();
    }

    @Test
    public void jdbcInsertSelectTest() throws SQLException {
        var url = postgres.getJdbcUrl();
        var username = postgres.getUsername();
        var password = postgres.getPassword();
        try (var connection = getConnection(url, username, password)) {
            try (var statement = connection.prepareStatement("""
                    create table participant (
                        id integer,
                        name text
                    );
                    """
            )) {
                statement.execute();
            }

            try (var statement = connection.prepareStatement(
                    "insert into participant(id, name) values(?,?)")) {
                statement.setInt(1, 1);
                statement.setString(2, "Alice");
                statement.execute();
            }

            try (var statement = connection.prepareStatement(
                    "select name from participant where id=?")) {
                statement.setInt(1, 1);
                try (var resultSet = statement.executeQuery()) {
                    Assertions.assertTrue(resultSet.next());
                    var name = resultSet.getString(1);
                    Assertions.assertEquals("Alice", name);
                }
            }
        }
    }
}