/*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.facebook.presto.sql.planner.assertions;
import com.facebook.presto.sql.parser.SqlParser;
import com.facebook.presto.sql.tree.Expression;
import com.facebook.presto.sql.tree.SymbolReference;
import org.testng.annotations.Test;
import static com.facebook.presto.sql.ExpressionUtils.rewriteIdentifiersToSymbolReferences;
import static org.testng.Assert.assertTrue;
import static org.testng.AssertJUnit.assertFalse;
public class TestExpressionVerifier
{
private final SqlParser parser = new SqlParser();
@Test
public void test()
{
Expression actual = expression("NOT(orderkey = 3 AND custkey = 3 AND orderkey < 10)");
SymbolAliases symbolAliases = SymbolAliases.builder()
.put("X", new SymbolReference("orderkey"))
.put("Y", new SymbolReference("custkey"))
.build();
ExpressionVerifier verifier = new ExpressionVerifier(symbolAliases);
assertTrue(verifier.process(actual, expression("NOT(X = 3 AND Y = 3 AND X < 10)")));
assertThrows(() -> verifier.process(actual, expression("NOT(X = 3 AND Y = 3 AND Z < 10)")));
assertFalse(verifier.process(actual, expression("NOT(X = 3 AND X = 3 AND X < 10)")));
}
@Test
public void testCast()
throws Exception
{
SymbolAliases aliases = SymbolAliases.builder()
.put("X", new SymbolReference("orderkey"))
.build();
ExpressionVerifier verifier = new ExpressionVerifier(aliases);
assertTrue(verifier.process(expression("CAST('2' AS varchar)"), expression("CAST('2' AS varchar)")));
assertFalse(verifier.process(expression("CAST('2' AS varchar)"), expression("CAST('2' AS bigint)")));
assertTrue(verifier.process(expression("CAST(orderkey AS varchar)"), expression("CAST(X AS varchar)")));
}
@Test
public void testBetween()
throws Exception
{
SymbolAliases symbolAliases = SymbolAliases.builder()
.put("X", new SymbolReference("orderkey"))
.put("Y", new SymbolReference("custkey"))
.build();
ExpressionVerifier verifier = new ExpressionVerifier(symbolAliases);
// Complete match
assertTrue(verifier.process(expression("orderkey BETWEEN 1 AND 2"), expression("X BETWEEN 1 AND 2")));
// Different value
assertFalse(verifier.process(expression("orderkey BETWEEN 1 AND 2"), expression("Y BETWEEN 1 AND 2")));
assertFalse(verifier.process(expression("custkey BETWEEN 1 AND 2"), expression("X BETWEEN 1 AND 2")));
// Different min or max
assertFalse(verifier.process(expression("orderkey BETWEEN 2 AND 4"), expression("X BETWEEN 1 AND 2")));
assertFalse(verifier.process(expression("orderkey BETWEEN 1 AND 2"), expression("X BETWEEN '1' AND '2'")));
assertFalse(verifier.process(expression("orderkey BETWEEN 1 AND 2"), expression("X BETWEEN 4 AND 7")));
}
private Expression expression(String sql)
{
return rewriteIdentifiersToSymbolReferences(parser.createExpression(sql));
}
private static void assertThrows(Runnable runnable)
{
try {
runnable.run();
throw new AssertionError("Method didn't throw exception as expected");
}
catch (Exception expected) {
}
}
}