/*
 * Licensed to Crate under one or more contributor license agreements.
 * See the NOTICE file distributed with this work for additional
 * information regarding copyright ownership.  Crate 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.
 *
 * However, if you have executed another commercial license agreement
 * with Crate these terms will supersede the license and you may use the
 * software solely pursuant to the terms of the relevant commercial
 * agreement.
 */

package io.crate.lucene;

import io.crate.test.integration.CrateDummyClusterServiceUnitTest;
import io.crate.testing.QueryTester;
import org.elasticsearch.Version;
import org.hamcrest.Matchers;
import org.junit.Test;

public class CIDRRangeQueryTest extends CrateDummyClusterServiceUnitTest {

    @Test
    public void test_ipv4_cidr_operator() throws Throwable {
        test("ip_addr",
             new String[]{"192.168.0.1", "192.168.1.1", "192.168.1.7", "192.168.1.255", "192.168.2.0", "10.0.0.1"},
             "ip_addr << '192.168.1.1/24'",
             "192.168.1.1", "192.168.1.7", "192.168.1.255");
    }

    @Test
    public void test_ipv4_cidr_operator_right_operand_is_ip() throws Throwable {
        // operand [192.168.1.0] must conform with CIDR notation
        expectedException.expect(IllegalArgumentException.class);
        test("ip_addr",
             new String[]{"192.168.0.1", "192.168.1.1", "192.168.1.7", "192.168.1.255", "192.168.2.0", "10.0.0.1"},
             "ip_addr << '192.168.1.0'::ip",
             "192.168.1.1", "192.168.1.7");
    }

    @Test
    public void test_ipv4_cidr_operator_right_operand_is_text() throws Throwable {
        // operand [random text] must conform with CIDR notation
        expectedException.expect(IllegalArgumentException.class);
        test("ip_addr",
             new String[]{"192.168.0.1", "192.168.1.1", "192.168.1.7", "192.168.1.255", "192.168.2.0", "10.0.0.1"},
             "ip_addr << 'random text'",
             "192.168.1.1", "192.168.1.7");
    }

    private void test(String colName, String [] valuesToIndex, String queryStr, String ... expectedResults) throws Throwable {
        try (QueryTester tester = createQueryTester(colName, valuesToIndex)) {
            assertThat(tester.runQuery(colName, queryStr), Matchers.contains(expectedResults));
        }
    }

    private QueryTester createQueryTester(String colname, Object ... toIndex) throws Throwable {
        QueryTester.Builder builder = new QueryTester.Builder(
            createTempDir(),
            THREAD_POOL,
            clusterService,
            Version.CURRENT.CURRENT,
            "create table t (ip_addr ip)"
        );
        builder.indexValues(colname, toIndex);
        return builder.build();
    }
}

@@ -1366,36 +1366,25 @@
        test(
-%22ip_addr%22,%0A 
+%0A
             
@@ -1829,36 +1829,25 @@
        test(
-%22ip_addr%22,%0A 
+%0A
             
@@ -2286,20 +2286,9 @@
 est(
-%22ip_addr%22,%0A 
+%0A
     
@@ -2516,31 +2516,14 @@
 est(
-String colName, String 
+Object
 %5B%5D v
@@ -2559,17 +2559,16 @@
 , String
- 
 ... expe
@@ -2603,297 +2603,8 @@
 e %7B%0A
-        try (QueryTester tester = createQueryTester(colName, valuesToIndex)) %7B%0A            assertThat(tester.runQuery(colName, queryStr), Matchers.contains(expectedResults));%0A        %7D%0A    %7D%0A%0A    private QueryTester createQueryTester(String colname, Object ... toIndex) throws Throwable %7B%0A
     
@@ -2771,24 +2771,16 @@
 .CURRENT
-.CURRENT
 ,%0A      
@@ -2858,58 +2858,195 @@
 ues(
-colname, toIndex);%0A        return builder.build();
+%22ip_addr%22, valuesToIndex);%0A        try (QueryTester tester = builder.build()) %7B%0A            assertThat(tester.runQuery(%22ip_addr%22, queryStr), Matchers.contains(expectedResults));%0A        %7D
 %0A   

Fix varargs warning in CIDRRangeQueryTest