Migrate TestCustomFunctions to AbstractTestFunctions
*/
package com.facebook.presto.operator.scalar;
-import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import static com.facebook.presto.spi.type.BigintType.BIGINT;
import static com.facebook.presto.spi.type.BooleanType.BOOLEAN;
-import static io.airlift.testing.Closeables.closeAllRuntimeException;
public class TestCustomFunctions
+ extends AbstractTestFunctions
{
- private FunctionAssertions functionAssertions;
-
@BeforeClass
public void setupClass()
{
- functionAssertions = new FunctionAssertions().addScalarFunctions(CustomFunctions.class);
- }
-
- @AfterClass(alwaysRun = true)
- public void tearDown()
- {
- closeAllRuntimeException(functionAssertions);
- functionAssertions = null;
+ registerScalar(CustomFunctions.class);
}
@Test
public void testCustomAdd()
{
- functionAssertions.assertFunction("custom_add(123, 456)", BIGINT, 579L);
+ assertFunction("custom_add(123, 456)", BIGINT, 579L);
}
@Test
public void testSliceIsNull()
{
- functionAssertions.assertFunction("custom_is_null(CAST(NULL AS VARCHAR))", BOOLEAN, true);
- functionAssertions.assertFunction("custom_is_null('not null')", BOOLEAN, false);
+ assertFunction("custom_is_null(CAST(NULL AS VARCHAR))", BOOLEAN, true);
+ assertFunction("custom_is_null('not null')", BOOLEAN, false);
}
@Test
public void testLongIsNull()
{
- functionAssertions.assertFunction("custom_is_null(CAST(NULL AS BIGINT))", BOOLEAN, true);
- functionAssertions.assertFunction("custom_is_null(0)", BOOLEAN, false);
+ assertFunction("custom_is_null(CAST(NULL AS BIGINT))", BOOLEAN, true);
+ assertFunction("custom_is_null(0)", BOOLEAN, false);
}
}
