[PIG-2650] Convenience mock Loader and Storer to simplify unit testing of Pig scripts

• Type: New Feature
• Status: Closed
• Priority: Major
• Resolution: Fixed
• Fix Version/s: 0.11, 0.10.1
• Patch Info: Patch Available
• Assignee: Julien Le Dem
• Reporter: Julien Le Dem
• Created: 13/Apr/12 00:07
• Updated: 06/Jan/13 23:57
• Resolved: 26/Apr/12 20:30

Description

A test would look as follows:

PigServer pigServer = new PigServer(ExecType.LOCAL);
TupleFactory tf = TupleFactory.getInstance();
Data data = Storage.resetData(pigServer.getPigContext());
data.set("foo", Arrays.asList(
    tf.newTuple("a"),
    tf.newTuple("b"),
    tf.newTuple("c")
    ));

pigServer.registerQuery("A = LOAD 'foo' USING mock.Storage();");
// some complex script to test
pigServer.registerQuery("STORE A INTO 'bar' USING mock.Storage();");

Iterator<Tuple> out = data.get("bar").iterator();
assertEquals("a", out.next().get(0));
assertEquals("b", out.next().get(0));
assertEquals("c", out.next().get(0));