PlatefoodPPCSearchCommandTest.java

/* Copyright (2012) Schibsted ASA
 * This file is part of Possom.
 *
 *   Possom is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU Lesser General Public License as published by
 *   the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   Possom is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU Lesser General Public License for more details.
 *
 *   You should have received a copy of the GNU Lesser General Public License
 *   along with Possom.  If not, see <http://www.gnu.org/licenses/>.
 */
package no.sesat.search.mode.command;

import java.util.Collections;
import no.sesat.search.site.SiteKeyedFactoryInstantiationException;
import no.sesat.search.view.config.SearchTab;
import no.sesat.search.view.config.SearchTab.EnrichmentHint;
import no.sesat.search.view.config.SearchTab.EnrichmentPlacementHint;
import no.sesat.search.view.config.SearchTab.Layout;
import no.sesat.search.view.config.SearchTab.Scope;
import org.testng.annotations.Test;
import static org.testng.AssertJUnit.*;

/**
 *
 * @version $Id$
 */
public class PlatefoodPPCSearchCommandTest extends AbstractSearchCommandTest {

    @Test
    public void testSingleTerm()  throws Exception{
        executeTestOfQuery("test", "test", "");
    }


    @Test
    public void testTwoTerms()  throws Exception{
        executeTestOfQuery("test1 test2", "test1  test2", "");
    }

    @Test
    public void testThreeTerms()  throws Exception{
        executeTestOfQuery("test1 test2 test3", "test1  test2  test3", "");
    }

    @Test
    public void testTwoTermsPlus()  throws Exception{
        executeTestOfQuery("+test1 +test2", "test1  test2", "");
    }


    @Test
    public void testNot()  throws Exception{
        executeTestOfQuery("test1 -test2", "test1 NOT test2", "");
    }

    @Test
    public void testLeadingNot()  throws Exception{
        executeTestOfQuery("-test1 test2", "NOT test1  test2", "");
    }

    @Test
    public void testOr()  throws Exception{
        executeTestOfQuery("(test1 test2)", "test1 OR test2", "");
    }

    @Test
    public void testLeadingNotAndOr()  throws Exception{
        executeTestOfQuery("-a (test1 test2)", "NOT a  test1 OR test2", "");
    }

    @Test
    public void testLotsOfNots()  throws Exception{
        executeTestOfQuery("-a d -b -e c -f", "NOT a  d NOT b NOT e  c NOT f", "");
        executeTestOfQuery("-a d -b -e c g -f", "NOT a  d NOT b NOT e  c  g NOT f", "");
        executeTestOfQuery("-a -b c", "NOT a NOT b  c", "");
        executeTestOfQuery("-a (e f) c", "NOT a  e OR f  c", "");
    }


    @Test
    public void testSiteRestriction() throws Exception{

        assertEquals(
                "Site filter not found",
                "aftonbladet.se",executeTestOfQuery("site:aftonbladet.se banan", "banan", "+site:aftonbladet.se")
                .getFilterBuilder().getFilter("site"));

        assertEquals(
                "Site filter not found",
                "aftonbladet.se",executeTestOfQuery("banan site:aftonbladet.se", "banan", "+site:aftonbladet.se")
                .getFilterBuilder().getFilter("site"));
    }

    /**
     * Asserts that generated query equals the expected generated query.
     * Asserts that generated filter equals the expected filter.
     *
     * @param query         The query.
     * @param wantedQuery   The expected query.
     * @param wantedFilter  The expected filter.
     */
    private AbstractYahooSearchCommand executeTestOfQuery(
            final String query,
            final String wantedQuery,
            final String wantedFilter)  throws SiteKeyedFactoryInstantiationException{

        final SearchTab fakeTab = new SearchTab(null, "fake-view", "default-mode", "test", null, null, true,
                null, Collections.<EnrichmentPlacementHint>emptyList(), Collections.<EnrichmentHint>emptyList(), null,
                0, 0, Collections.<String>emptyList(), Collections.<String>emptyList(), false, false,
                null, Collections.<String,Layout>emptyMap(), Scope.REQUEST);

        final SearchCommand.Context cxt = createCommandContext(query, fakeTab, "default-platefood-ppc-command");
        final AbstractYahooSearchCommand cmd = new PlatefoodPPCSearchCommand(cxt);
        final String generatedQuery = cmd.getQueryRepresentation();
        assertEquals("Generated query does not match wanted query", wantedQuery, generatedQuery.trim());
        assertEquals("Generated filter does not match wanter filter", wantedFilter, cmd.getFilter());
        return cmd;
    }
}