com.plugtree.solradvert.util.AbstractAdvertTestCase.java Source code

Introduction

Here is the source code for com.plugtree.solradvert.util.AbstractAdvertTestCase.java

Source

package com.plugtree.solradvert.util;

/**
 *      Copyright 2011 Plugtree LLC
 *
 *  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.
 */

import static org.junit.Assert.assertNull;

import java.util.Collection;
import java.util.Date;
import java.util.LinkedList;

import org.apache.lucene.search.Query;
import org.apache.solr.common.params.CommonParams;
import org.apache.solr.request.SolrQueryRequest;
import org.apache.solr.search.QParser;
import org.easymock.EasyMock;
import org.junit.After;
import org.junit.Rule;
import org.junit.rules.TemporaryFolder;

public abstract class AbstractAdvertTestCase {

    private Collection<Object> mocks = null;

    @Rule
    public TemporaryFolder tmpFolder = new TemporaryFolder();

    @Rule
    public TestHarnessProvider harnessProvider = new TestHarnessProvider();

    @After
    public void validateMocks() {
        if (mocks != null) {
            EasyMock.verify(mocks.toArray());
            mocks = null;
        }
    }

    protected <T> T createMock(Class<T> clazz) {
        T mock = EasyMock.createMock(clazz);

        if (mocks == null) {
            mocks = new LinkedList<Object>();
        }

        mocks.add(mock);

        return mock;
    }

    protected void replayAllMocks() {
        if (mocks != null) {
            EasyMock.replay(mocks.toArray());
        }
    }

    public String getDataDirectory() {
        return tmpFolder.newFolder("data").getAbsolutePath();
    }

    protected Query getQuery(String qstr) throws Exception {
        SolrQueryRequest req = newRequest("q", qstr);
        QParser qparser = QParser.getParser(req.getParams().get(CommonParams.Q), "lucene", req);
        return qparser.getQuery();
    }

    protected void assertAddDoc(String id, String product, String brand, String description, Date date,
            Double price) throws Exception {
        assertNull("Error adding document", harnessProvider.getHarness().validateAddDoc("id", id, "product",
                product, "brand", brand, "description", description, "price", price.toString()));
    }

    protected void assertCommit() throws Exception {
        assertNull("Error comitting", harnessProvider.getHarness().validateUpdate("<commit/>"));
    }

    protected SolrQueryRequest newRequest(String... args) {
        return harnessProvider.getRequestFactory().makeRequest(args);
    }

    protected void assertQuery(SolrQueryRequest req, String... tests) throws Exception {
        assertNull(harnessProvider.getHarness().validateQuery(req, tests));
    }

}