diff options
Diffstat (limited to 'vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java')
-rw-r--r-- | vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java b/vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java new file mode 100644 index 00000000000..c98e7b1c02c --- /dev/null +++ b/vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java @@ -0,0 +1,50 @@ +package com.yahoo.vespa.hadoop.util; + +import com.yahoo.vespa.hadoop.mapreduce.util.TupleTools; +import org.apache.pig.data.DataType; +import org.apache.pig.data.Tuple; +import org.apache.pig.data.TupleFactory; +import org.apache.pig.impl.logicalLayer.schema.Schema; +import org.junit.Test; + +import java.io.IOException; + +import static org.junit.Assert.assertEquals; + +public class TupleToolsTest { + + @Test + public void requireThatTupleToStringHandlesSimpleTypes() throws IOException { + Schema schema = new Schema(); + Tuple tuple = TupleFactory.getInstance().newTuple(); + + addToTuple("id", DataType.CHARARRAY, "123", schema, tuple); + addToTuple("rank", DataType.INTEGER, 1, schema, tuple); + + String template = "Id is <id> and rank is <rank>"; + String result = TupleTools.toString(schema, tuple, template); + + assertEquals("Id is 123 and rank is 1", result); + } + + + private void addToTuple(String alias, byte type, Object value, Schema schema, Tuple tuple) { + schema.add(new Schema.FieldSchema(alias, type)); + tuple.append(value); + } + + @Test + public void requireThatTupleToStringHandlesStringCharacters() throws IOException { + Schema schema = new Schema(); + Tuple tuple = TupleFactory.getInstance().newTuple(); + + addToTuple("id", DataType.CHARARRAY, "_!@#$%^&*()", schema, tuple); + addToTuple("rank", DataType.INTEGER, 1, schema, tuple); + + String template = "Id is <id> and rank is <rank>"; + String result = TupleTools.toString(schema, tuple, template); + + assertEquals("Id is _!@#$%^&*() and rank is 1", result); + } + +} |