ContentIndexServiceImpl_countTest.java

/*
 * Copyright 2000-2013 Enonic AS
 * http://www.enonic.com/license
 */

package com.enonic.cms.itest.search;

import org.junit.Test;

import com.enonic.cms.core.content.ContentKey;
import com.enonic.cms.core.content.contenttype.ContentTypeKey;
import com.enonic.cms.core.content.index.ContentIndexQuery;
import com.enonic.cms.core.content.resultset.ContentResultSet;
import com.enonic.cms.core.search.ContentIndexServiceImpl;
import com.enonic.cms.core.search.query.ContentDocument;

import static org.junit.Assert.*;

public class ContentIndexServiceImpl_countTest
    extends ContentIndexServiceTestBase
{


    @Test
    public void testLargeCountNumber()
        throws Exception
    {
        indexContent( 1 );
        indexContent( 2 );
        indexContent( 3 );
        indexContent( 4 );
        indexContent( 5 );

        flushIndex();

        ContentIndexQuery query = new ContentIndexQuery( "title = 'test'" );
        query.setCount( Integer.MAX_VALUE );

        final ContentResultSet result = contentIndexService.query( query );

        // Will fail with exception if count not working
    }

    private void indexContent( int contentKey )
    {
        ContentDocument contentDoc = new ContentDocument( new ContentKey( contentKey ) );
        contentDoc.setTitle( "test" );
        contentDoc.setContentTypeKey( new ContentTypeKey( 1 ) );
        contentDoc.setContentTypeName( "testContentType" );

        contentIndexService.index( contentDoc );
    }

    @Test
    public void testQueryThreshold()
    {
        for ( int i = 1; i <= ContentIndexServiceImpl.COUNT_OPTIMIZER_THRESHOULD_VALUE + 10; i++ )
        {
            indexContent( i );
        }

        flushIndex();

        ContentIndexQuery query = new ContentIndexQuery( "" );

        final ContentResultSet result = contentIndexService.query( query );

        assertEquals( ContentIndexServiceImpl.COUNT_OPTIMIZER_THRESHOULD_VALUE, result.getLength() );
    }

    @Test
    public void testReturnAllContent()
    {
        for ( int i = 1; i <= ContentIndexServiceImpl.COUNT_OPTIMIZER_THRESHOULD_VALUE + 10; i++ )
        {
             indexContent( i );
        }

        flushIndex();

        ContentIndexQuery query = new ContentIndexQuery( "" );
        query.setReturnAllHits( true );

        final ContentResultSet result = contentIndexService.query( query );

        assertEquals( ContentIndexServiceImpl.COUNT_OPTIMIZER_THRESHOULD_VALUE + 10, result.getLength() );
    }


}