QueryTranslator_filterStatusTest.java

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

package com.enonic.cms.core.search.query;

import org.elasticsearch.index.query.FilterBuilder;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.junit.Test;

import com.enonic.cms.core.content.ContentStatus;
import com.enonic.cms.core.content.index.ContentIndexQuery;
import com.enonic.cms.core.search.query.factory.FilterQueryBuilderFactory;

public class QueryTranslator_filterStatusTest
    extends QueryTranslatorTestBase
{
    FilterQueryBuilderFactory filterQueryBuilderFactory = new FilterQueryBuilderFactory();

    @Test
    public void testStatusFilter()
    {
        String expected = "{\n" +
            "  \"post_filter\" : {\n" +
            "    \"term\" : {\n" +
            "      \"status\" : \"2\"\n" +
            "    }\n" +
            "  }\n" +
            "}";

        SearchSourceBuilder builder = new SearchSourceBuilder();

        ContentIndexQuery query = new ContentIndexQuery( "" );

        query.setContentStatusFilter( ContentStatus.APPROVED.getKey() );

        final FilterBuilder filterToApply = filterQueryBuilderFactory.buildFilter( query );
        if ( filterToApply != null )
        {
            builder.filter( filterToApply );
        }

        compareStringsIgnoreFormatting( expected, builder.toString() );
    }


}