KaneAI - World’s First E2E Software Testing Agent.Signup for Private Beta Join TestMu Conference

How to use TiDBContainer class of org.testcontainers.tidb package

Best Testcontainers-java code snippet using org.testcontainers.tidb.TiDBContainer

Source:AppTest.java

2import com.mysql.cj.jdbc.MysqlDataSource;3import org.junit.jupiter.api.BeforeAll;4import org.junit.jupiter.api.Test;5import org.testcontainers.junit.jupiter.Container;6import org.testcontainers.tidb.TiDBContainer;7import javax.sql.DataSource;8import java.sql.ResultSet;9import java.sql.SQLException;10import java.sql.Statement;11public class AppTest12{13    // Default TiDB version at DockerHub: https://hub.docker.com/r/pingcap/tidb14    @Container15    private static final TiDBContainer tidb = new TiDBContainer("pingcap/tidb");16    public static void printTiDBParams(TiDBContainer tidb) {17        System.out.println("db: " + tidb.getDatabaseName());18        System.out.println("username: " + tidb.getUsername());19        System.out.println("password: " + tidb.getPassword());20        System.out.println("jdbc: " + tidb.getJdbcUrl());21    }22    @BeforeAll23    public static void beforeAll() {24        tidb.start();25    }26    protected ResultSet performQuery(TiDBContainer container, String sql) throws SQLException {27        DataSource ds = getDataSource(container);28        Statement statement = ds.getConnection().createStatement();29        statement.execute(sql);30        return statement.getResultSet();31    }32    protected DataSource getDataSource(TiDBContainer container) throws SQLException {33        MysqlDataSource dataSource = new MysqlDataSource();34        dataSource.setURL(container.getJdbcUrl());35        dataSource.setUser(container.getUsername());36        dataSource.setPassword(container.getPassword());37        dataSource.setUseSSL(false);38        return dataSource;39    }40    @Test41    public void printDefaultTiDBVersion() throws SQLException {42        System.out.println("------------ TiDB default version ------------");43        printTiDBParams(tidb);44        ResultSet rs = performQuery(tidb, "SELECT VERSION()");45        while (rs.next()) {46            System.out.println(rs.getString(1));47        }48    }49    @Test50    public void printParticularTiDBVersion() throws SQLException {51        // Appoint TiDB version to v6.2.052        try (TiDBContainer tidb62 = new TiDBContainer("pingcap/tidb:v6.2.0")) {53            tidb62.start();54            System.out.println("------------ TiDB v6.2.0 version ------------");55            printTiDBParams(tidb62);56            ResultSet rs = performQuery(tidb62, "SELECT VERSION()");57            while (rs.next()) {58                System.out.println(rs.getString(1));59            }60        }61    }62}...

Source:SimpleTiDBTest.java

1package org.testcontainers.junit.tidb;2import org.junit.Test;3import org.testcontainers.TiDBTestImages;4import org.testcontainers.db.AbstractContainerDatabaseTest;5import org.testcontainers.tidb.TiDBContainer;6import java.sql.ResultSet;7import java.sql.SQLException;8import static org.assertj.core.api.Assertions.assertThat;9public class SimpleTiDBTest extends AbstractContainerDatabaseTest {10    @Test11    public void testSimple() throws SQLException {12        try (TiDBContainer tidb = new TiDBContainer(TiDBTestImages.TIDB_IMAGE)) {13            tidb.start();14            ResultSet resultSet = performQuery(tidb, "SELECT 1");15            int resultSetInt = resultSet.getInt(1);16            assertThat(resultSetInt).isEqualTo(1);17            assertHasCorrectExposedAndLivenessCheckPorts(tidb);18        }19    }20    @Test21    public void testExplicitInitScript() throws SQLException {22        try (23            TiDBContainer tidb = new TiDBContainer(TiDBTestImages.TIDB_IMAGE).withInitScript("somepath/init_tidb.sql")24        ) { // TiDB is expected to be compatible with MySQL25            tidb.start();26            ResultSet resultSet = performQuery(tidb, "SELECT foo FROM bar");27            String firstColumnValue = resultSet.getString(1);28            assertThat(firstColumnValue).isEqualTo("hello world");29        }30    }31    @Test32    public void testWithAdditionalUrlParamInJdbcUrl() {33        TiDBContainer tidb = new TiDBContainer(TiDBTestImages.TIDB_IMAGE).withUrlParam("sslmode", "disable");34        try {35            tidb.start();36            String jdbcUrl = tidb.getJdbcUrl();37            assertThat(jdbcUrl).contains("?");38            assertThat(jdbcUrl).contains("sslmode=disable");39        } finally {40            tidb.stop();41        }42    }43    private void assertHasCorrectExposedAndLivenessCheckPorts(TiDBContainer tidb) {44        Integer tidbPort = 4000;45        Integer restApiPort = 10080;46        assertThat(tidb.getExposedPorts()).containsExactlyInAnyOrder(tidbPort, restApiPort);47        assertThat(tidb.getLivenessCheckPortNumbers())48            .containsExactlyInAnyOrder(tidb.getMappedPort(tidbPort), tidb.getMappedPort(restApiPort));49    }50}...

Source:TiDBTest.java

...7import org.springframework.test.context.DynamicPropertyRegistry;8import org.springframework.test.context.DynamicPropertySource;9import org.testcontainers.junit.jupiter.Container;10import org.testcontainers.junit.jupiter.Testcontainers;11import org.testcontainers.tidb.TiDBContainer;12import static org.assertj.core.api.Assertions.assertThat;13@Testcontainers14@DataJdbcTest15@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)16class TiDBTest {17	@Container18	private static final TiDBContainer postgres = new TiDBContainer("pingcap/tidb:v6.1.0");19	@Autowired20	private JdbcTemplate jdbcTemplate;21	@DynamicPropertySource22	static void sqlserverProperties(DynamicPropertyRegistry registry) {23		registry.add("spring.datasource.url", postgres::getJdbcUrl);24		registry.add("spring.datasource.username", postgres::getUsername);25		registry.add("spring.datasource.password", postgres::getPassword);26	}27	@Test28	void test() {29		var records = this.jdbcTemplate.queryForList("select count(*) from profile");30		assertThat(records).hasSize(1);31	}32}...

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2public class 1 {3    public static void main(String[] args) {4        try (TiDBContainer tidb = new TiDBContainer()) {5            tidb.start();6            System.out.println(tidb.getJdbcUrl());7            System.out.println(tidb.getPassword());8        }9    }10}11import org.testcontainers.containers.TiDBContainer;12public class 2 {13    public static void main(String[] args) {14        try (TiDBContainer tidb = new TiDBContainer()) {15            tidb.start();16            System.out.println(tidb.getJdbcUrl());17            System.out.println(tidb.getPassword());18        }19    }20}21public TiDBContainer() {22        this(TiDBContainer.IMAGE + ":" + TiDBContainer.DEFAULT_TAG);23    }24    public TiDBContainer(final String dockerImageName) {25        super(dockerImageName);26        this.addExposedPorts(4000, 10080);27        this.withCommand("tikv-server --store=tikv --path=

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2import org.testcontainers.containers.output.Slf4jLogConsumer;3import org.slf4j.Logger;4import org.slf4j.LoggerFactory;5public class 1 {6    public static void main(String[] args) {7        Logger logger = LoggerFactory.getLogger(1.class);8        TiDBContainer tidb = new TiDBContainer()9                .withLogConsumer(new Slf4jLogConsumer(logger));10        tidb.start();11        System.out.println(tidb.getJdbcUrl());12        tidb.stop();13    }14}15import org.testcontainers.containers.output.Slf4jLogConsumer;16import org.testcontainers.containers.TiDBContainer;17import org.slf4j.Logger;18import org.slf4j.LoggerFactory;19public class 2 {20    public static void main(String[] args) {21        Logger logger = LoggerFactory.getLogger(2.class);22        TiDBContainer tidb = new TiDBContainer()23                .withLogConsumer(new Slf4jLogConsumer(logger));24        tidb.start();25        System.out.println(tidb.getJdbcUrl());26        tidb.stop();27    }28}

TiDBContainer

Using AI Code Generation

1package org.testcontainers.tidb;2import org.testcontainers.containers.ContainerLaunchException;3import org.testcontainers.containers.GenericContainer;4import org.testcontainers.containers.output.Slf4jLogConsumer;5import org.testcontainers.containers.wait.strategy.Wait;6import org.testcontainers.utility.DockerImageName;7import java.io.IOException;8import java.time.Duration;9import java.util.ArrayList;10import java.util.List;11public class TiDBContainer<SELF extends TiDBContainer<SELF>> extends GenericContainer<SELF> {12    private static final int DEFAULT_PORT = 4000;13    private static final int DEFAULT_TIDB_STATUS_PORT = 10080;14    private static final int DEFAULT_TIDB_DEBUG_PORT = 10090;15    private static final int DEFAULT_TIDB_STATUS_HEALTH_PORT = 10280;16    private static final int DEFAULT_TIDB_DEBUG_HEALTH_PORT = 10290;17    private static final int DEFAULT_TIDB_PORT = 10080;18    private static final int DEFAULT_TIDB_HEALTH_PORT = 10280;19    private static final int DEFAULT_TIDB_PPROF_PORT = 10090;20    private static final int DEFAULT_TIDB_HEALTH_PPROF_PORT = 10290;21    private static final int DEFAULT_TIDB_METRICS_PORT = 9090;22    private static final int DEFAULT_TIDB_HEALTH_METRICS_PORT = 9190;23    private static final String DEFAULT_TAG = "latest";24    private static final String DEFAULT_NETWORK_ALIAS = "tidb";25    private static final String DEFAULT_DOCKER_IMAGE_NAME = "pingcap/tidb";26    private static final String DEFAULT_USERNAME = "root";27    private static final String DEFAULT_PASSWORD = "";28    private static final String DEFAULT_DATABASE_NAME = "test";29    private static final String DEFAULT_CHARACTER_SET = "utf8";30    private static final String DEFAULT_COLLATION = "utf8_general_ci";31    private static final String DEFAULT_TIDB_INIT_SQL_PATH = "/docker-entrypoint-initdb.d";32    private static final String DEFAULT_TIDB_INIT_SQL_FILE = "init.sql";33    private static final String DEFAULT_TIDB_INIT_SQL = "CREATE DATABASE IF NOT EXISTS " + DEFAULT_DATABASE_NAME + " CHARACTER SET " + DEFAULT_CHARACTER_SET + " COLLATE " + DEFAULT_COLLATION + ";";34    private static final String DEFAULT_TIDB_DATA_DIR = "/tidb-data";

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2import org.testcontainers.containers.output.Slf4jLogConsumer;3import org.slf4j.Logger;4import org.slf4j.LoggerFactory;5import org.testcontainers.containers.output.Slf4jLogConsumer;6import org.testcontainers.containers.output.WaitingConsumer;7import org.testcontainers.containers.output.OutputFrame;8public class 1 {9    public static void main(String[] args) {10        Logger logger = LoggerFactory.getLogger(1.class);11        try (TiDBContainer tidb = new TiDBContainer("pingcap/tidb:v3.0.0-rc.1")) {12            tidb.start();13            System.out.println("TiDBContainer started");14            System.out.println("JDBC URL: " + tidb.getJdbcUrl());15            System.out.println("MySQL URL: " + tidb.getJdbcUrl());16            System.out.println("Username: " + tidb.getUsername());17            System.out.println("Password: " + tidb.getPassword());18            System.out.println("Port: " + tidb.getMappedPort(4000));19        }20    }21}22import org.testcontainers.tidb.TiDBContainer;23import org.testcontainers.containers.output.Slf4jLogConsumer;24import org.slf4j.Logger;25import org.slf4j.LoggerFactory;26import org.testcontainers.containers.output.Slf4jLogConsumer;27import org.testcontainers.containers.output.WaitingConsumer;28import org.testcontainers.containers.output.OutputFrame;29public class 2 {30    public static void main(String[] args) {31        Logger logger = LoggerFactory.getLogger(2.class);32        try (TiDBContainer tidb = new TiDBContainer("pingcap/tidb:v3.0.0-rc.1")33                .withConfigurationOverride("conf/tidb.toml")) {34            tidb.start();35            System.out.println("TiDBContainer started");

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2import org.testcontainers.containers.output.Slf4jLogConsumer;3import org.slf4j.Logger;4import org.slf4j.LoggerFactory;5import org.testcontainers.containers.output.Slf4jLogConsumer;6import org.testcontainers.containers.output.WaitingConsumer;7import org.testcontainers.containers.output.OutputFrame;8public class 1 {9    public static void main(String[] args) {10        Logger logger = LoggerFactory.getLogger(1.class);11        try (TiDBContainer tidb = new TiDBContainer("pingcap/tidb:v3.0.0-rc.1")) {12            tidb.start();13            System.out.println("TiDBContainer started");14            System.out.println("JDBC URL: " + tidb.getJdbcUrl());15            System.out.println("MySQL URL: " + tidb.getJdbcUrl());16            System.out.println("Username: " + tidb.getUsername());17            System.out.println("Password: " + tidb.getPassword());18            System.out.println("Port: " + tidb.getMappedPort(4000));19        }20    }21}22import org.testcontainers.tidb.TiDBContainer;23import org.testcontainers.containers.output.Slf4jLogConsumer;24import org.slf4j.Logger;25import org.slf4j.LoggerFactory;26import org.testcontainers.containers.output.Slf4jLogConsumer;27import org.testcontainers.containers.output.WaitingConsumer;28import org.testcontainers.containers.output.OutputFrame;29public class 2 {30    public static void main(String[] args) {31        Logger logger = LoggerFactory.getLogger(2.class);32        try (TiDBContainer tidb = new TiDBContainer("pingcap/tidb:v3.0.0-rc.1")33                .withConfigurationOverride("conf/tidb.toml")) {34            tidb.start();35            System.out.println("TiDBContainer started");

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2import org.testcontainers.containers.wait.strategy.Wait;3import org.testcontainers.containers.GenericContainer;4public class TiDBContainerTest {5    public static void main(String[] args) {6        try (TiDBContainer tidb = new TiDBContainer<>()) {7            tidb.start();8        }9    }10}11import org.testcontainers.containers.wait.strategy.Wait;12import org.testcontainers.containers.GenericContainer;13import org.testcontainers.containers.MySQLContainer;14public class TiDBContainerTest {15    public static void main(String[] args) {16        try (MySQLContainer tidb = new MySQLContainer<>()) {17            tidb.start();18        }19    }20}

TiDBContainer

Using AI Code Generation

1import org.testcontainers.containers.TiDBContainer;2import org.testcontainers.containers.output.Slf4jLogConsumer;3import org.slf4j.Logger;4import org.slf4j.LoggerFactory;5import org.testcontainers.containers.output.OutputFrame;6public class 1 {7    public static void main(String[] args) {8        Logger logger = LoggerFactory.getLogger(1.class);9        TiDBContainer tidb = new TiDBContainer("pingcap/tidb:latest");10        tidb.withLogConsumer(new Slf4jLogConsumer(logger));11        tidb.start();12        tidb.stop();13    }14}15import org.testcontainers.containers.TiDBContainer;16import org.testcontainers.containers.output.Slf4jLogConsumer;17import org.slf4j.Logger;18import org.slf4j.LoggerFactory;19import org.testcontainers.containers.output.OutputFrame;20public class 2 {21    public static void main(String[] args) {22        Logger logger = LoggerFactory.getLogger(2.class);23        TiDBContainer tidb = new TiDBContainer("pingcap/tidb:latest");24        tidb.withLogConsumer(new Slf4jLogConsumer(logger));25        tidb.start();26        tidb.stop();27    }28}29[INFO] --- maven-compiler-plugin:3.1:compile (default-compile) @ 1 ---30[INFO] --- maven-resources-plugin:2.6:resources (default-resources) @ 1 ---31[INFO] --- maven-compiler-plugin:3.1:compile (default-compile) @ 1 ---

TiDBContainer

Using AI Code Generation

1import org.testcontainers.tidb.TiDBContainer;2import org.testcontainers.containers.wait.strategy.Wait;3import org.testcontainers.containers.GenericContainer;4public class TiDBContainerTest {5    public static void main(String[] args) {6        try (TiDBContainer tidb = new TiDBContainer<>()) {7            tidb.start();8        }9    }10}11import org.testcontainers.containers.wait.strategy.Wait;12import org.testcontainers.containers.GenericContainer;13import org.testcontainers.containers.MySQLContainer;14public class TiDBContainerTest {15    public static void main(String[] args) {16        try (MySQLContainer tidb = new MySQLContainer<>()) {17            tidb.start();18        }19    }20}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.