forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDataFusionOptions.java
More file actions
43 lines (36 loc) · 1.63 KB
/
DataFusionOptions.java
File metadata and controls
43 lines (36 loc) · 1.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package sqlancer.datafusion;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.List;
import com.beust.jcommander.Parameter;
import com.beust.jcommander.Parameters;
import sqlancer.DBMSSpecificOptions;
import sqlancer.OracleFactory;
import sqlancer.common.oracle.TestOracle;
import sqlancer.datafusion.DataFusionOptions.DataFusionOracleFactory;
import sqlancer.datafusion.DataFusionProvider.DataFusionGlobalState;
import sqlancer.datafusion.test.DataFusionNoRECOracle;
import sqlancer.datafusion.test.DataFusionQueryPartitioningWhereTester;
@Parameters(commandDescription = "DataFusion")
public class DataFusionOptions implements DBMSSpecificOptions<DataFusionOracleFactory> {
@Parameter(names = "--debug-info", description = "Show debug messages related to DataFusion", arity = 0)
public boolean showDebugInfo;
@Override
public List<DataFusionOracleFactory> getTestOracleFactory() {
return Arrays.asList(DataFusionOracleFactory.NOREC, DataFusionOracleFactory.QUERY_PARTITIONING_WHERE);
}
public enum DataFusionOracleFactory implements OracleFactory<DataFusionGlobalState> {
NOREC {
@Override
public TestOracle<DataFusionGlobalState> create(DataFusionGlobalState globalState) throws SQLException {
return new DataFusionNoRECOracle(globalState);
}
},
QUERY_PARTITIONING_WHERE {
@Override
public TestOracle<DataFusionGlobalState> create(DataFusionGlobalState globalState) throws SQLException {
return new DataFusionQueryPartitioningWhereTester(globalState);
}
}
}
}