/*
 * Copyright (c) 2006, 2021 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 */

// Contributors:
//     Oracle - initial API and implementation
//
package org.eclipse.persistence.jpa.tests.jpql.parser;

import org.junit.Test;
import static org.eclipse.persistence.jpa.tests.jpql.parser.JPQLParserTester.*;

@SuppressWarnings("nls")
public final class IndexExpressionTest extends JPQLParserTest {

    private JPQLQueryStringFormatter buildQueryFormatter_03() {
        return new JPQLQueryStringFormatter() {
            @Override
            public String format(String query) {
                return query.replace("INDEX(BETWEEN", "INDEX( BETWEEN");
            }
        };
    }

    @Test
    public void test_JPQLQuery_01() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX(c) BETWEEN 0 AND 9";

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index("c").between(numeric(0), numeric(9)))
        );

        testQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_02() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX() BETWEEN 0 AND 9";

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index(nullExpression()).between(numeric(0), numeric(9)))
        );

        testInvalidQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_03() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX( BETWEEN 0 AND 9";

        IndexExpressionTester index = index(nullExpression());
        index.hasLeftParenthesis  = true;
        index.hasRightParenthesis = false;

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index.between(numeric(0), numeric(9)))
        );

        testInvalidQuery(query, selectStatement, buildQueryFormatter_03());
    }

    @Test
    public void test_JPQLQuery_04() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX BETWEEN 0 AND 9";

        IndexExpressionTester index = index(nullExpression());
        index.hasLeftParenthesis  = false;
        index.hasRightParenthesis = false;

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index.between(numeric(0), numeric(9)))
        );

        testInvalidQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_05() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX) ";

        IndexExpressionTester index = index(nullExpression());
        index.hasLeftParenthesis  = false;
        index.hasRightParenthesis = true;

        SelectStatementTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index)
        );

        selectStatement.hasSpaceAfterWhere = true;

        testInvalidQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_06() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX e)";

        IndexExpressionTester index = index("e");
        index.hasLeftParenthesis  = false;
        index.hasRightParenthesis = true;

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index)
        );

        testInvalidQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_07() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX(e";

        IndexExpressionTester index = index("e");
        index.hasLeftParenthesis  = true;
        index.hasRightParenthesis = false;

        ExpressionTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index)
        );

        testInvalidQuery(query, selectStatement);
    }

    @Test
    public void test_JPQLQuery_08() {

        String query = "SELECT c FROM CreditCard c WHERE INDEX(e ";

        IndexExpressionTester index = index("e");
        index.hasLeftParenthesis  = true;
        index.hasRightParenthesis = false;

        SelectStatementTester selectStatement = selectStatement(
            select(variable("c")),
            from("CreditCard", "c"),
            where(index)
        );

        selectStatement.hasSpaceAfterWhere = true;

        testInvalidQuery(query, selectStatement);
    }
}
