src/sqlancer/cockroachdb/gen/CockroachDBShowGenerator.java - master - NikolayIvkin/sqlancer2

src/sqlancer/cockroachdb/gen/CockroachDBShowGenerator.java

62 строки2 KB

Patrick

Continuation of Reworking Connection to DB to support non-SQL connections

26 ноя 2020, 15:40

26 ноя 2020, 15:4062e302e

package sqlancer.cockroachdb.gen;

import sqlancer.Randomly;
import sqlancer.cockroachdb.CockroachDBErrors;
import sqlancer.cockroachdb.CockroachDBProvider.CockroachDBGlobalState;
import sqlancer.common.query.ExpectedErrors;
import sqlancer.common.query.SQLQueryAdapter;

public final class CockroachDBShowGenerator {

    private CockroachDBShowGenerator() {
    }

    private enum Option {
        EXPERIMENTAL_FINGERPRINTS, // https://github.com/cockroachdb/cockroach/issues/44237
        DATABASES, JOBS, RANGES, LOCALITY, SEQUENCES, TRACE_FOR_SESSION
    }

    public static SQLQueryAdapter show(CockroachDBGlobalState globalState) {
        ExpectedErrors errors = new ExpectedErrors();
        StringBuilder sb = new StringBuilder();
        switch (Randomly.fromOptions(Option.values())) {
        case EXPERIMENTAL_FINGERPRINTS:
            sb.append("SHOW EXPERIMENTAL_FINGERPRINTS FROM TABLE ");
            sb.append(globalState.getSchema().getRandomTable(t -> !t.isView()).getName());
            errors.add("as type bytes: bytea encoded value ends with incomplete escape sequence");
            errors.add("invalid bytea escape sequence");
            break;
        case DATABASES:
            sb.append("SHOW DATABASES");
            break;
        case JOBS:
            sb.append("SHOW JOBS");
            break;
        case RANGES:
            sb.append("SHOW RANGES FROM TABLE ");
            sb.append(globalState.getSchema().getRandomTable(t -> !t.isView()).getName());
            break;
        case LOCALITY:
            sb.append("SHOW LOCALITY");
            break;
        case SEQUENCES:
            sb.append("SHOW SEQUENCES");
            break;
        case TRACE_FOR_SESSION:
            sb.append("SHOW ");
            if (Randomly.getBoolean()) {
                sb.append("COMPACT ");
            }
            if (Randomly.getBoolean()) {
                sb.append("KV ");
            }
            sb.append("TRACE FOR SESSION;");
            break;
        default:
            throw new AssertionError();
        }
        CockroachDBErrors.addTransactionErrors(errors);
        return new SQLQueryAdapter(sb.toString(), errors);
    }

}