|
1 | 1 | package sqlancer.cockroachdb; |
2 | 2 |
|
3 | 3 | import java.sql.SQLException; |
| 4 | +import java.util.ArrayList; |
| 5 | +import java.util.List; |
4 | 6 |
|
5 | 7 | import com.beust.jcommander.Parameter; |
6 | 8 | import com.beust.jcommander.Parameters; |
7 | 9 |
|
| 10 | +import sqlancer.CompositeTestOracle; |
8 | 11 | import sqlancer.TestOracle; |
9 | 12 | import sqlancer.cockroachdb.CockroachDBProvider.CockroachDBGlobalState; |
10 | 13 | import sqlancer.cockroachdb.oracle.CockroachDBNoRECOracle; |
|
14 | 17 | import sqlancer.cockroachdb.oracle.tlp.CockroachDBTLPGroupByOracle; |
15 | 18 | import sqlancer.cockroachdb.oracle.tlp.CockroachDBTLPHavingOracle; |
16 | 19 | import sqlancer.cockroachdb.oracle.tlp.CockroachDBTLPJoinOracle; |
17 | | -import sqlancer.cockroachdb.oracle.tlp.CockroachDBTLPOracle; |
18 | 20 | import sqlancer.cockroachdb.oracle.tlp.CockroachDBTLPWhereOracle; |
19 | 21 |
|
20 | 22 | @Parameters(separators = "=", commandDescription = "Test CockroachDB") |
@@ -77,7 +79,14 @@ public TestOracle create(CockroachDBGlobalState globalState) throws SQLException |
77 | 79 | QUERY_PARTITIONING { |
78 | 80 | @Override |
79 | 81 | public TestOracle create(CockroachDBGlobalState globalState) throws SQLException { |
80 | | - return new CockroachDBTLPOracle(globalState); |
| 82 | + List<TestOracle> oracles = new ArrayList<>(); |
| 83 | + oracles.add(new CockroachDBTLPAggregateOracle(globalState)); |
| 84 | + oracles.add(new CockroachDBTLPHavingOracle(globalState)); |
| 85 | + oracles.add(new CockroachDBTLPWhereOracle(globalState)); |
| 86 | + oracles.add(new CockroachDBTLPGroupByOracle(globalState)); |
| 87 | + oracles.add(new CockroachDBTLPExtendedWhereOracle(globalState)); |
| 88 | + oracles.add(new CockroachDBTLPDistinctOracle(globalState)); |
| 89 | + return new CompositeTestOracle(oracles); |
81 | 90 | } |
82 | 91 | }; |
83 | 92 |
|
|
0 commit comments