Equivalent of xUnit for testing Pig.
Call {@link PigTest#getCluster()} then construct a test and call an assert method.Have a look to the test of this class for more example.
218219220221222223224225226
public void testArgFiles() throws ParseException, IOException { String[] argsFile = { "test/data/pigunit/top_queries_params.txt" }; test = new PigTest(PIG_SCRIPT, null, argsFile); test.assertOutput(new File("test/data/pigunit/top_queries_expected_top_3.txt")); }
234235236237238239240241242243244
"queries_ordered = ORDER queries_sum BY count DESC PARALLEL 1;", "queries_limit = LIMIT queries_ordered 3;", "STORE queries_limit INTO 'top_3_queries';", }; test = new PigTest(script); String expected = "(yahoo,25)\n" + "(facebook,15)\n" + "(twitter,7)";
256257258259260261262263264265266
"queries_ordered = ORDER queries BY query DESC PARALLEL 1;", "queries_limit = LIMIT queries_ordered 3;", "STORE queries_limit INTO 'top_3_queries';", }; test = new PigTest(script); String[] output = { "(yahoo,{(yahoo)})", "(yahoo,{(yahoo)})", "(twitter,{(twitter)})",
275276277278279280281282283284285
"n=3", "reducers=1", "input=top_queries_input_data.txt", "output=top_3_queries", }; test = new PigTest(PIG_SCRIPT, args); // By default PigUnit removes all the STORE and DUMP test.unoverride("STORE"); test.runScript();
298299300301302303304305306
}; PigServer mockServer = null; Cluster mockCluster = null; test = new PigTest(PIG_SCRIPT, args, mockServer, mockCluster); test.assertOutput(new File("data/top_queries_expected_top_3.txt")); }
361362363364365366367368369370371
pig = new PigServer(ExecType.LOCAL); } final Cluster cluster = new Cluster(pig.getPigContext()); test = new PigTest(scriptPath, args, pig, cluster); String[] output = { "(yahoo,25)", "(facebook,15)", "(twitter,7)",
7172737475767778798081
"n=3", "reducers=1", "input=top_queries_input_data.txt", "output=top_3_queries", }; test = new PigTest(PIG_SCRIPT, args); String[] output = { "(yahoo,25)", "(facebook,15)", "(twitter,7)",
90919293949596979899100
109110111112113114115116117118119
"n=3", "reducers=1", "input=top_queries_input_data.txt", "output=top_3_queries", }; test = new PigTest(PIG_SCRIPT, args); String[] input = { "yahoo\t10", "twitter\t7", "facebook\t10",
141142143144145146147148149150151
"n=3", "reducers=1", "input=top_queries_input_data.txt", "output=top_3_queries", }; test = new PigTest(PIG_SCRIPT, args); String[] input = { "yahoo,10", "twitter,7", "facebook,10",