Method testWithDereferenceOperator

test/org/apache/pig/test/TestCase.java:245–280

Verify that CASE statement works when expressions contain dereference operators. @throws Exception

()

     * @throws Exception
     */
    @Test
    public void testWithDereferenceOperator() throws Exception {
        PigServer pigServer = new PigServer(Util.getLocalTestMode());
        Data data = resetData(pigServer);

        data.set("foo",
                tuple("a","x",1),
                tuple("a","y",1),
                tuple("b","x",2),
                tuple("b","y",2),
                tuple("c","x",3),
                tuple("c","y",3)
                );

        pigServer.registerQuery("A = LOAD 'foo' USING mock.Storage() AS (c1:chararray, c2:chararray, i:int);");
        pigServer.registerQuery("B = GROUP A BY (c1, i);");
        pigServer.registerQuery("C = FOREACH B GENERATE group.i, (" +
                "  CASE group.i % 3" +
                "    WHEN 0 THEN '3n'" +
                "    WHEN 1 THEN '3n+1'" +
                "    ELSE        '3n+2'" +
                "  END" +
                "), A.(c1, c2);");
        pigServer.registerQuery("STORE C INTO 'bar' USING mock.Storage();");

        List<Tuple> out = data.get("bar");

        String[] expected = new String[] {
                "(1,3n+1,{(a,x),(a,y)})",
                "(2,3n+2,{(b,x),(b,y)})",
                "(3,3n,{(c,x),(c,y)})"
        };
        Schema s = pigServer.dumpSchema("C");
        Util.checkQueryOutputs(out.iterator(), expected, org.apache.pig.newplan.logical.Util.translateSchema(s),
                Util.isSparkExecType(Util.getLocalTestMode()));
    }

    /**
     * Verify that FrontendException is thrown when case expression is missing,