QueryOnRecordIdTest.java

/*
 * Copyright (c) 2013-2017 Cinchapi Inc.
 * 
 * Licensed 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 com.cinchapi.concourse;

import java.util.Set;

import org.junit.Assert;
import org.junit.Test;
import com.cinchapi.concourse.lang.Criteria;
import com.cinchapi.concourse.test.ConcourseIntegrationTest;
import com.cinchapi.concourse.thrift.Operator;
import com.google.common.collect.Sets;

/**
 * Class includes test cases for a feature in findAtomic() method in
 * ConcourseServer class.
 * It includes test for handling id related expression in query at the Server
 * rather than passing the query to Engine.
 * 
 * @author Raghav
 */
public class QueryOnRecordIdTest extends ConcourseIntegrationTest {

    @Override
    protected void beforeEachTest() {
        for (int i = 30; i <= 50; i++) {
            client.add("name", "foo" + i, i);
        }
    }

    @Test
    public void testRecordRetrievalMatchingId() {
        Set<Long> set = Sets.newHashSet();
        set.add(new Long(35));
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(35).build()));
    }

    @Test
    public void testRecordRetrievalNotMatchingId() {
        Set<Long> set = Sets.newHashSet();
        for (long i = 30; i <= 50; i++) {
            if(i != 35) {
                set.add(i);
            }
        }
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.NOT_EQUALS).value(35).build()));
    }

    @Test
    public void testRecordRetrievaIWithIdAndOperator() {
        Set<Long> set = Sets.newHashSet();
        set.add(new Long(35));
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(35).and().key("name")
                        .operator(Operator.EQUALS).value("foo35").build()));
        set = Sets.newHashSet();
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(55).and().key("name")
                        .operator(Operator.EQUALS).value("foo35").build()));
    }

    @Test
    public void testRecordRetrievaIWithIdOrOperator() {
        Set<Long> set = Sets.newHashSet();
        set.add(new Long(35));
        set.add(new Long(40));
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(35).or().key("name")
                        .operator(Operator.EQUALS).value("foo40").build()));

        set = Sets.newHashSet();
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(55).and().key("name")
                        .operator(Operator.EQUALS).value("foo46").build()));
    }

    @Test
    public void testRecordRetrievalNonExistingId() {
        Set<Long> set = Sets.newHashSet();
        set.add(new Long(55));
        Assert.assertEquals(
                set,
                client.find(Criteria.where()
                        .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                        .operator(Operator.EQUALS).value(55).build()));
    }

    @Test(expected = Exception.class)
    public void testQueryOnRecordIdNonEqualOrNonEqualsOperator() {
        Operator operator = Operator.GREATER_THAN;
        client.find(Criteria.where()
                .key(Constants.JSON_RESERVED_IDENTIFIER_NAME)
                .operator(operator).value(55).build());
    }
    
    @Test
    public void testQueryOnRecordIdComplexCcl(){
        for (int i = 30; i <= 50; i++) {
            client.add("name", "foo", i);
        }
        for(int i = 0; i < 20; ++i){
            client.add("bar", i, i);
        }
        String ccl = "(name = foo OR bar > 15) AND $id$ != 40";
        Set<Long> actual = client.find(ccl);
        Assert.assertFalse(actual.contains(40));
    }

}