Commit 3d61209e authored by caohao's avatar caohao
Browse files

for #601, add index metadata assertion support

parent 82b7e1ad

sharding-jdbc/src/test/java/io/shardingsphere/dbtest/cases/dataset/metadata/DataSetIndex.java

+2 −0
Original line number Diff line number Diff line
@@ -18,12 +18,14 @@ 
package io.shardingsphere.dbtest.cases.dataset.metadata;



import lombok.Getter;

import lombok.Setter;



import javax.xml.bind.annotation.XmlAccessType;

import javax.xml.bind.annotation.XmlAccessorType;

import javax.xml.bind.annotation.XmlAttribute;



@Getter

@Setter

@XmlAccessorType(XmlAccessType.FIELD)

public final class DataSetIndex {

sharding-jdbc/src/test/java/io/shardingsphere/dbtest/engine/ddl/BaseDDLIntegrateTest.java

+34 −6
Original line number Diff line number Diff line
@@ -21,6 +21,7 @@ import io.shardingsphere.core.constant.DatabaseType; 
import io.shardingsphere.dbtest.cases.assertion.ddl.DDLIntegrateTestCaseAssertion;

import io.shardingsphere.dbtest.cases.dataset.DataSet;

import io.shardingsphere.dbtest.cases.dataset.metadata.DataSetColumn;

import io.shardingsphere.dbtest.cases.dataset.metadata.DataSetIndex;

import io.shardingsphere.dbtest.cases.dataset.metadata.DataSetMetadata;

import io.shardingsphere.dbtest.engine.SingleIntegrateTest;

import io.shardingsphere.dbtest.env.DatabaseTypeEnvironment;
@@ -77,16 +78,20 @@ public abstract class BaseDDLIntegrateTest extends SingleIntegrateTest { 
        }

        String tableName = assertion.getTable();

        List<DataSetColumn> actualColumns = getActualColumns(connection, tableName);

        assertMetadata(actualColumns, expected.findMetadata(tableName));

        List<DataSetIndex> actualIndexes = getActualIndexes(connection, tableName);

        assertMetadata(actualColumns, actualIndexes, expected.findMetadata(tableName));

    }

    

    private void assertMetadata(final List<DataSetColumn> actual, final DataSetMetadata expected) {

    private void assertMetadata(final List<DataSetColumn> actualColumns, List<DataSetIndex> actualIndexes, final DataSetMetadata expected) {

        for (DataSetColumn each : expected.getColumns()) {

            assertMetadata(actual, each);

            assertColumnMetadata(actualColumns, each);

        }

        for (DataSetIndex each : expected.getIndexes()) {

            assertIndexMetadata(actualIndexes, each);

        }

    }

    

    private void assertMetadata(final List<DataSetColumn> actual, final DataSetColumn expect) {

    private void assertColumnMetadata(final List<DataSetColumn> actual, final DataSetColumn expect) {

        for (DataSetColumn each : actual) {

            if (expect.getName().equals(each.getName())) {

                if (DatabaseType.MySQL == databaseType && "integer".equals(expect.getType())) {
@@ -100,6 +105,14 @@ public abstract class BaseDDLIntegrateTest extends SingleIntegrateTest { 
        }

    }



    private void assertIndexMetadata(final List<DataSetIndex> actual, final DataSetIndex expect) {

        for (DataSetIndex each : actual) {

            if (expect.getName().equals(each.getName())) {

                assertThat(each.isUnique(), is(expect.isUnique()));

            }

        }

    }



    private List<DataSetColumn> getActualColumns(final Connection connection, final String tableName) throws SQLException {

        DatabaseMetaData metaData = connection.getMetaData();

        boolean isTableExisted = metaData.getTables(null, null, tableName, new String[] {"TABLE"}).next();
@@ -118,6 +131,21 @@ public abstract class BaseDDLIntegrateTest extends SingleIntegrateTest { 
        }

    }



    private List<DataSetIndex> getActualIndexes(final Connection connection, final String tableName) throws SQLException {

        DatabaseMetaData metaData = connection.getMetaData();

        try (ResultSet resultSet = metaData.getIndexInfo(null, null, tableName, false, false)) {

            List<DataSetIndex> result = new LinkedList<>();

            while (resultSet.next()) {

                DataSetIndex each = new DataSetIndex();

                each.setName(resultSet.getString("INDEX_NAME"));

                each.setUnique(!resultSet.getBoolean("NON_UNIQUE"));

                each.setColumns(resultSet.getString("COLUMN_NAME"));

                result.add(each);

            }

            return result;

        }

    }



    protected void dropTableIfExisted(final Connection connection) {

        try (PreparedStatement preparedStatement = connection.prepareStatement(String.format("DROP TABLE %s", assertion.getTable()))) {

            preparedStatement.executeUpdate();