src/sqlancer/mongodb/MongoDBComparatorHelper.java - master - NikolayIvkin/sqlancer2

src/sqlancer/mongodb/MongoDBComparatorHelper.java
97 строк4 KB
Patrick
Clean up and improve code for pull request
28 мар 2021, 18:06
28 мар 2021, 18:06e0b272d
package sqlancer.mongodb;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.bson.Document;

import sqlancer.IgnoreMeException;
import sqlancer.Main;
import sqlancer.common.query.ExpectedErrors;
import sqlancer.mongodb.MongoDBProvider.MongoDBGlobalState;
import sqlancer.mongodb.query.MongoDBSelectQuery;

public final class MongoDBComparatorHelper {

    private MongoDBComparatorHelper() {
    }

    public static List<Document> getResultSetAsDocumentList(MongoDBSelectQuery adapter, MongoDBGlobalState state)
            throws Exception {
        ExpectedErrors errors = adapter.getExpectedErrors();
        List<Document> result;
        try {
            adapter.executeAndGet(state);
            Main.nrSuccessfulActions.addAndGet(1);
            result = adapter.getResultSet();
            return result;
        } catch (Exception e) {
            if (e instanceof IgnoreMeException) {
                throw e;
            }
            Main.nrUnsuccessfulActions.addAndGet(1);
            if (e.getMessage() == null) {
                throw new AssertionError(adapter.getLogString(), e);
            }
            if (errors.errorIsExpected(e.getMessage())) {
                throw new IgnoreMeException();
            }
            throw new AssertionError(adapter.getLogString(), e);
        }
    }

    public static void assumeCountIsEqual(List<Document> resultSet, List<Document> secondResultSet,
            MongoDBSelectQuery originalQuery) {
        int originalSize = resultSet.size();
        if (secondResultSet.isEmpty()) {
            if (originalSize == 0) {
                return;
            } else {
                String assertMessage = String.format("The Count of the result set mismatches!\n %s",
                        originalQuery.getLogString());
                throw new AssertionError(assertMessage);
            }
        }
        if (secondResultSet.size() != 1) {
            throw new AssertionError(
                    String.format("Count query result bigger than one \n %s", originalQuery.getLogString()));
        }
        int withCount = (int) secondResultSet.get(0).get("count");
        if (originalSize != withCount) {
            String assertMessage = String.format("The Count of the result set mismatches!\n %s",
                    originalQuery.getLogString());
            throw new AssertionError(assertMessage);
        }
    }

    public static void assumeResultSetsAreEqual(List<Document> resultSet, List<Document> secondResultSet,
            MongoDBSelectQuery originalQuery) {
        if (resultSet.size() != secondResultSet.size()) {
            String assertionMessage = String.format("The Size of the result sets mismatch (%d and %d)!\n%s",
                    resultSet.size(), secondResultSet.size(), originalQuery.getLogString());
            throw new AssertionError(assertionMessage);
        }

        Set<Document> firstHashSet = new HashSet<>(resultSet);
        Set<Document> secondHashSet = new HashSet<>(secondResultSet);

        if (!firstHashSet.equals(secondHashSet)) {
            Set<Document> firstResultSetMisses = new HashSet<>(firstHashSet);
            firstResultSetMisses.removeAll(secondHashSet);
            Set<Document> secondResultSetMisses = new HashSet<>(secondHashSet);
            secondResultSetMisses.removeAll(firstHashSet);
            StringBuilder firstMisses = new StringBuilder();
            for (Document document : firstResultSetMisses) {
                firstMisses.append(document.toJson()).append(" ");
            }
            StringBuilder secondMisses = new StringBuilder();
            for (Document document : secondResultSetMisses) {
                secondMisses.append(document.toJson()).append(" ");
            }
            String assertMessage = String.format("The Content of the result sets mismatch!\n %s \n %s\n %s",
                    firstMisses.toString(), secondMisses.toString(), originalQuery.getLogString());
            throw new AssertionError(assertMessage);
        }
    }
}