public class TestLoaderStorerShipCacheFilesTez extends TestLoaderStorerShipCacheFiles {
@Override
protected void checkPlan(PhysicalPlan pp, String[] expectedFiles, int size, PigContext pigContext) throws Exception {
TezLauncher launcher = new TezLauncher();
TezPlanContainer tezPlanContainer = launcher.compile(pp, pigContext);
assertPlanContains(tezPlanContainer.getRoots().get(0).getTezOperPlan(), expectedFiles, size);
}
private void assertPlanContains(TezOperPlan plan, String[] expectedFiles, int size) throws VisitorException {
TezPOUserFuncVisitor udfVisitor = new TezPOUserFuncVisitor(plan);