modules/indexing/src/test/java/org/apache/ignite/internal/processors/query/SqlPushDownFunctionTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.ignite.internal.processors.query;

import java.util.List;
import org.apache.ignite.cache.query.SqlFieldsQuery;
import org.apache.ignite.internal.IgniteEx;
import org.apache.ignite.internal.processors.cache.index.AbstractIndexingCommonTest;
import org.junit.Test;

/**
 * Tests for schemas.
 */
public class SqlPushDownFunctionTest extends AbstractIndexingCommonTest {
    /** Node. */
    private IgniteEx node;

    /** {@inheritDoc} */
    @Override protected void beforeTest() throws Exception {
        node = (IgniteEx)startGrid();

        startGrid(2);
    }

    /** {@inheritDoc} */
    @Override protected void afterTest() throws Exception {
        stopAllGrids();
    }

    /**
     */
    @Test
    public void testPushDownFunction() {
        sql("CREATE TABLE Person(id INTEGER PRIMARY KEY, company_id INTEGER)");
        sql("CREATE TABLE Company(id INTEGER PRIMARY KEY, name VARCHAR)");

        sql("INSERT INTO Person(id,company_id) VALUES (1, 1), (2, 2), (3, 3)");
        sql("INSERT INTO Company(id,name) VALUES (1,'n1'), (2,'n2'), (3,'n3')");

        sql("SELECT p.id, sum(p.id) FROM person p " +
            "LEFT JOIN (select distinct id from company) as c on c.id=p.company_id " +
            "GROUP BY p.id");

        sql("SELECT count(1) FROM person p " +
            "LEFT JOIN (select id from company union select id from company) as c on c.id=p.company_id");

        sql("SELECT count(1) FROM person p " +
            "LEFT JOIN (select id from company union select id from company) as c on c.id=p.company_id");

        sql("SELECT 1, (select count(1)) FROM person p " +
            "LEFT JOIN (select id from company union select id from company) as c on c.id=p.company_id");
    }

    /**
     * @param sql SQL query.
     * @return Results.
     */
    private List<List<?>> sql(String sql) {
        GridQueryProcessor qryProc = node.context().query();

        SqlFieldsQuery qry = new SqlFieldsQuery(sql).setSchema("PUBLIC");

        return qryProc.querySqlFields(qry, true).getAll();
    }
}