package com.browseengine.bobo.test;
/**
* Bobo Browse Engine - High performance faceted/parametric search implementation
* that handles various types of semi-structured data. Written in Java.
*
* Copyright (C) 2005-2006 John Wang
*
* This library 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 2.1 of the License, or (at your option) any later version.
*
* This library 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 this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*
* To contact the project administrators for the bobo-browse project,
* please go to https://sourceforge.net/projects/bobo-browse/, or
* send mail to owner@browseengine.com.
*/
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.List;
import junit.framework.TestCase;
import org.apache.log4j.Logger;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.StringField;
import org.apache.lucene.index.DirectoryReader;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.RAMDirectory;
import com.browseengine.bobo.api.BoboBrowser;
import com.browseengine.bobo.api.BoboMultiReader;
import com.browseengine.bobo.api.BrowseException;
import com.browseengine.bobo.api.BrowseRequest;
import com.browseengine.bobo.api.BrowseResult;
import com.browseengine.bobo.api.BrowseSelection;
import com.browseengine.bobo.api.FacetSpec;
import com.browseengine.bobo.api.FacetSpec.FacetSortSpec;
import com.browseengine.bobo.facets.FacetHandler;
import com.browseengine.bobo.facets.impl.SimpleFacetHandler;
import com.browseengine.bobo.index.BoboIndexer;
import com.browseengine.bobo.index.digest.DataDigester;
/**
* This class is to test the case when facetName is different from the underlying indexingFieldName for simpleFacetHandler
* @author hyan
*
*/
public class FacetNameTest extends TestCase {
static Logger log = Logger.getLogger(FacetNameTest.class);
private final List<FacetHandler<?>> _facetHandlers;
private final int _documentSize;
private static class TestDataDigester extends DataDigester {
private final Document[] _data;
TestDataDigester(List<FacetHandler<?>> facetHandlers, Document[] data) {
super();
_data = data;
}
@Override
public void digest(DataHandler handler) throws IOException {
for (int i = 0; i < _data.length; ++i) {
handler.handleDocument(_data[i]);
}
}
}
public FacetNameTest(String testName) {
super(testName);
_facetHandlers = createFacetHandlers();
_documentSize = 10;
String confdir = System.getProperty("conf.dir");
if (confdir == null) confdir = "./resource";
System.setProperty("log.home", ".");
org.apache.log4j.PropertyConfigurator.configure(confdir + "/log4j.properties");
}
public Document[] createData() {
ArrayList<Document> dataList = new ArrayList<Document>();
for (int i = 0; i < _documentSize; ++i) {
String color = null;
if (i == 0) color = "red";
else if (i == 1) color = "green";
else if (i == 2) color = "blue";
else if (i % 2 == 0) color = "yellow";
else color = "white";
String make = null;
if (i == 0) make = "camry";
else if (i == 1) make = "accord";
else if (i == 2) make = "4runner";
else if (i % 2 == 0) make = "rav4";
else make = "prius";
String ID = Integer.toString(i);
Document d = new Document();
d.add(new StringField("id", ID, Field.Store.YES));
d.add(new StringField("color", color, Field.Store.YES));
d.add(new StringField("make", make, Field.Store.YES));
dataList.add(d);
}
return dataList.toArray(new Document[dataList.size()]);
}
private Directory createIndex() {
Directory dir = new RAMDirectory();
try {
Document[] data = createData();
TestDataDigester testDigester = new TestDataDigester(_facetHandlers, data);
BoboIndexer indexer = new BoboIndexer(testDigester, dir);
indexer.index();
DirectoryReader r = DirectoryReader.open(dir);
r.close();
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
}
return dir;
}
public static List<FacetHandler<?>> createFacetHandlers() {
List<FacetHandler<?>> facetHandlers = new ArrayList<FacetHandler<?>>();
facetHandlers.add(new SimpleFacetHandler("id"));
facetHandlers.add(new SimpleFacetHandler("make"));
facetHandlers.add(new SimpleFacetHandler("mycolor", "color"));
return facetHandlers;
}
public void testFacetNameForSimpleFacetHandler() throws Exception {
BrowseRequest br = new BrowseRequest();
br.setCount(20);
br.setOffset(0);
BrowseSelection colorSel = new BrowseSelection("mycolor");
colorSel.addValue("yellow");
br.addSelection(colorSel);
BrowseSelection makeSel = new BrowseSelection("make");
makeSel.addValue("rav4");
br.addSelection(makeSel);
FacetSpec spec = new FacetSpec();
spec.setExpandSelection(true);
spec.setOrderBy(FacetSortSpec.OrderHitsDesc);
spec.setMaxCount(15);
br.setFacetSpec("mycolor", spec);
br.setFacetSpec("id", spec);
br.setFacetSpec("make", spec);
BrowseResult result = null;
BoboBrowser boboBrowser = null;
int expectedHitNum = 3;
try {
Directory ramIndexDir = createIndex();
DirectoryReader srcReader = DirectoryReader.open(ramIndexDir);
boboBrowser = new BoboBrowser(BoboMultiReader.getInstance(srcReader, _facetHandlers));
result = boboBrowser.browse(br);
assertEquals(expectedHitNum, result.getNumHits());
} catch (BrowseException e) {
e.printStackTrace();
fail(e.getMessage());
} catch (IOException ioe) {
fail(ioe.getMessage());
} finally {
if (boboBrowser != null) {
try {
if (result != null) result.close();
boboBrowser.close();
} catch (IOException e) {
fail(e.getMessage());
}
}
}
}
}