Add a base class for all TLP oracles

mrigger · mrigger · commit d9fab80e9b52 · 2020-07-16T13:52:27.000+02:00
diff --git a/src/sqlancer/TernaryLogicPartitioningOracleBase.java b/src/sqlancer/TernaryLogicPartitioningOracleBase.java
@@ -0,0 +1,13 @@
+package sqlancer;
+
+/**
+ * This is the base class of the Ternary Logic Partitioning (TLP) oracles. The core idea of TLP is to partition a given
+ * so-called original query to three so-called partitioning queries, each of which computes a partition of the original
+ * query's result.
+ *
+ * @param <E>
+ *            the expression type
+ */
+public abstract class TernaryLogicPartitioningOracleBase<E> {
+
+}
diff --git a/src/sqlancer/clickhouse/oracle/tlp/ClickHouseTLPBase.java b/src/sqlancer/clickhouse/oracle/tlp/ClickHouseTLPBase.java
@@ -2,6 +2,7 @@
 
 import ru.yandex.clickhouse.domain.ClickHouseDataType;
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.clickhouse.ClickHouseErrors;
 import sqlancer.clickhouse.ClickHouseProvider.ClickHouseGlobalState;
@@ -26,7 +27,7 @@
 import java.util.Set;
 import java.util.stream.Collectors;
 
-public class ClickHouseTLPBase implements TestOracle {
+public class ClickHouseTLPBase extends TernaryLogicPartitioningOracleBase<ClickHouseExpression> implements TestOracle {
 
     final ClickHouseGlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/cockroachdb/oracle/tlp/CockroachDBTLPBase.java b/src/sqlancer/cockroachdb/oracle/tlp/CockroachDBTLPBase.java
@@ -8,6 +8,7 @@
 import java.util.stream.Collectors;
 
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.cockroachdb.CockroachDBErrors;
 import sqlancer.cockroachdb.CockroachDBProvider.CockroachDBGlobalState;
@@ -26,7 +27,8 @@
 import sqlancer.cockroachdb.gen.CockroachDBExpressionGenerator;
 import sqlancer.cockroachdb.oracle.CockroachDBNoRECOracle;
 
-public class CockroachDBTLPBase implements TestOracle {
+public class CockroachDBTLPBase extends TernaryLogicPartitioningOracleBase<CockroachDBExpression>
+        implements TestOracle {
 
     final CockroachDBGlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/duckdb/test/DuckDBQueryPartitioningBase.java b/src/sqlancer/duckdb/test/DuckDBQueryPartitioningBase.java
@@ -8,6 +8,7 @@
 import java.util.stream.Collectors;
 
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.ast.newast.ColumnReferenceNode;
 import sqlancer.ast.newast.NewUnaryPostfixOperatorNode;
@@ -27,7 +28,8 @@
 import sqlancer.duckdb.gen.DuckDBExpressionGenerator.DuckDBUnaryPostfixOperator;
 import sqlancer.duckdb.gen.DuckDBExpressionGenerator.DuckDBUnaryPrefixOperator;
 
-public class DuckDBQueryPartitioningBase implements TestOracle {
+public class DuckDBQueryPartitioningBase extends TernaryLogicPartitioningOracleBase<Node<DuckDBExpression>>
+        implements TestOracle {
 
     final DuckDBGlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/mysql/oracle/MySQLQueryPartitioningBase.java b/src/sqlancer/mysql/oracle/MySQLQueryPartitioningBase.java
@@ -7,6 +7,7 @@
 import java.util.Set;
 import java.util.stream.Collectors;
 
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.mysql.MySQLErrors;
 import sqlancer.mysql.MySQLGlobalState;
@@ -22,7 +23,8 @@
 import sqlancer.mysql.ast.MySQLUnaryPrefixOperation.MySQLUnaryPrefixOperator;
 import sqlancer.mysql.gen.MySQLExpressionGenerator;
 
-public abstract class MySQLQueryPartitioningBase implements TestOracle {
+public abstract class MySQLQueryPartitioningBase extends TernaryLogicPartitioningOracleBase<MySQLExpression>
+        implements TestOracle {
 
     final MySQLGlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/postgres/oracle/tlp/PostgresTLPBase.java b/src/sqlancer/postgres/oracle/tlp/PostgresTLPBase.java
@@ -8,6 +8,7 @@
 import java.util.stream.Collectors;
 
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.postgres.PostgresGlobalState;
 import sqlancer.postgres.PostgresSchema;
@@ -27,7 +28,7 @@
 import sqlancer.postgres.gen.PostgresExpressionGenerator;
 import sqlancer.postgres.oracle.PostgresNoRECOracle;
 
-public class PostgresTLPBase implements TestOracle {
+public class PostgresTLPBase extends TernaryLogicPartitioningOracleBase<PostgresExpression> implements TestOracle {
 
     final PostgresGlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/sqlite3/oracle/tlp/SQLite3TLPBase.java b/src/sqlancer/sqlite3/oracle/tlp/SQLite3TLPBase.java
@@ -8,6 +8,7 @@
 import java.util.stream.Collectors;
 
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.sqlite3.SQLite3Errors;
 import sqlancer.sqlite3.SQLite3Provider.SQLite3GlobalState;
@@ -26,7 +27,7 @@
 import sqlancer.sqlite3.schema.SQLite3Schema.SQLite3Table;
 import sqlancer.sqlite3.schema.SQLite3Schema.SQLite3Tables;
 
-public class SQLite3TLPBase implements TestOracle {
+public class SQLite3TLPBase extends TernaryLogicPartitioningOracleBase<SQLite3Expression> implements TestOracle {
 
     final SQLite3GlobalState state;
     final Set<String> errors = new HashSet<>();
diff --git a/src/sqlancer/tidb/oracle/TiDBTLPBase.java b/src/sqlancer/tidb/oracle/TiDBTLPBase.java
@@ -8,6 +8,7 @@
 import java.util.stream.Collectors;
 
 import sqlancer.Randomly;
+import sqlancer.TernaryLogicPartitioningOracleBase;
 import sqlancer.TestOracle;
 import sqlancer.tidb.TiDBErrors;
 import sqlancer.tidb.TiDBExpressionGenerator;
@@ -26,7 +27,7 @@
 import sqlancer.tidb.ast.TiDBUnaryPrefixOperation.TiDBUnaryPrefixOperator;
 import sqlancer.tidb.gen.TiDBHintGenerator;
 
-public abstract class TiDBTLPBase implements TestOracle {
+public abstract class TiDBTLPBase extends TernaryLogicPartitioningOracleBase<TiDBExpression> implements TestOracle {
 
     final TiDBGlobalState state;
     final Set<String> errors = new HashSet<>();
