// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition;

import com.yahoo.searchdefinition.parser.ParseException;
import com.yahoo.yolean.Exceptions;
import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

/**
 * @author bratseth
 */
public class RankingExpressionLoopDetectionTestCase {

    @Test
    public void testSelfLoop() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "        field a type string { \n" +
                "            indexing: index \n" +
                "        }\n" +
                "    }\n" +
                "    \n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo\n" +
                "        }\n" +
                "        function foo() {\n" +
                "            expression: foo\n" +
                "        }\n" +
                "    }\n" +
                "\n" +
                "}\n");
        try {
            builder.build();
            fail("Excepted exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("In schema 'test', rank profile 'test': The function 'foo' is invalid: foo is invalid: Invocation loop: foo -> foo",
                         Exceptions.toMessageString(e));
        }
    }

    @Test
    public void testNestedLoop() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "        field a type string { \n" +
                "            indexing: index \n" +
                "        }\n" +
                "    }\n" +
                "    \n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo\n" +
                "        }\n" +
                "        function foo() {\n" +
                "            expression: arg(5)\n" +
                "        }\n" +
                "        function arg(a1) {\n" +
                "            expression: foo + a1*2\n" +
                "        }\n" +
                "    }\n" +
                "\n" +
                "}\n");
        try {
            builder.build();
            fail("Excepted exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("In schema 'test', rank profile 'test': The function 'foo' is invalid: arg(5) is invalid: foo is invalid: arg(5) is invalid: Invocation loop: arg(5) -> foo -> arg(5)",
                         Exceptions.toMessageString(e));
        }
    }

    @Test
    public void testSelfArgumentLoop() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "        field a type string { \n" +
                "            indexing: index \n" +
                "        }\n" +
                "    }\n" +
                "    \n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo\n" +
                "        }\n" +
                "        function foo() {\n" +
                "            expression: arg(foo)\n" +
                "        }\n" +
                "        function arg(a1) {\n" +
                "            expression: a1*2\n" +
                "        }\n" +
                "    }\n" +
                "\n" +
                "}\n");
        try {
            builder.build();
            fail("Excepted exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("In schema 'test', rank profile 'test': The function 'foo' is invalid: arg(foo) is invalid: a1 is invalid: foo is invalid: arg(foo) is invalid: Invocation loop: arg(foo) -> foo -> arg(foo)",
                         Exceptions.toMessageString(e));
        }
    }

    @Test
    public void testNoLoopWithSameLocalArgument() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "        field a type string { \n" +
                "            indexing: index \n" +
                "        }\n" +
                "    }\n" +
                "    \n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo(3)\n" +
                "        }\n" +
                "        function foo(a1) {\n" +
                "            expression: bar(3)\n" +
                "        }\n" +
                "        function bar(a1) {\n" +
                "            expression: a1*2\n" +
                "        }\n" +
                "    }\n" +
                "\n" +
                "}\n");
        builder.build();
    }

    @Test
    public void testNoLoopWithMultipleInvocations() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "        field a type string { \n" +
                "            indexing: index \n" +
                "        }\n" +
                "    }\n" +
                "    \n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo(3)\n" +
                "        }\n" +
                "        function foo(a1) {\n" +
                "            expression: bar(3) + bar(a1)\n" +
                "        }\n" +
                "        function bar(a1) {\n" +
                "            expression: a1*2\n" +
                "        }\n" +
                "    }\n" +
                "\n" +
                "}\n");
        builder.build();
    }

    @Test
    public void testNoLoopWithBoundIdentifiers() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "    }\n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo(bar(2))\n" +
                "        }\n" +
                "        function foo(x) {\n" +
                "            expression: x * x\n" +
                "        }\n" +
                "        function bar(x) {\n" +
                "            expression: x + x\n" +
                "        }\n" +
                "    }\n" +
                "}\n");
        builder.build();
    }

    @Test
    public void testNoLoopWithTheSameNestedIdentifierWhichIsUnbound() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "    }\n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo()\n" +
                "        }\n" +
                "        function foo() {\n" +
                "            expression: bar(x)\n" +
                "        }\n" +
                "        function bar(x) {\n" +
                "            expression: x + x\n" +
                "        }\n" +
                "    }\n" +
                "}\n");
        builder.build();
    }

    @Test
    public void testNoLoopWithTheSameAlternatingNestedIdentifierWhichIsUnbound() throws ParseException {
        RankProfileRegistry rankProfileRegistry = new RankProfileRegistry();
        SchemaBuilder builder = new SchemaBuilder(rankProfileRegistry);
        builder.importString(
                "search test {\n" +
                "    document test { \n" +
                "    }\n" +
                "    rank-profile test {\n" +
                "        first-phase {\n" +
                "            expression: foo()\n" +
                "        }\n" +
                "        function foo() {\n" +
                "            expression: bar(x)\n" +
                "        }\n" +
                "        function bar(y) {\n" +
                "            expression: baz(y)\n" +
                "        }\n" +
                "        function baz(x) {\n" +
                "            expression: x + x\n" +
                "        }\n" +
                "    }\n" +
                "}\n");
        builder.build();
    }

}