← Back to branch summary

~slub.team/goobi-indexserver/3.x

« back to all changes in this revision

Viewing changes to lucene/contrib/facet/src/test/org/apache/lucene/facet/taxonomy/directory/TestDirectoryTaxonomyWriter.java

  • Committer: Sebastian Meyer
  • Date: 2012-08-03 09:12:40 UTC
  • Revision ID: sebastian.meyer@slub-dresden.de-20120803091240-x6861b0vabq1xror
https://launchpad.net/bugs/985487
Remove Lucene and Solr source code and add patches instead
Fix Bug #985487: Auto-suggestion for the search interface

Show diffs side-by-side

added added

removed removed

Lines of Context:
lucene/contrib/facet/src/test/org/apache/lucene/facet/taxonomy/directory/TestDirectoryTaxonomyWriter.java
1
 
package org.apache.lucene.facet.taxonomy.directory;
2
 
 
3
 
import java.util.HashMap;
4
 
import java.util.Map;
5
 
 
6
 
import org.apache.lucene.index.IndexReader;
7
 
import org.apache.lucene.index.IndexWriterConfig.OpenMode;
8
 
import org.apache.lucene.store.AlreadyClosedException;
9
 
import org.apache.lucene.store.Directory;
10
 
import org.junit.Test;
11
 
 
12
 
import org.apache.lucene.util.LuceneTestCase;
13
 
import org.apache.lucene.facet.taxonomy.CategoryPath;
14
 
import org.apache.lucene.facet.taxonomy.directory.DirectoryTaxonomyWriter;
15
 
import org.apache.lucene.facet.taxonomy.writercache.TaxonomyWriterCache;
16
 
 
17
 
/**
18
 
 * Licensed to the Apache Software Foundation (ASF) under one or more
19
 
 * contributor license agreements.  See the NOTICE file distributed with
20
 
 * this work for additional information regarding copyright ownership.
21
 
 * The ASF licenses this file to You under the Apache License, Version 2.0
22
 
 * (the "License"); you may not use this file except in compliance with
23
 
 * the License.  You may obtain a copy of the License at
24
 
 *
25
 
 *     http://www.apache.org/licenses/LICENSE-2.0
26
 
 *
27
 
 * Unless required by applicable law or agreed to in writing, software
28
 
 * distributed under the License is distributed on an "AS IS" BASIS,
29
 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
30
 
 * See the License for the specific language governing permissions and
31
 
 * limitations under the License.
32
 
 */
33
 
 
34
 
public class TestDirectoryTaxonomyWriter extends LuceneTestCase {
35
 
 
36
 
  // A No-Op TaxonomyWriterCache which always discards all given categories, and
37
 
  // always returns true in put(), to indicate some cache entries were cleared.
38
 
  private static class NoOpCache implements TaxonomyWriterCache {
39
 
 
40
 
    NoOpCache() { }
41
 
    
42
 
    public void close() {}
43
 
    public int get(CategoryPath categoryPath) { return -1; }
44
 
    public int get(CategoryPath categoryPath, int length) { return get(categoryPath); }
45
 
    public boolean put(CategoryPath categoryPath, int ordinal) { return true; }
46
 
    public boolean put(CategoryPath categoryPath, int prefixLen, int ordinal) { return true; }
47
 
    public boolean hasRoom(int numberOfEntries) { return false; }
48
 
    
49
 
  }
50
 
  
51
 
  @Test
52
 
  public void testCommit() throws Exception {
53
 
    // Verifies that nothing is committed to the underlying Directory, if
54
 
    // commit() wasn't called.
55
 
    Directory dir = newDirectory();
56
 
    DirectoryTaxonomyWriter ltw = new DirectoryTaxonomyWriter(dir, OpenMode.CREATE_OR_APPEND, new NoOpCache());
57
 
    assertFalse(IndexReader.indexExists(dir));
58
 
    ltw.commit(); // first commit, so that an index will be created
59
 
    ltw.addCategory(new CategoryPath("a"));
60
 
    
61
 
    IndexReader r = IndexReader.open(dir);
62
 
    assertEquals("No categories should have been committed to the underlying directory", 1, r.numDocs());
63
 
    r.close();
64
 
    ltw.close();
65
 
    dir.close();
66
 
  }
67
 
  
68
 
  @Test
69
 
  public void testCommitUserData() throws Exception {
70
 
    // Verifies that committed data is retrievable
71
 
    Directory dir = newDirectory();
72
 
    DirectoryTaxonomyWriter ltw = new DirectoryTaxonomyWriter(dir, OpenMode.CREATE_OR_APPEND, new NoOpCache());
73
 
    assertFalse(IndexReader.indexExists(dir));
74
 
    ltw.commit(); // first commit, so that an index will be created
75
 
    ltw.addCategory(new CategoryPath("a"));
76
 
    ltw.addCategory(new CategoryPath("b"));
77
 
    Map <String, String> userCommitData = new HashMap<String, String>();
78
 
    userCommitData.put("testing", "1 2 3");
79
 
    ltw.commit(userCommitData);
80
 
    ltw.close();
81
 
    IndexReader r = IndexReader.open(dir);
82
 
    assertEquals("2 categories plus root should have been committed to the underlying directory", 3, r.numDocs());
83
 
    Map <String, String> readUserCommitData = r.getCommitUserData();
84
 
    assertTrue("wrong value extracted from commit data", 
85
 
        "1 2 3".equals(readUserCommitData.get("testing")));
86
 
    r.close();
87
 
    dir.close();
88
 
  }
89
 
  
90
 
  @Test
91
 
  public void testRollback() throws Exception {
92
 
    // Verifies that if callback is called, DTW is closed.
93
 
    Directory dir = newDirectory();
94
 
    DirectoryTaxonomyWriter dtw = new DirectoryTaxonomyWriter(dir);
95
 
    dtw.addCategory(new CategoryPath("a"));
96
 
    dtw.rollback();
97
 
    try {
98
 
      dtw.addCategory(new CategoryPath("a"));
99
 
      fail("should not have succeeded to add a category following rollback.");
100
 
    } catch (AlreadyClosedException e) {
101
 
      // expected
102
 
    }
103
 
    dir.close();
104
 
  }
105
 
  
106
 
  @Test
107
 
  public void testEnsureOpen() throws Exception {
108
 
    // verifies that an exception is thrown if DTW was closed
109
 
    Directory dir = newDirectory();
110
 
    DirectoryTaxonomyWriter dtw = new DirectoryTaxonomyWriter(dir);
111
 
    dtw.close();
112
 
    try {
113
 
      dtw.addCategory(new CategoryPath("a"));
114
 
      fail("should not have succeeded to add a category following close.");
115
 
    } catch (AlreadyClosedException e) {
116
 
      // expected
117
 
    }
118
 
    dir.close();
119
 
  }
120
 
  
121
 
}