blob: dd40822038269829f943d3df10ef8a44ccf687db [file] [log] [blame]
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.tajo.engine.function;
import org.apache.commons.lang.StringEscapeUtils;
import org.apache.tajo.catalog.Schema;
import org.apache.tajo.catalog.SchemaFactory;
import org.apache.tajo.datum.NullDatum;
import org.apache.tajo.engine.eval.ExprTestBase;
import org.apache.tajo.exception.TajoException;
import org.junit.Test;
import static org.apache.tajo.common.TajoDataTypes.Type.*;
public class TestStringOperatorsAndFunctions extends ExprTestBase {
@Test
public void testConcatenateOnLiteral() throws TajoException {
testSimpleEval("select ('abc' || 'def') col1 ", new String[]{"abcdef"});
testSimpleEval("select 'abc' || 'def' as col1 ", new String[]{"abcdef"});
testSimpleEval("select 1 || 'def' as col1 ", new String[]{"1def"});
testSimpleEval("select 'abc' || 2 as col1 ", new String[]{"abc2"});
}
@Test
public void testConcatenateOnExpressions() throws TajoException {
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", INT4);
schema.addColumn("col3", FLOAT8);
testSimpleEval("select (1+3) || 2 as col1 ", new String[]{"42"});
testEval(schema, "table1", "abc,2,3.14", "select col1 || col2 || col3 from table1", new String[]{"abc23.14"});
testEval(schema, "table1", "abc,2,3.14", "select col1 || '---' || col3 from table1", new String[]{"abc---3.14"});
}
@Test
public void testFunctionCallIngoreCases() throws TajoException {
testSimpleEval("select ltrim(' trim') ", new String[]{"trim"});
testSimpleEval("select LTRIM(' trim') ", new String[]{"trim"});
testSimpleEval("select lTRim(' trim') ", new String[]{"trim"});
testSimpleEval("select ltrIM(' trim') ", new String[]{"trim"});
}
@Test
public void testLTrim() throws TajoException {
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
testSimpleEval("select ltrim(' trim') ", new String[]{"trim"});
testSimpleEval("select ltrim('xxtrim', 'xx') ", new String[]{"trim"});
testSimpleEval("select trim(leading 'xx' from 'xxtrim') ", new String[]{"trim"});
testSimpleEval("select trim(leading from ' trim') ", new String[]{"trim"});
testSimpleEval("select trim(' trim') ", new String[]{"trim"});
testEval(schema, "table1", " trim,abc", "select ltrim(col1) from table1", new String[]{"trim"});
testEval(schema, "table1", "xxtrim,abc", "select ltrim(col1, 'xx') from table1", new String[]{"trim"});
testEval(schema, "table1", "xxtrim,abc", "select trim(leading 'xx' from col1) from table1", new String[]{"trim"});
testEval(schema, "table1", " trim, abc", "select ltrim(col1) || ltrim(col2) from table1",
new String[]{"trimabc"});
}
@Test
public void testRTrim() throws TajoException {
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
testSimpleEval("select rtrim('trim ') ", new String[]{"trim"});
testSimpleEval("select rtrim('trimxx', 'xx') ", new String[]{"trim"});
testSimpleEval("select trim(trailing 'xx' from 'trimxx') ", new String[]{"trim"});
testSimpleEval("select trim(trailing from 'trim ') ", new String[]{"trim"});
testSimpleEval("select trim('trim ') ", new String[]{"trim"});
testEval(schema, "table1", "trim ,abc", "select rtrim(col1) from table1", new String[]{"trim"});
testEval(schema, "table1", "trimxx,abc", "select rtrim(col1, 'xx') from table1", new String[]{"trim"});
testEval(schema, "table1", "trimxx,abc", "select trim(trailing 'xx' from col1) from table1", new String[]{"trim"});
testEval(schema, "table1", "trim ,abc ", "select rtrim(col1) || rtrim(col2) from table1",
new String[]{"trimabc"});
}
@Test
public void testTrim() throws TajoException {
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
testSimpleEval("select trim(' trim ') ", new String[]{"trim"});
testSimpleEval("select btrim('xxtrimxx', 'xx') ", new String[]{"trim"});
testSimpleEval("select trim(both 'xx' from 'xxtrimxx') ", new String[]{"trim"});
testSimpleEval("select trim(both from ' trim ') ", new String[]{"trim"});
testSimpleEval("select trim(' trim ') ", new String[]{"trim"});
testEval(schema, "table1", " trim ,abc", "select trim(col1) from table1", new String[]{"trim"});
testEval(schema, "table1", "xxtrimxx,abc", "select trim(col1, 'xx') from table1", new String[]{"trim"});
testEval(schema, "table1", "xxtrimxx,abc", "select trim(both 'xx' from col1) from table1", new String[]{"trim"});
testEval(schema, "table1", " trim ,xxabcxx", "select trim(col1) || trim(col2,'xx') from table1",
new String[]{"trimabc"});
}
@Test
public void testRegexReplace() throws TajoException {
testSimpleEval("select regexp_replace('abcdef','bc','--') as col1 ", new String[]{"a--def"});
// null test
testSimpleEval("select regexp_replace(null, 'bc', '--') as col1 ", new String[]{NullDatum.get().toString()});
testSimpleEval("select regexp_replace('abcdef', null, '--') as col1 ", new String[]{NullDatum.get().toString()});
testSimpleEval("select regexp_replace('abcdef','bc', null) as col1 ", new String[]{NullDatum.get().toString()});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
// find matches and replace from column values
testEval(schema, "table1", "------,(^--|--$),ab", "select regexp_replace(col1, col2, col3) as str from table1",
new String[]{"ab--ab"});
// null test from a table
testEval(schema, "table1", ",(^--|--$),ab", "select regexp_replace(col1, col2, col3) as str from table1",
new String[]{NullDatum.get().toString()});
testEval(schema, "table1", "------,(^--|--$),", "select regexp_replace(col1, col2, col3) as str from table1",
new String[]{NullDatum.get().toString()});
}
@Test
public void testLeft() throws TajoException {
testSimpleEval("select left('abcdef',1) as col1 ", new String[]{"a"});
testSimpleEval("select left('abcdef',2) as col1 ", new String[]{"ab"});
testSimpleEval("select left('abcdef',3) as col1 ", new String[]{"abc"});
testSimpleEval("select left('abcdef',4) as col1 ", new String[]{"abcd"});
testSimpleEval("select left('abcdef',5) as col1 ", new String[]{"abcde"});
testSimpleEval("select left('abcdef',6) as col1 ", new String[]{"abcdef"});
testSimpleEval("select left('abcdef',7) as col1 ", new String[]{"abcdef"});
testSimpleEval("select left('abcdef',-1) as col1 ", new String[]{"abcde"});
testSimpleEval("select left('abcdef',-2) as col1 ", new String[]{"abcd"});
testSimpleEval("select left('abcdef',-3) as col1 ", new String[]{"abc"});
testSimpleEval("select left('abcdef',-4) as col1 ", new String[]{"ab"});
testSimpleEval("select left('abcdef',-5) as col1 ", new String[]{"a"});
testSimpleEval("select left('abcdef',-6) as col1 ", new String[]{""});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", INT4);
schema.addColumn("col3", TEXT);
// for null tests
testEval(schema, "table1", ",1,ghi", "select left(col1,1) is null from table1", new String[]{"t"});
testEval(schema, "table1", "abc,,ghi", "select left(col1,col2) is null from table1", new String[]{"t"});
testEval(schema, "table1", "abc,1,ghi", "select left(col1,1) || left(col3,3) from table1", new String[]{"aghi"});
}
@Test
public void testRight() throws TajoException {
testSimpleEval("select right('abcdef',1) as col1 ", new String[]{"f"});
testSimpleEval("select right('abcdef',2) as col1 ", new String[]{"ef"});
testSimpleEval("select right('abcdef',3) as col1 ", new String[]{"def"});
testSimpleEval("select right('abcdef',4) as col1 ", new String[]{"cdef"});
testSimpleEval("select right('abcdef',5) as col1 ", new String[]{"bcdef"});
testSimpleEval("select right('abcdef',6) as col1 ", new String[]{"abcdef"});
testSimpleEval("select right('abcdef',7) as col1 ", new String[]{"abcdef"});
testSimpleEval("select right('abcdef',-1) as col1 ", new String[]{"bcdef"});
testSimpleEval("select right('abcdef',-2) as col1 ", new String[]{"cdef"});
testSimpleEval("select right('abcdef',-3) as col1 ", new String[]{"def"});
testSimpleEval("select right('abcdef',-4) as col1 ", new String[]{"ef"});
testSimpleEval("select right('abcdef',-5) as col1 ", new String[]{"f"});
testSimpleEval("select right('abcdef',-6) as col1 ", new String[]{""});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", INT4);
schema.addColumn("col3", TEXT);
// for null tests
testEval(schema, "table1", ",1,ghi", "select right(col1,1) is null from table1", new String[]{"t"});
testEval(schema, "table1", "abc,,ghi", "select right(col1,col2) is null from table1", new String[]{"t"});
testEval(schema, "table1", "abc,1,ghi", "select right(col1,1) || right(col3,3) from table1", new String[]{"cghi"});
}
@Test
public void testReverse() throws TajoException {
testSimpleEval("select reverse('abcdef') as col1 ", new String[]{"fedcba"});
testSimpleEval("select reverse('가') as col1 ", new String[]{"가"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "abc,efg,3.14", "select reverse(col1) || reverse(col2) from table1",
new String[]{"cbagfe"});
}
@Test
public void testRepeat() throws TajoException {
testSimpleEval("select repeat('ab',4) as col1 ", new String[]{"abababab"});
testSimpleEval("select repeat('가',3) as col1 ", new String[]{"가가가"});
testSimpleEval("select repeat('a',2) as col1 ", new String[]{"aa"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "abc,efg,3.14", "select repeat(col1,2) from table1", new String[]{"abcabc"});
}
@Test
public void testUpper() throws TajoException {
testSimpleEval("select upper('abcdef') as col1 ", new String[]{"ABCDEF"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "abc,efg,3.14", "select upper(col1), upper(col2) from table1",
new String[]{"ABC", "EFG"});
testEval(schema, "table1", "abc,efg,3.14", "select upper(col1) || upper(col2) from table1", new String[]{"ABCEFG"});
}
@Test
public void testLower() throws TajoException {
testSimpleEval("select lower('ABCdEF') as col1 ", new String[]{"abcdef"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABC,DEF,3.14", "select lower(col1), lower(col2) from table1",
new String[]{"abc", "def"});
testEval(schema, "table1", "ABC,DEF,3.14", "select lower(col1) || lower(col2) from table1", new String[]{"abcdef"});
}
@Test
public void testCharLength() throws TajoException {
testSimpleEval("select char_length('123456') as col1 ", new String[]{"6"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABC,DEF,3.14", "select character_length(lower(col1) || lower(col2)) from table1",
new String[]{"6"});
}
@Test
public void testLength() throws TajoException {
testSimpleEval("select length('123456') as col1 ", new String[]{"6"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABC,DEF,3.14", "select length(lower(col1) || lower(col2)) from table1",
new String[]{"6"});
}
@Test
public void testMd5() throws TajoException {
testSimpleEval("select md5('1') as col1 ", new String[]{"c4ca4238a0b923820dcc509a6f75849b"});
testSimpleEval("select md5('tajo') as col1 ", new String[]{"742721b3a79f71a9491681b8e8a7ce85"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "abc,efg,3.14", "select md5(col1) from table1",
new String[]{"900150983cd24fb0d6963f7d28e17f72"});
}
@Test
public void testDigest() throws TajoException {
testSimpleEval("select digest('tajo', 'md2') as col1 ", new String[]{"bf523bce8241982f6bea9af0f7fd37ff"});
testSimpleEval("select digest('tajo', 'md5') as col1 ", new String[]{"742721b3a79f71a9491681b8e8a7ce85"});
testSimpleEval("select digest('tajo', 'sha1') as col1 ", new String[]{"02b0e20540b89f0b735092bbac8093eb2e3804cf"});
testSimpleEval("select digest('tajo', 'sha256') as col1 ",
new String[]{"6440083be076869a9f9d0271a4bf298d98c8aa3ecb49df841895fbcddbb04a70"});
testSimpleEval("select digest('tajo', 'sha384') as col1 ",
new String[]{"59ff99b0e274eb3d8e10f221b6b949bfc1244d2a1226c5c720062fb03d82272be633e4a0f2babccffbfdff7cc1cb06fb"});
testSimpleEval("select digest('tajo', 'sha512') as col1 ",
new String[]{"ee8ba254d331ddfb1bca9aaf0c4b8c58aea5331928cbd20168c87828afb853b0c096af71ec69a23b669217a1dddd2934edaac33b1296fe526b22abd28a15c4b3"});
testSimpleEval("select digest('tajo', 'not') as col1 ", new String[]{NullDatum.get().toString()});
}
@Test
public void testHex() throws TajoException {
testSimpleEval("select to_hex(1) as col1 ", new String[]{"1"});
testSimpleEval("select to_hex(10) as col1 ", new String[]{"a"});
testSimpleEval("select to_hex(1234) as col1 ", new String[]{"4d2"});
testSimpleEval("select to_hex(1023456788888888) as col1 ", new String[]{"3a2d41a583d38"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", ",abcdef,3.14", "select to_hex(10) from table1",
new String[]{"a"});
}
@Test
public void testBin() throws TajoException {
testSimpleEval("select to_bin(1) as col1 ", new String[]{"1"});
testSimpleEval("select to_bin(10) as col1 ", new String[]{"1010"});
testSimpleEval("select to_bin(1234) as col1 ", new String[]{"10011010010"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", ",abcdef,3.14", "select to_bin(20) from table1",
new String[]{"10100"});
}
@Test
public void testOctetLength() throws TajoException {
testSimpleEval("select octet_length('123456') as col1 ", new String[]{"6"});
testSimpleEval("select octet_length('1') as col1 ", new String[]{"1"});
testSimpleEval("select octet_length('가') as col1 ", new String[]{"3"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABC,DEF,3.14", "select octet_length(lower(col1) || lower(col2)) from table1",
new String[]{"6"});
}
@Test
public void testSplitPart() throws TajoException {
testSimpleEval("select split_part('1386577650.123', '.', 1) as col1 ", new String[]{"1386577650"});
testSimpleEval("select split_part('1386577650.123', '.', 2) as col1 ", new String[]{"123"});
// If part is larger than the number of string portions, it will returns NULL.
testSimpleEval("select split_part('1386577650.123', '.', 3) is null", new String[]{"t"});
// null handling tests
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "t1", ",.,1", "select split_part(col1, col2, col3::int) is null from t1", new String[]{"t"});
testEval(schema, "t1", "1386577650.123,,1", "select split_part(col1, col2, col3::int) from t1",
new String[]{"1386577650.123"});
testEval(schema, "t1", "1386577650.123,.,", "select split_part(col1, col2, col3::int) is null from t1",
new String[]{"t"});
}
@Test
public void testSubstr() throws TajoException {
testSimpleEval("select substr('abcdef', 3, 2) as col1 ", new String[]{"cd"});
testSimpleEval("select substr('abcdef', 3) as col1 ", new String[]{"cdef"});
testSimpleEval("select substr('abcdef', 1, 1) as col1 ", new String[]{"a"});
testSimpleEval("select substr('abcdef', 0, 1) as col1 ", new String[]{""});
testSimpleEval("select substr('abcdef', 0, 2) as col1 ", new String[]{"a"});
testSimpleEval("select substr('abcdef', 0) as col1 ", new String[]{"abcdef"});
testSimpleEval("select substr('abcdef', 1, 100) as col1 ", new String[]{"abcdef"});
testSimpleEval("select substr('abcdef', 0, 100) as col1 ", new String[]{"abcdef"});
testSimpleEval("select substr('일이삼사오', 2, 2) as col1 ", new String[]{"이삼"});
testSimpleEval("select substr('일이삼사오', 3) as col1 ", new String[]{"삼사오"});
testSimpleEval("select substr('abcdef', -1) as col1 ", new String[]{"abcdef"});
testSimpleEval("select substr('abcdef', -1, 100) as col1 ", new String[]{"abcdef"});
testSimpleEval("select substr('abcdef', -1, 3) as col1 ", new String[]{"a"});
testSimpleEval("select substr('abcdef', -1, 1) as col1 ", new String[]{""});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", ",abcdef,3.14", "select substr(lower(col2), 2, 3) from table1",
new String[]{"bcd"});
}
@Test
public void testLocate() throws TajoException {
// normal case
testSimpleEval("select locate('abcdef', 'a') as col1 ", new String[]{"1"});
testSimpleEval("select locate('abcdef', 'a', 0) as col1 ", new String[]{"1"});
testSimpleEval("select locate('abcdef', 'a', 1) as col1 ", new String[]{"1"});
testSimpleEval("select locate('abcdef', 'z') as col1 ", new String[]{"0"});
testSimpleEval("select locate('abcdef', 'z', 1) as col1 ", new String[]{"0"});
testSimpleEval("select locate('foobarbar', 'bar') as col1 ", new String[]{"4"});
testSimpleEval("select locate('foobarbar', 'bar', 0) as col1 ", new String[]{"4"});
testSimpleEval("select locate('foobarbar', 'bar', 1) as col1 ", new String[]{"4"});
testSimpleEval("select locate('foobarbar', 'bar', 5) as col1 ", new String[]{"7"});
testSimpleEval("select locate('foobarbar', 'bar', 9) as col1 ", new String[]{"0"});
testSimpleEval("select locate('가나다라마라마', '라마') as col1 ", new String[]{"4"});
testSimpleEval("select locate('가나다라마라마', '라마', 5) as col1 ", new String[]{"6"});
// empty string
testSimpleEval("select locate('abcdef', '') as col1 ", new String[]{"1"});
testSimpleEval("select locate('abcdef', '', 2) as col1 ", new String[]{"2"});
// pos = last index of string (expected value(6) is tested on mysql)
testSimpleEval("select locate('abcdef', '', 6) as col1 ", new String[]{"6"});
// pos = last index + 1 (expected value(7) is tested on mysql)
testSimpleEval("select locate('abcdef', '', 7) as col1 ", new String[]{"7"});
// pos = greater then last index + 1 (expected value(0) is tested on mysql)
testSimpleEval("select locate('abcdef', '', 8) as col1 ", new String[]{"0"});
// pos = greater then last index + 1 (expected value(0) is tested on mysql)
testSimpleEval("select locate('abcdef', '', 9) as col1 ", new String[]{"0"});
testSimpleEval("select locate('가나다라', '', 2) as col1 ", new String[]{"2"});
testSimpleEval("select locate('가나다라', '', 4) as col1 ", new String[]{"4"});
testSimpleEval("select locate('가나다라', '', 5) as col1 ", new String[]{"5"});
testSimpleEval("select locate('가나다라', '', 6) as col1 ", new String[]{"0"});
// negative pos
testSimpleEval("select locate('abcdef', 'a', -1) as col1 ", new String[]{"0"});
testSimpleEval("select locate('abcdef', 'a', -5) as col1 ", new String[]{"0"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", ",abcdef,3.14", "select locate(col2, 'cd') from table1", new String[]{"3"});
testEval(schema, "table1", ",abcdef,3.14", "select locate(col2, 'cd', 1) from table1", new String[]{"3"});
testEval(schema, "table1", ",abcdef,3.14", "select locate(col2, 'cd', 4) from table1", new String[]{"0"});
testEval(schema, "table1", ",abcdef,3.14", "select locate(col2, 'xy') from table1", new String[]{"0"});
// null string
testEval(schema, "table1", ",abcdef,3.14", "select locate(col1, 'cd') is null from table1", new String[]{"t"});
// nul substring
testEval(schema, "table1", ",abcdef,3.14", "select locate('cd', col1) is null from table1", new String[]{"t"});
}
@Test
public void testBitLength() throws TajoException {
testSimpleEval("select bit_length('123456') as col1 ", new String[]{"48"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABC,DEF,3.14", "select bit_length(lower(col1) || lower(col2)) from table1",
new String[]{"48"});
}
@Test
public void testStrpos() throws TajoException {
testSimpleEval("select strpos('tajo','jo') as col1 ", new String[]{"3"});
testSimpleEval("select strpos('tajo','') as col1 ", new String[]{"1"});
testSimpleEval("select strpos('tajo','abcdef') as col1 ", new String[]{"0"});
testSimpleEval("select strpos('일이삼사오육','삼사') as col1 ", new String[]{"3"});
testSimpleEval("select strpos('일이삼사오육','일이삼') as col1 ", new String[]{"1"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABCDEF,HIJKLMN,3.14", "select strpos(lower(col1) || lower(col2), 'fh') from table1",
new String[]{"6"});
}
@Test
public void testStrposb() throws TajoException {
testSimpleEval("select strposb('tajo','jo') as col1 ", new String[]{"3"});
testSimpleEval("select strposb('tajo','') as col1 ", new String[]{"1"});
testSimpleEval("select strposb('tajo','abcdef') as col1 ", new String[]{"0"});
testSimpleEval("select strposb('일이삼사오육','삼사') as col1 ", new String[]{"7"}); //utf8 1 korean word = 3 chars
testSimpleEval("select strposb('일이삼사오육','삼사일') as col1 ", new String[]{"0"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
schema.addColumn("col3", TEXT);
testEval(schema, "table1", "ABCDEF,HIJKLMN,3.14", "select strposb(lower(col1) || lower(col2), 'fh') from table1",
new String[]{"6"});
}
@Test
public void testInitcap() throws TajoException {
testSimpleEval("select initcap('hi bro') ", new String[]{"Hi Bro"});
testSimpleEval("select initcap('HI BRO') ", new String[]{"Hi Bro"});
}
@Test
public void testAscii() throws TajoException {
testSimpleEval("select ascii('abc') as col1 ", new String[]{"97"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
testEval(schema, "table1", "abc", "select ascii(col1) from table1",
new String[]{"97"});
testEval(schema, "table1", "12", "select ascii(col1) from table1",
new String[]{"49"});
}
@Test
public void testChr() throws TajoException {
testSimpleEval("select chr(48) as col1 ", new String[]{"0"});
testSimpleEval("select chr(49) as col1 ", new String[]{"1"});
testSimpleEval("select chr(50) as col1 ", new String[]{"2"});
testSimpleEval("select chr(64) as col1 ", new String[]{"@"});
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", INT4);
testEval(schema, "table1", "65", "select chr(col1) from table1", new String[]{"A"});
testEval(schema, "table1", "66", "select chr(col1) from table1", new String[]{"B"});
testEval(schema, "table1", "52512", "select chr(col1) from table1", new String[]{"촠"});
}
@Test
public void testLpad() throws TajoException {
testSimpleEval("select lpad('hi', 5, 'xy') ", new String[]{"xyxhi"});
testSimpleEval("select LPAD('hello', 7, 'xy') ", new String[]{"xyhello"});
testSimpleEval("select LPAD('hello', 3, 'xy') ", new String[]{"hel"});
testSimpleEval("select lPAD('hello', 7) ", new String[]{" hello"});
testSimpleEval("select lPAD('가나다라', 3) ", new String[]{"가나다"});
}
@Test
public void testRpad() throws TajoException {
testSimpleEval("select rpad('hi', 5, 'xy') ", new String[]{"hixyx"});
testSimpleEval("select RPAD('hello', 7, 'xy') ", new String[]{"helloxy"});
testSimpleEval("select RPAD('hello', 3, 'xy') ", new String[]{"hel"});
testSimpleEval("select rPAD('hello', 7) ", new String[]{"hello "});
testSimpleEval("select rPAD('가나다라', 3) ", new String[]{"가나다"});
}
@Test
public void testQuote_ident() throws TajoException {
testSimpleEval("select quote_ident('Foo bar') ", new String[]{"\"Foo bar\""});
testSimpleEval("select QUOTE_IDENT('Tajo Function') ", new String[]{"\"Tajo Function\""});
}
@Test
public void testEncode() throws TajoException {
testSimpleEval("select encode('Hello\nworld', 'base64') ", new String[]{"SGVsbG8Kd29ybGQ="});
testSimpleEval("select encode('Hello\nworld', 'hex') ",
new String[]{"0x480x650x6c0x6c0x6f0x0a0x770x6f0x720x6c0x64"});
testSimpleEval("select encode('한글', 'base64') ", new String[]{"7ZWc6riA"});
testSimpleEval("select encode('한글', 'hex') ", new String[]{"0xd55c0xae00"});
testSimpleEval("select encode('한글\n테스트\t입니다.', 'hex') ",
new String[]{"0xd55c0xae000x0a0xd14c0xc2a40xd2b80x090xc7850xb2c80xb2e40x2e"});
}
@Test
public void testDecode() throws TajoException {
testSimpleEval("select decode('SGVsbG8Kd29ybGQ=', 'base64') ",
new String[]{StringEscapeUtils.escapeJava("Hello\nworld")});
testSimpleEval("select decode('0x480x650x6c0x6c0x6f0x0a0x770x6f0x720x6c0x64', 'hex') ",
new String[]{StringEscapeUtils.escapeJava("Hello\nworld")});
testSimpleEval("select decode('7ZWc6riA', 'base64') ", new String[]{StringEscapeUtils.escapeJava("한글")});
testSimpleEval("select decode('0xd55c0xae00', 'hex') ", new String[]{StringEscapeUtils.escapeJava("한글")});
testSimpleEval("select decode('0xd55c0xae000x0a0xd14c0xc2a40xd2b80x090xc7850xb2c80xb2e40x2e', 'hex') ",
new String[]{StringEscapeUtils.escapeJava("한글\n" + "테스트\t입니다.")});
}
@Test
public void testFindInSet() throws TajoException {
// abnormal cases
testSimpleEval("select find_in_set('cr','crt') as col1 ", new String[]{"0"}); // there is no matched string
testSimpleEval("select find_in_set('c,r','crt,c,cr,c,def') as col1 ", new String[]{"0"}); // abnormal parameter
// normal cases
testSimpleEval("select find_in_set('crt','crt,c,cr,d,def') as col1 ", new String[]{"1"});
testSimpleEval("select find_in_set('c','crt,c,cr,d,def') as col1 ", new String[]{"2"});
testSimpleEval("select find_in_set('def','crt,c,cr,d,def') as col1 ", new String[]{"5"});
// unicode test
testSimpleEval("select find_in_set('딸기','사과,배,옥수수,감자,딸기,수박') as col1 ", new String[]{"5"});
// null test
Schema schema = SchemaFactory.newV1();
schema.addColumn("col1", TEXT);
schema.addColumn("col2", TEXT);
testEval(schema, "table1", "|crt,c,cr,c,def", "select find_in_set(col1, col2) is null from table1",
new String[]{"t"}, '|', true);
testEval(schema, "table1", "cr|", "select find_in_set(col1, col2) is null from table1",
new String[]{"t"}, '|', true);
}
@Test
public void testConcat() throws TajoException {
testSimpleEval("select concat('333', '22') ", new String[]{"33322"});
testSimpleEval("select concat('한글', '22') ", new String[]{"한글22"});
testSimpleEval("select concat(null, '22') ", new String[]{"22"});
testSimpleEval("select concat('22',null) ", new String[]{"22"});
testSimpleEval("select concat('22',null, '33', null, '44') ", new String[]{"223344"});
}
@Test
public void testConcat_ws() throws TajoException {
testSimpleEval("select concat_ws(',', '333', '22') ", new String[]{"333,22"});
testSimpleEval("select concat_ws(',', '한글', '22') ", new String[]{"한글,22"});
testSimpleEval("select concat_ws(',', '22', null) ", new String[]{"22"});
testSimpleEval("select concat_ws(',', '22', '33', '33') ", new String[]{"22,33,33"});
testSimpleEval("select concat_ws(',', null, '22') ", new String[]{"22"});
}
}