Method testBzipStoreInMultiQuery

test/org/apache/pig/test/TestBZip.java:503–535

()

    }

    @Test
    public void testBzipStoreInMultiQuery() throws Exception {
        String[] inputData = new String[] {
                "1\t2\r3\t4"
        };

        try {
            String inputFileName = "input.txt";
            Util.createInputFile(cluster, inputFileName, inputData);

            PigServer pig = new PigServer(cluster.getExecType(), properties);

            pig.setBatchOn();
            pig.registerQuery("a = load '" +  inputFileName + "';");
            pig.registerQuery("store a into 'output.bz2';");
            pig.registerQuery("store a into 'output';");
            pig.executeBatch();

            FileSystem fs = FileSystem.get(ConfigurationUtil.toConfiguration(
                    pig.getPigContext().getProperties()));
            FileStatus[] outputFiles = fs.listStatus(new Path("output"),
                    Util.getSuccessMarkerPathFilter());
            assertTrue(outputFiles[0].getLen() > 0);

            outputFiles = fs.listStatus(new Path("output.bz2"),
                    Util.getSuccessMarkerPathFilter());
            assertTrue(outputFiles[0].getLen() > 0);
        } finally {
            Util.deleteFile(cluster, "input.txt");
            Util.deleteFile(cluster, "output.bz2");
            Util.deleteFile(cluster, "output");
        }
    }

    @Test
    public void testBzipStoreInMultiQuery2() throws Exception {