How to use getViburDataSourceWithDriverClassName method of org.testcontainers.jdbc.mysql.JDBCDriverWithPoolTest class

Best Testcontainers-java code snippet using org.testcontainers.jdbc.mysql.JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName

Source:JDBCDriverWithPoolTest.java

...34            JDBCDriverWithPoolTest::getTomcatDataSourceWithDriverClassName,35            JDBCDriverWithPoolTest::getTomcatDataSource,36            JDBCDriverWithPoolTest::getHikariDataSourceWithDriverClassName,37            JDBCDriverWithPoolTest::getHikariDataSource,38            JDBCDriverWithPoolTest::getViburDataSourceWithDriverClassName,39            JDBCDriverWithPoolTest::getViburDataSource40        );41    }42    public JDBCDriverWithPoolTest(Supplier<DataSource> dataSourceSupplier) {43        this.dataSource = dataSourceSupplier.get();44    }45    private ExecutorService executorService = Executors.newFixedThreadPool(5);46    @Test47    public void testMySQLWithConnectionPoolUsingSameContainer() throws SQLException, InterruptedException {48        // Populate the database with some data in multiple threads, so that multiple connections from the pool will be used49        for (int i = 0; i < 100; i++) {50            executorService.submit(() -> {51                try {52                    new QueryRunner(dataSource).insert("INSERT INTO my_counter (n) VALUES (5)",53                        (ResultSetHandler<Object>) rs -> true);54                } catch (SQLException e) {55                    e.printStackTrace();56                }57            });58        }59        // Complete population of the database60        executorService.shutdown();61        executorService.awaitTermination(5, TimeUnit.MINUTES);62        // compare to expected results63        int count = new QueryRunner(dataSource).query("SELECT COUNT(1) FROM my_counter", rs -> {64            rs.next();65            return rs.getInt(1);66        });67        assertEquals("Reuse of a datasource points to the same DB container", 100, count);68        int sum = new QueryRunner(dataSource).query("SELECT SUM(n) FROM my_counter", rs -> {69            rs.next();70            return rs.getInt(1);71        });72        // 100 records * 5 = 500 expected73        assertEquals("Reuse of a datasource points to the same DB container", 500, sum);74    }75    private static DataSource getTomcatDataSourceWithDriverClassName() {76        PoolProperties poolProperties = new PoolProperties();77        poolProperties.setUrl(URL + ";TEST=TOMCAT_WITH_CLASSNAME"); // append a dummy URL element to ensure different DB per test78        poolProperties.setValidationQuery("SELECT 1");79        poolProperties.setMinIdle(3);80        poolProperties.setMaxActive(10);81        poolProperties.setDriverClassName(ContainerDatabaseDriver.class.getName());82        return new org.apache.tomcat.jdbc.pool.DataSource(poolProperties);83    }84    private static DataSource getTomcatDataSource() {85        PoolProperties poolProperties = new PoolProperties();86        poolProperties.setUrl(URL + ";TEST=TOMCAT"); // append a dummy URL element to ensure different DB per test87        poolProperties.setValidationQuery("SELECT 1");88        poolProperties.setInitialSize(3);89        poolProperties.setMaxActive(10);90        return new org.apache.tomcat.jdbc.pool.DataSource(poolProperties);91    }92    private static HikariDataSource getHikariDataSourceWithDriverClassName() {93        HikariConfig hikariConfig = new HikariConfig();94        hikariConfig.setJdbcUrl(URL + ";TEST=HIKARI_WITH_CLASSNAME"); // append a dummy URL element to ensure different DB per test95        hikariConfig.setConnectionTestQuery("SELECT 1");96        hikariConfig.setMinimumIdle(3);97        hikariConfig.setMaximumPoolSize(10);98        hikariConfig.setDriverClassName(ContainerDatabaseDriver.class.getName());99        return new HikariDataSource(hikariConfig);100    }101    private static HikariDataSource getHikariDataSource() {102        HikariConfig hikariConfig = new HikariConfig();103        hikariConfig.setJdbcUrl(URL + ";TEST=HIKARI"); // append a dummy URL element to ensure different DB per test104        hikariConfig.setConnectionTestQuery("SELECT 1");105        hikariConfig.setMinimumIdle(3);106        hikariConfig.setMaximumPoolSize(10);107        return new HikariDataSource(hikariConfig);108    }109    private static DataSource getViburDataSourceWithDriverClassName() {110        ViburDBCPDataSource ds = new ViburDBCPDataSource();111        ds.setJdbcUrl(URL + ";TEST=VIBUR_WITH_CLASSNAME");112        ds.setUsername("any");  // Recent versions of Vibur require a username, even though it will not be used113        ds.setPassword("");114        ds.setPoolInitialSize(3);115        ds.setPoolMaxSize(10);116        ds.setTestConnectionQuery("SELECT 1");117        ds.setDriverClassName(ContainerDatabaseDriver.class.getName());118        ds.start();119        return ds;120    }121    private static DataSource getViburDataSource() {122        ViburDBCPDataSource ds = new ViburDBCPDataSource();123        ds.setJdbcUrl(URL + ";TEST=VIBUR");...

getViburDataSourceWithDriverClassName

Using AI Code Generation

1    public void testViburDataSource() throws SQLException {2        DataSource dataSource = getViburDataSourceWithDriverClassName();3        try (Connection connection = dataSource.getConnection()) {4            try (Statement statement = connection.createStatement()) {5                try (ResultSet resultSet = statement.executeQuery("SELECT 1")) {6                    resultSet.next();7                    assertThat(resultSet.getInt(1), is(1));8                }9            }10        }11    }12    private DataSource getViburDataSourceWithDriverClassName() throws SQLException {13        JdbcDatabaseContainer container = new MySQLContainer()14            .withDatabaseName("test")15            .withUsername("test")16            .withPassword("test");17        container.start();18        ViburDBCPDataSource dataSource = new ViburDBCPDataSource();19        dataSource.setJdbcUrl(container.getJdbcUrl());20        dataSource.setUsername(container.getUsername());21        dataSource.setPassword(container.getPassword());22        dataSource.setDriverClassName(container.getDriverClassName());23        dataSource.setConnectionTimeoutInMs(1000);24        dataSource.setLoginTimeoutInSeconds(1);25        dataSource.setTestConnectionQuery("SELECT 1");26        dataSource.start();27        return dataSource;28    }29}

getViburDataSourceWithDriverClassName

Using AI Code Generation

1ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.cj.jdbc.Driver");2ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.jdbc.Driver");3ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.cj.jdbc.Driver", "mysql");4ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.jdbc.Driver", "mysql");5ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.cj.jdbc.Driver", "mysql", "mysql");6ViburDataSource dataSource = JDBCDriverWithPoolTest.getViburDataSourceWithDriverClassName("com.mysql.jdbc.Driver", "mysql", "mysql");

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.