Commit 7dc1b9f8 authored by tristaZero's avatar tristaZero
Browse files

modify mockDataSource

parent c4425d78

sharding-jdbc/src/test/java/io/shardingsphere/core/jdbc/core/datasource/ShardingDataSourceTest.java

+7 −4
Original line number Diff line number Diff line
@@ -26,7 +26,6 @@ import io.shardingsphere.core.constant.DatabaseType; 
import io.shardingsphere.core.constant.properties.ShardingProperties;

import io.shardingsphere.core.constant.properties.ShardingPropertiesConstant;

import io.shardingsphere.core.orche.eventbus.config.jdbc.ShardingConfigurationEventBusEvent;

import io.shardingsphere.core.executor.ShardingExecuteEngine;

import io.shardingsphere.core.rule.ShardingRule;

import org.junit.Test;

import org.mockito.ArgumentMatchers;
@@ -126,15 +125,19 @@ public final class ShardingDataSourceTest { 
        ResultSet resultSet = mock(ResultSet.class);

        when(resultSet.next()).thenReturn(false);

        when(statement.getResultSet()).thenReturn(resultSet);

        when(connection.getMetaData()).thenReturn(databaseMetaData);

        when(databaseMetaData.getDatabaseProductName()).thenReturn(dataBaseProductName);

        when(result.getConnection()).thenReturn(connection);

        when(connection.getMetaData()).thenReturn(databaseMetaData);

        when(statement.getConnection()).thenReturn(connection);

        when(connection.createStatement()).thenReturn(statement);

        when(statement.executeQuery(ArgumentMatchers.<String>any())).thenReturn(resultSet);

        when(statement.getConnection()).thenReturn(connection);

        when(statement.getConnection().getMetaData().getTables(ArgumentMatchers.<String>any(), ArgumentMatchers.<String>any(),

                ArgumentMatchers.<String>any(), ArgumentMatchers.<String[]>any())).thenReturn(resultSet);

        if ("MySQL".equals(dataBaseProductName)) {

            when(result.getConnection().getMetaData().getURL()).thenReturn("jdbc:mysql://localhost:3306/test");

        } else if ("H2".equals(dataBaseProductName)) {

            when(statement.getConnection().getMetaData().getURL()).thenReturn("jdbc:h2:mem:demo_ds;DB_CLOSE_DELAY=-1;DATABASE_TO_UPPER=false;MODE=MySQL");

        }

        return result;

    }