cvs commit: jakarta-lucene-sandbox/contributions/miscellaneous/src/test/org/apache/lucene/misc ChainedFilterTest.java

[eh...@apache.org]

ehatcher    2004/05/24 09:20:46

  Added:       contributions/miscellaneous/src/test/org/apache/lucene/misc
                        ChainedFilterTest.java
  Log:
  rudimentary test case for ChainedFilter, used initially to learn how it works
  
  Revision  Changes    Path
  1.1                  jakarta-lucene-sandbox/contributions/miscellaneous/src/test/org/apache/lucene/misc/ChainedFilterTest.java
  
  Index: ChainedFilterTest.java
  ===================================================================
  package org.apache.lucene.misc;
  
  /**
   * Copyright 2004 The Apache Software Foundation
   *
   * 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 junit.framework.TestCase;
  import java.util.Calendar;
  import java.util.Date;
  import java.text.ParseException;
  import java.text.SimpleDateFormat;
  import org.apache.lucene.index.IndexWriter;
  import org.apache.lucene.index.Term;
  import org.apache.lucene.store.RAMDirectory;
  import org.apache.lucene.analysis.WhitespaceAnalyzer;
  import org.apache.lucene.document.Document;
  import org.apache.lucene.document.Field;
  import org.apache.lucene.search.Filter;
  import org.apache.lucene.search.DateFilter;
  import org.apache.lucene.search.Hits;
  import org.apache.lucene.search.IndexSearcher;
  import org.apache.lucene.search.TermQuery;
  import org.apache.lucene.search.Query;
  import org.apache.lucene.search.BooleanQuery;
  import org.apache.lucene.search.QueryFilter;
  
  public class ChainedFilterTest extends TestCase {
    public static final int MAX = 500;
  
    private RAMDirectory directory;
    private IndexSearcher searcher;
    private Query query;
    private DateFilter dateFilter;
    private QueryFilter bobFilter;
    private QueryFilter sueFilter;
  
    public void setUp() throws Exception {
      directory = new RAMDirectory();
      IndexWriter writer =
         new IndexWriter(directory, new WhitespaceAnalyzer(), true);
  
      Calendar cal = Calendar.getInstance();
      cal.setTimeInMillis(1041397200000L); // 2003 January 01
  
      for (int i = 0; i < MAX; i++) {
        Document doc = new Document();
        doc.add(Field.Keyword("key", "" + (i + 1)));
        doc.add(
            Field.Keyword("owner", (i < MAX / 2) ? "bob" : "sue"));
        doc.add(Field.Keyword("date", cal.getTime()));
        writer.addDocument(doc);
  
        cal.add(Calendar.DATE, 1);
      }
  
      writer.close();
  
      searcher = new IndexSearcher(directory);
  
      // query for everything to make life easier
      BooleanQuery bq = new BooleanQuery();
      bq.add(new TermQuery(new Term("owner", "bob")), false, false);
      bq.add(new TermQuery(new Term("owner", "sue")),false, false);
      query = bq;
  
      // date filter matches everything too
      Date pastTheEnd = parseDate("2099 Jan 1");
      dateFilter = DateFilter.Before("date", pastTheEnd);
  
      bobFilter = new QueryFilter(
          new TermQuery(new Term("owner", "bob")));
      sueFilter = new QueryFilter(
          new TermQuery(new Term("owner", "sue")));
    }
  
    public void testSingleFilter() throws Exception {
      ChainedFilter chain = new ChainedFilter(
          new Filter[] {dateFilter});
  
      Hits hits = searcher.search(query, chain);
      assertEquals(MAX, hits.length());
  
      chain = new ChainedFilter(new Filter[] {bobFilter});
      hits = searcher.search(query, chain);
      assertEquals(MAX / 2, hits.length());
    }
  
    public void testOR() throws Exception {
      ChainedFilter chain = new ChainedFilter(
        new Filter[] {sueFilter, bobFilter});
  
      Hits hits = searcher.search(query, chain);
      assertEquals("OR matches all", MAX, hits.length());
    }
  
    public void testAND() throws Exception {
      ChainedFilter chain = new ChainedFilter(
        new Filter[] {dateFilter, bobFilter}, ChainedFilter.AND);
  
      Hits hits = searcher.search(query, chain);
      assertEquals("AND matches just bob", MAX / 2, hits.length());
      assertEquals("bob", hits.doc(0).get("owner"));
    }
  
    public void testXOR() throws Exception {
      ChainedFilter chain = new ChainedFilter(
        new Filter[]{dateFilter, bobFilter}, ChainedFilter.XOR);
  
      Hits hits = searcher.search(query, chain);
      assertEquals("XOR matches sue", MAX / 2, hits.length());
      assertEquals("sue", hits.doc(0).get("owner"));
    }
  
    public void testANDNOT() throws Exception {
      ChainedFilter chain = new ChainedFilter(
        new Filter[]{dateFilter, sueFilter},
          new int[] {ChainedFilter.AND, ChainedFilter.ANDNOT});
  
      Hits hits = searcher.search(query, chain);
      assertEquals("ANDNOT matches just bob",
          MAX / 2, hits.length());
      assertEquals("bob", hits.doc(0).get("owner"));
    }
  
    private Date parseDate(String s) throws ParseException {
      return new SimpleDateFormat("yyyy MMM dd").parse(s);
    }
  
  }
  
  
  

---------------------------------------------------------------------
To unsubscribe, e-mail: lucene-dev-unsubscribe@jakarta.apache.org
For additional commands, e-mail: lucene-dev-help@jakarta.apache.org