[Postgres] Implement the expected value visitor for LIKE

mrigger · mrigger · commit f642b8a4057e · 2020-08-25T21:47:03.000+02:00
diff --git a/src/sqlancer/postgres/PostgresExpectedValueVisitor.java b/src/sqlancer/postgres/PostgresExpectedValueVisitor.java
@@ -10,6 +10,7 @@
 import sqlancer.postgres.ast.PostgresExpression;
 import sqlancer.postgres.ast.PostgresFunction;
 import sqlancer.postgres.ast.PostgresInOperation;
+import sqlancer.postgres.ast.PostgresLikeOperation;
 import sqlancer.postgres.ast.PostgresOrderByTerm;
 import sqlancer.postgres.ast.PostgresPOSIXRegularExpression;
 import sqlancer.postgres.ast.PostgresPostfixOperation;
@@ -169,4 +170,11 @@ public void visit(PostgresBinaryLogicalOperation op) {
         visit(op.getRight());
     }
 
+    @Override
+    public void visit(PostgresLikeOperation op) {
+        print(op);
+        visit(op.getLeft());
+        visit(op.getRight());
+    }
+
 }
diff --git a/src/sqlancer/postgres/PostgresToStringVisitor.java b/src/sqlancer/postgres/PostgresToStringVisitor.java
@@ -3,6 +3,7 @@
 import java.util.Optional;
 
 import sqlancer.Randomly;
+import sqlancer.common.visitor.BinaryOperation;
 import sqlancer.common.visitor.ToStringVisitor;
 import sqlancer.postgres.PostgresSchema.PostgresDataType;
 import sqlancer.postgres.ast.PostgresAggregate;
@@ -17,6 +18,7 @@
 import sqlancer.postgres.ast.PostgresInOperation;
 import sqlancer.postgres.ast.PostgresJoin;
 import sqlancer.postgres.ast.PostgresJoin.PostgresJoinType;
+import sqlancer.postgres.ast.PostgresLikeOperation;
 import sqlancer.postgres.ast.PostgresOrderByTerm;
 import sqlancer.postgres.ast.PostgresPOSIXRegularExpression;
 import sqlancer.postgres.ast.PostgresPostfixOperation;
@@ -340,13 +342,12 @@ public void visit(PostgresCollate op) {
 
     @Override
     public void visit(PostgresBinaryLogicalOperation op) {
-        sb.append("(");
-        visit(op.getLeft());
-        sb.append(")");
-        sb.append(op.getOperatorRepresentation());
-        sb.append("(");
-        visit(op.getRight());
-        sb.append(")");
+        super.visit((BinaryOperation<PostgresExpression>) op);
+    }
+
+    @Override
+    public void visit(PostgresLikeOperation op) {
+        super.visit((BinaryOperation<PostgresExpression>) op);
     }
 
 }
diff --git a/src/sqlancer/postgres/PostgresVisitor.java b/src/sqlancer/postgres/PostgresVisitor.java
@@ -14,6 +14,7 @@
 import sqlancer.postgres.ast.PostgresExpression;
 import sqlancer.postgres.ast.PostgresFunction;
 import sqlancer.postgres.ast.PostgresInOperation;
+import sqlancer.postgres.ast.PostgresLikeOperation;
 import sqlancer.postgres.ast.PostgresOrderByTerm;
 import sqlancer.postgres.ast.PostgresPOSIXRegularExpression;
 import sqlancer.postgres.ast.PostgresPostfixOperation;
@@ -63,6 +64,8 @@ public interface PostgresVisitor {
 
     void visit(PostgresBinaryLogicalOperation op);
 
+    void visit(PostgresLikeOperation op);
+
     default void visit(PostgresExpression expression) {
         if (expression instanceof PostgresConstant) {
             visit((PostgresConstant) expression);
@@ -98,8 +101,8 @@ default void visit(PostgresExpression expression) {
             visit((PostgresFromTable) expression);
         } else if (expression instanceof PostgresSubquery) {
             visit((PostgresSubquery) expression);
-        } else if (expression instanceof PostgresBinaryLogicalOperation) {
-            visit((PostgresBinaryLogicalOperation) expression);
+        } else if (expression instanceof PostgresLikeOperation) {
+            visit((PostgresLikeOperation) expression);
         } else {
             throw new AssertionError(expression);
         }
