summaryrefslogtreecommitdiffstats
path: root/vespa-hadoop/src/test/java/com/yahoo/vespa/hadoop/util/TupleToolsTest.java
diff options
context:
space:
mode:
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.java50
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);
+ }
+
+}