src/sqlancer/databend/test/DatabendQueryPartitioningDistinctTester.java - master - NikolayIvkin/sqlancer2

src/sqlancer/databend/test/DatabendQueryPartitioningDistinctTester.java

44 строки2 KB

Yisong Han

fix(databend): mvn verify succeed

09 сен 2022, 19:09

09 сен 2022, 19:095f07a2d

package sqlancer.databend.test;

import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import sqlancer.ComparatorHelper;
import sqlancer.Randomly;
import sqlancer.databend.DatabendErrors;
import sqlancer.databend.DatabendProvider.DatabendGlobalState;
import sqlancer.databend.DatabendToStringVisitor;

public class DatabendQueryPartitioningDistinctTester extends DatabendQueryPartitioningBase {

    public DatabendQueryPartitioningDistinctTester(DatabendGlobalState state) {
        super(state);
        DatabendErrors.addGroupByErrors(errors);
    }

    @Override
    public void check() throws SQLException {
        super.check();
        select.setDistinct(true);
        select.setWhereClause(null);
        String originalQueryString = DatabendToStringVisitor.asString(select);

        List<String> resultSet = ComparatorHelper.getResultSetFirstColumnAsString(originalQueryString, errors, state);
        if (Randomly.getBoolean()) {
            select.setDistinct(false);
        }
        select.setWhereClause(predicate);
        String firstQueryString = DatabendToStringVisitor.asString(select);
        select.setWhereClause(negatedPredicate);
        String secondQueryString = DatabendToStringVisitor.asString(select);
        select.setWhereClause(isNullPredicate);
        String thirdQueryString = DatabendToStringVisitor.asString(select);
        List<String> combinedString = new ArrayList<>();
        List<String> secondResultSet = ComparatorHelper.getCombinedResultSetNoDuplicates(firstQueryString,
                secondQueryString, thirdQueryString, combinedString, true, state, errors);
        ComparatorHelper.assumeResultSetsAreEqual(resultSet, secondResultSet, originalQueryString, combinedString,
                state, DatabendQueryPartitioningBase::canonicalizeResultValue);
    }

}