org.eclipse.smila.lucene.test.TestLuceneIndexService.java Source code

Introduction

Here is the source code for org.eclipse.smila.lucene.test.TestLuceneIndexService.java

Source

/***********************************************************************************************************************
 * Copyright (c) 2008 empolis GmbH and brox IT Solutions GmbH. All rights reserved. This program and the accompanying
 * materials are made available under the terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors: Daniel Stucky (empolis GmbH) - initial API and implementation, 
 * Drazen Cindric (Attensity Europe GmbH) - using LuceneIndexPipelet
 **********************************************************************************************************************/
package org.eclipse.smila.lucene.test;

import java.io.File;

import org.apache.lucene.index.IndexReader;
import org.eclipse.smila.blackboard.Blackboard;
import org.eclipse.smila.blackboard.BlackboardFactory;
import org.eclipse.smila.datamodel.Record;
import org.eclipse.smila.lucene.LuceneIndexService;
import org.eclipse.smila.lucene.LuceneService;
import org.eclipse.smila.lucene.pipelets.LuceneIndexPipelet;
import org.eclipse.smila.test.DeclarativeServiceTestCase;
import org.eclipse.smila.utils.workspace.WorkspaceHelper;

/**
 * The Class TestLuceneIndexService.
 */
public class TestLuceneIndexService extends DeclarativeServiceTestCase {

    /** The Constant DOCS_QUANTITY. */
    private static final int DOCS_QUANTITY = 5;

    /**
     * Name of the test index.
     */
    private static final String TEST_INDEX_NAME = "test_index";

    /** the Blackboard. */
    private Blackboard _blackboard;

    /** the LuceneIndexService. */
    private LuceneIndexService _luceneIndexService;

    /** the index pipelet to test. */
    private LuceneIndexPipelet _luceneIndexPipelet;

    /**
     * the test index directory.
     */
    private File _testIndexDir;

    /**
     * {@inheritDoc}
     * 
     * @see junit.framework.TestCase#setUp()
     */
    @Override
    protected void setUp() throws Exception {
        final BlackboardFactory factory = getService(BlackboardFactory.class);
        assertNotNull(factory);
        _blackboard = factory.createPersistingBlackboard();
        assertNotNull(_blackboard);
        _luceneIndexService = getService(LuceneIndexService.class);
        assertNotNull(_luceneIndexService);
        _testIndexDir = WorkspaceHelper.createWorkingDir(LuceneService.BUNDLE_NAME, TEST_INDEX_NAME);
        assertNotNull(_testIndexDir);
        if (!_luceneIndexService.isIndexExists(TEST_INDEX_NAME)) {
            _luceneIndexService.createIndex(TEST_INDEX_NAME);
        }
        _luceneIndexPipelet = new LuceneIndexPipelet();
        assertNotNull(_luceneIndexPipelet);
    }

    /**
     * {@inheritDoc}
     * 
     * @see junit.framework.TestCase#tearDown()
     */
    @Override
    protected void tearDown() throws Exception {
        _blackboard = null;
        _luceneIndexService = null;
        _luceneIndexPipelet = null;
    }

    /**
     * Test add and delete document.
     * 
     * @throws Exception
     *           if any error occurs
     */
    public void testAddDeleteDocument() throws Exception {
        addDocument();
        deleteDocument();
    }

    /**
     * Add documents.
     * 
     * @throws Exception
     *           if any error occurs
     */
    private void addDocument() throws Exception {
        IndexReader indexReader = IndexReader.open(_testIndexDir);
        assertEquals(0, indexReader.numDocs());

        final Record[] records = AllTests.createRecords(5);
        assertEquals(DOCS_QUANTITY, records.length);
        final String[] recordIds = new String[DOCS_QUANTITY];
        for (int i = 0; i < records.length; i++) {
            recordIds[i] = records[i].getId();
            _blackboard.setRecord(records[i]);
            AllTests.setAnnotations(_blackboard, records[i].getId(), LuceneIndexService.ExecutionMode.ADD,
                    TEST_INDEX_NAME);
        }
        final String[] result = _luceneIndexPipelet.process(_blackboard, recordIds);
        assertEquals(DOCS_QUANTITY, result.length);

        indexReader = indexReader.reopen();
        assertEquals(DOCS_QUANTITY, indexReader.numDocs());
        indexReader.close();
    }

    /**
     * Delete documents.
     * 
     * @throws Exception
     *           if any error occurs
     */
    private void deleteDocument() throws Exception {
        IndexReader indexReader = IndexReader.open(_testIndexDir);
        assertEquals(DOCS_QUANTITY, indexReader.numDocs());

        final Record[] records = AllTests.createRecords(5);
        assertEquals(DOCS_QUANTITY, records.length);
        final String[] recordIds = new String[DOCS_QUANTITY];
        for (int i = 0; i < records.length; i++) {
            recordIds[i] = records[i].getId();
            AllTests.setAnnotations(_blackboard, records[i].getId(), LuceneIndexService.ExecutionMode.DELETE,
                    TEST_INDEX_NAME);
        }
        final String[] result = _luceneIndexPipelet.process(_blackboard, recordIds);
        assertEquals(DOCS_QUANTITY, result.length);
        indexReader = indexReader.reopen();
        assertEquals(0, indexReader.numDocs());
        indexReader.close();
    }
}