} else {
assertEquals(stats.getJobGraph().size(), 4);
}
assertTrue(stats.getJobGraph().getSinks().size() == 1);
assertTrue(stats.getJobGraph().getSources().size() == 1);
JobStats js = (JobStats) stats.getJobGraph().getSinks().get(0);
assertEquals(OUTPUT_FILE, js.getOutputs().get(0).getName());
assertEquals(2, js.getOutputs().get(0).getNumberRecords());
assertEquals(12, js.getOutputs().get(0).getBytes());
assertEquals(OUTPUT_FILE, stats.getOutputNames().get(0));
assertEquals(2, stats.getRecordWritten());
assertEquals(12, stats.getBytesWritten());
if (execType.equals("tez")) {
assertEquals("A,B", ((JobStats) stats.getJobGraph().getSources().get(
0)).getAlias());
// TODO: alias is not set for sample-aggregation/partition/sort job.
// Need to investigate
// assertEquals("B", ((JobStats) stats.getJobGraph().getPredecessors(
// js).get(0)).getAlias());
} else {
assertEquals("A", ((JobStats) stats.getJobGraph().getSources().get(
0)).getAlias());
assertEquals("B", ((JobStats) stats.getJobGraph().getPredecessors(
js).get(0)).getAlias());
assertEquals("B", js.getAlias());
}
} finally {
new File(PIG_FILE).delete();
Util.deleteFile(cluster, OUTPUT_FILE);
}