/* * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You 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. */ using System; using NUnit.Framework; //using TestRunner = junit.textui.TestRunner; using Document = Lucene.Net.Documents.Document; using Field = Lucene.Net.Documents.Field; using FieldOption = Lucene.Net.Index.IndexReader.FieldOption; using Lucene.Net.Store; using WhitespaceAnalyzer = Lucene.Net.Analysis.WhitespaceAnalyzer; using StandardAnalyzer = Lucene.Net.Analysis.Standard.StandardAnalyzer; using Hits = Lucene.Net.Search.Hits; using IndexSearcher = Lucene.Net.Search.IndexSearcher; using TermQuery = Lucene.Net.Search.TermQuery; using LuceneTestCase = Lucene.Net.Util.LuceneTestCase; using _TestUtil = Lucene.Net.Util._TestUtil; namespace Lucene.Net.Index { [TestFixture] public class TestIndexReader : LuceneTestCase { /// Main for running test case by itself. [STAThread] public static void Main(System.String[] args) { // NUnit.Core.TestRunner(new NUnit.Core.TestSuite(typeof(TestIndexReader))); // {{Aroush}} where is 'TestRunner'? // TestRunner.run (new TestIndexReader("testBasicDelete")); // TestRunner.run (new TestIndexReader("testDeleteReaderWriterConflict")); // TestRunner.run (new TestIndexReader("testDeleteReaderReaderConflict")); // TestRunner.run (new TestIndexReader("testFilesOpenClose")); } // public TestIndexReader(System.String name) // { // } public virtual void TestIsCurrent() { RAMDirectory d = new MockRAMDirectory(); IndexWriter writer = new IndexWriter(d, new StandardAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); // set up reader: IndexReader reader = IndexReader.Open(d); Assert.IsTrue(reader.IsCurrent()); // modify index by adding another document: writer = new IndexWriter(d, new StandardAnalyzer(), false); AddDocumentWithFields(writer); writer.Close(); Assert.IsFalse(reader.IsCurrent()); // re-create index: writer = new IndexWriter(d, new StandardAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); Assert.IsFalse(reader.IsCurrent()); reader.Close(); d.Close(); } /// Tests the IndexReader.getFieldNames implementation /// Exception on error public virtual void TestGetFieldNames() { RAMDirectory d = new MockRAMDirectory(); // set up writer IndexWriter writer = new IndexWriter(d, new StandardAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); // set up reader IndexReader reader = IndexReader.Open(d); System.Collections.ICollection fieldNames = reader.GetFieldNames(IndexReader.FieldOption.ALL); Assert.IsTrue(CollectionContains(fieldNames, "keyword")); Assert.IsTrue(CollectionContains(fieldNames, "text")); Assert.IsTrue(CollectionContains(fieldNames, "unindexed")); Assert.IsTrue(CollectionContains(fieldNames, "unstored")); reader.Close(); // add more documents writer = new IndexWriter(d, new StandardAnalyzer(), false); // want to get some more segments here for (int i = 0; i < 5 * writer.GetMergeFactor(); i++) { AddDocumentWithFields(writer); } // new fields are in some different segments (we hope) for (int i = 0; i < 5 * writer.GetMergeFactor(); i++) { AddDocumentWithDifferentFields(writer); } // new termvector fields for (int i = 0; i < 5 * writer.GetMergeFactor(); i++) { AddDocumentWithTermVectorFields(writer); } writer.Close(); // verify fields again reader = IndexReader.Open(d); fieldNames = reader.GetFieldNames(IndexReader.FieldOption.ALL); Assert.AreEqual(13, fieldNames.Count); // the following fields Assert.IsTrue(CollectionContains(fieldNames, "keyword")); Assert.IsTrue(CollectionContains(fieldNames, "text")); Assert.IsTrue(CollectionContains(fieldNames, "unindexed")); Assert.IsTrue(CollectionContains(fieldNames, "unstored")); Assert.IsTrue(CollectionContains(fieldNames, "keyword2")); Assert.IsTrue(CollectionContains(fieldNames, "text2")); Assert.IsTrue(CollectionContains(fieldNames, "unindexed2")); Assert.IsTrue(CollectionContains(fieldNames, "unstored2")); Assert.IsTrue(CollectionContains(fieldNames, "tvnot")); Assert.IsTrue(CollectionContains(fieldNames, "termvector")); Assert.IsTrue(CollectionContains(fieldNames, "tvposition")); Assert.IsTrue(CollectionContains(fieldNames, "tvoffset")); Assert.IsTrue(CollectionContains(fieldNames, "tvpositionoffset")); // verify that only indexed fields were returned fieldNames = reader.GetFieldNames(IndexReader.FieldOption.INDEXED); Assert.AreEqual(11, fieldNames.Count); // 6 original + the 5 termvector fields Assert.IsTrue(CollectionContains(fieldNames, "keyword")); Assert.IsTrue(CollectionContains(fieldNames, "text")); Assert.IsTrue(CollectionContains(fieldNames, "unstored")); Assert.IsTrue(CollectionContains(fieldNames, "keyword2")); Assert.IsTrue(CollectionContains(fieldNames, "text2")); Assert.IsTrue(CollectionContains(fieldNames, "unstored2")); Assert.IsTrue(CollectionContains(fieldNames, "tvnot")); Assert.IsTrue(CollectionContains(fieldNames, "termvector")); Assert.IsTrue(CollectionContains(fieldNames, "tvposition")); Assert.IsTrue(CollectionContains(fieldNames, "tvoffset")); Assert.IsTrue(CollectionContains(fieldNames, "tvpositionoffset")); // verify that only unindexed fields were returned fieldNames = reader.GetFieldNames(IndexReader.FieldOption.UNINDEXED); Assert.AreEqual(2, fieldNames.Count); // the following fields Assert.IsTrue(CollectionContains(fieldNames, "unindexed")); Assert.IsTrue(CollectionContains(fieldNames, "unindexed2")); // verify index term vector fields fieldNames = reader.GetFieldNames(IndexReader.FieldOption.TERMVECTOR); Assert.AreEqual(1, fieldNames.Count); // 1 field has term vector only Assert.IsTrue(CollectionContains(fieldNames, "termvector")); fieldNames = reader.GetFieldNames(IndexReader.FieldOption.TERMVECTOR_WITH_POSITION); Assert.AreEqual(1, fieldNames.Count); // 4 fields are indexed with term vectors Assert.IsTrue(CollectionContains(fieldNames, "tvposition")); fieldNames = reader.GetFieldNames(IndexReader.FieldOption.TERMVECTOR_WITH_OFFSET); Assert.AreEqual(1, fieldNames.Count); // 4 fields are indexed with term vectors Assert.IsTrue(CollectionContains(fieldNames, "tvoffset")); fieldNames = reader.GetFieldNames(IndexReader.FieldOption.TERMVECTOR_WITH_POSITION_OFFSET); Assert.AreEqual(1, fieldNames.Count); // 4 fields are indexed with term vectors Assert.IsTrue(CollectionContains(fieldNames, "tvpositionoffset")); reader.Close(); d.Close(); } [Test] public virtual void TestTermVectors() { RAMDirectory d = new MockRAMDirectory(); // set up writer IndexWriter writer = new IndexWriter(d, new StandardAnalyzer(), true); // want to get some more segments here // new termvector fields for (int i = 0; i < 5 * writer.GetMergeFactor(); i++) { Document doc = new Document(); doc.Add(new Field("tvnot", "one two two three three three", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.NO)); doc.Add(new Field("termvector", "one two two three three three", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.YES)); doc.Add(new Field("tvoffset", "one two two three three three", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_OFFSETS)); doc.Add(new Field("tvposition", "one two two three three three", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_POSITIONS)); doc.Add(new Field("tvpositionoffset", "one two two three three three", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_POSITIONS_OFFSETS)); writer.AddDocument(doc); } writer.Close(); IndexReader reader = IndexReader.Open(d); FieldSortedTermVectorMapper mapper = new FieldSortedTermVectorMapper(new TermVectorEntryFreqSortedComparator()); reader.GetTermFreqVector(0, mapper); System.Collections.IDictionary map = mapper.GetFieldToTerms(); Assert.IsTrue(map != null, "map is null and it shouldn't be"); Assert.IsTrue(map.Count == 4, "map Size: " + map.Count + " is not: " + 4); System.Collections.IDictionary set_Renamed = (System.Collections.IDictionary) map["termvector"]; for (System.Collections.IEnumerator iterator = set_Renamed.Keys.GetEnumerator(); iterator.MoveNext(); ) { TermVectorEntry entry = (TermVectorEntry) iterator.Current; Assert.IsTrue(entry != null, "entry is null and it shouldn't be"); System.Console.Out.WriteLine("Entry: " + entry); } } private void AssertTermDocsCount(System.String msg, IndexReader reader, Term term, int expected) { TermDocs tdocs = null; try { tdocs = reader.TermDocs(term); Assert.IsNotNull(tdocs, msg + ", null TermDocs"); int count = 0; while (tdocs.Next()) { count++; } Assert.AreEqual(expected, count, msg + ", count mismatch"); } finally { if (tdocs != null) tdocs.Close(); } } [Test] public virtual void TestBasicDelete() { Directory dir = new MockRAMDirectory(); IndexWriter writer = null; IndexReader reader = null; Term searchTerm = new Term("content", "aaa"); // add 100 documents with term : aaa writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 100; i++) { AddDoc(writer, searchTerm.Text()); } writer.Close(); // OPEN READER AT THIS POINT - this should fix the view of the // index at the point of having 100 "aaa" documents and 0 "bbb" reader = IndexReader.Open(dir); Assert.AreEqual(100, reader.DocFreq(searchTerm), "first docFreq"); AssertTermDocsCount("first reader", reader, searchTerm, 100); reader.Close(); // DELETE DOCUMENTS CONTAINING TERM: aaa int deleted = 0; reader = IndexReader.Open(dir); deleted = reader.DeleteDocuments(searchTerm); Assert.AreEqual(100, deleted, "deleted count"); Assert.AreEqual(100, reader.DocFreq(searchTerm), "deleted docFreq"); AssertTermDocsCount("deleted termDocs", reader, searchTerm, 0); // open a 2nd reader to make sure first reader can // commit its changes (.del) while second reader // is open: IndexReader reader2 = IndexReader.Open(dir); reader.Close(); // CREATE A NEW READER and re-test reader = IndexReader.Open(dir); Assert.AreEqual(100, reader.DocFreq(searchTerm), "deleted docFreq"); AssertTermDocsCount("deleted termDocs", reader, searchTerm, 0); reader.Close(); reader2.Close(); dir.Close(); } // Make sure attempts to make changes after reader is // closed throws IOException: [Test] public virtual void TestChangesAfterClose() { Directory dir = new RAMDirectory(); IndexWriter writer = null; IndexReader reader = null; Term searchTerm = new Term("content", "aaa"); // add 11 documents with term : aaa writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 11; i++) { AddDoc(writer, searchTerm.Text()); } writer.Close(); reader = IndexReader.Open(dir); // Close reader: reader.Close(); // Then, try to make changes: try { reader.DeleteDocument(4); Assert.Fail("deleteDocument after close failed to throw IOException"); } catch (AlreadyClosedException) { // expected } try { reader.SetNorm(5, "aaa", 2.0f); Assert.Fail("setNorm after close failed to throw IOException"); } catch (AlreadyClosedException) { // expected } try { reader.UndeleteAll(); Assert.Fail("undeleteAll after close failed to throw IOException"); } catch (AlreadyClosedException) { // expected } } // Make sure we get lock obtain failed exception with 2 writers: [Test] public virtual void TestLockObtainFailed() { Directory dir = new RAMDirectory(); IndexWriter writer = null; IndexReader reader = null; Term searchTerm = new Term("content", "aaa"); // add 11 documents with term : aaa writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 11; i++) { AddDoc(writer, searchTerm.Text()); } // Create reader: reader = IndexReader.Open(dir); // Try to make changes try { reader.DeleteDocument(4); Assert.Fail("deleteDocument should have hit LockObtainFailedException"); } catch (LockObtainFailedException) { // expected } try { reader.SetNorm(5, "aaa", 2.0f); Assert.Fail("setNorm should have hit LockObtainFailedException"); } catch (LockObtainFailedException) { // expected } try { reader.UndeleteAll(); Assert.Fail("undeleteAll should have hit LockObtainFailedException"); } catch (LockObtainFailedException) { // expected } writer.Close(); reader.Close(); } // Make sure you can set norms & commit even if a reader // is open against the index: [Test] public virtual void TestWritingNorms() { System.String tempDir = SupportClass.AppSettings.Get("tempDir", ""); if (tempDir == null) throw new System.IO.IOException("tempDir undefined, cannot run test"); System.IO.FileInfo indexDir = new System.IO.FileInfo(tempDir + "\\" + "lucenetestnormwriter"); Directory dir = FSDirectory.GetDirectory(indexDir); IndexWriter writer; IndexReader reader; Term searchTerm = new Term("content", "aaa"); // add 1 documents with term : aaa writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDoc(writer, searchTerm.Text()); writer.Close(); // now open reader & set norm for doc 0 reader = IndexReader.Open(dir); reader.SetNorm(0, "content", (float) 2.0); // we should be holding the write lock now: Assert.IsTrue(IndexReader.IsLocked(dir), "locked"); reader.Commit(); // we should not be holding the write lock now: Assert.IsTrue(!IndexReader.IsLocked(dir), "not locked"); // open a 2nd reader: IndexReader reader2 = IndexReader.Open(dir); // set norm again for doc 0 reader.SetNorm(0, "content", (float) 3.0); Assert.IsTrue(IndexReader.IsLocked(dir), "locked"); reader.Close(); // we should not be holding the write lock now: Assert.IsTrue(!IndexReader.IsLocked(dir), "not locked"); reader2.Close(); dir.Close(); RmDir(indexDir); } // Make sure you can set norms & commit, and there are // no extra norms files left: [Test] public virtual void TestWritingNormsNoReader() { Directory dir = new MockRAMDirectory(); IndexWriter writer = null; IndexReader reader = null; Term searchTerm = new Term("content", "aaa"); // add 1 documents with term : aaa writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); writer.SetUseCompoundFile(false); AddDoc(writer, searchTerm.Text()); writer.Close(); // now open reader & set norm for doc 0 (writes to // _0_1.s0) reader = IndexReader.Open(dir); reader.SetNorm(0, "content", (float) 2.0); reader.Close(); // now open reader again & set norm for doc 0 (writes to _0_2.s0) reader = IndexReader.Open(dir); reader.SetNorm(0, "content", (float) 2.0); reader.Close(); Assert.IsFalse(dir.FileExists("_0_1.s0"), "failed to remove first generation norms file on writing second generation"); dir.Close(); } [Test] public virtual void TestDeleteReaderWriterConflictUnoptimized() { DeleteReaderWriterConflict(false); } [Test] public virtual void TestOpenEmptyDirectory() { System.String dirName = "test.empty"; System.IO.FileInfo fileDirName = new System.IO.FileInfo(dirName); bool tmpBool; if (System.IO.File.Exists(fileDirName.FullName)) tmpBool = true; else tmpBool = System.IO.Directory.Exists(fileDirName.FullName); if (!tmpBool) { System.IO.Directory.CreateDirectory(fileDirName.FullName); } try { IndexReader reader = IndexReader.Open(fileDirName); Assert.Fail("opening IndexReader on empty directory failed to produce FileNotFoundException"); } catch (System.IO.FileNotFoundException) { // GOOD } RmDir(fileDirName); } [Test] public virtual void TestDeleteReaderWriterConflictOptimized() { DeleteReaderWriterConflict(true); } private void DeleteReaderWriterConflict(bool optimize) { //Directory dir = new RAMDirectory(); Directory dir = GetDirectory(); Term searchTerm = new Term("content", "aaa"); Term searchTerm2 = new Term("content", "bbb"); // add 100 documents with term : aaa IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 100; i++) { AddDoc(writer, searchTerm.Text()); } writer.Close(); // OPEN READER AT THIS POINT - this should fix the view of the // index at the point of having 100 "aaa" documents and 0 "bbb" IndexReader reader = IndexReader.Open(dir); Assert.AreEqual(100, reader.DocFreq(searchTerm), "first docFreq"); Assert.AreEqual(0, reader.DocFreq(searchTerm2), "first docFreq"); AssertTermDocsCount("first reader", reader, searchTerm, 100); AssertTermDocsCount("first reader", reader, searchTerm2, 0); // add 100 documents with term : bbb writer = new IndexWriter(dir, new WhitespaceAnalyzer(), false); for (int i = 0; i < 100; i++) { AddDoc(writer, searchTerm2.Text()); } // REQUEST OPTIMIZATION // This causes a new segment to become current for all subsequent // searchers. Because of this, deletions made via a previously open // reader, which would be applied to that reader's segment, are lost // for subsequent searchers/readers if (optimize) writer.Optimize(); writer.Close(); // The reader should not see the new data Assert.AreEqual(100, reader.DocFreq(searchTerm), "first docFreq"); Assert.AreEqual(0, reader.DocFreq(searchTerm2), "first docFreq"); AssertTermDocsCount("first reader", reader, searchTerm, 100); AssertTermDocsCount("first reader", reader, searchTerm2, 0); // DELETE DOCUMENTS CONTAINING TERM: aaa // NOTE: the reader was created when only "aaa" documents were in int deleted = 0; try { deleted = reader.DeleteDocuments(searchTerm); Assert.Fail("Delete allowed on an index reader with stale segment information"); } catch (StaleReaderException) { /* success */ } // Re-open index reader and try again. This time it should see // the new data. reader.Close(); reader = IndexReader.Open(dir); Assert.AreEqual(100, reader.DocFreq(searchTerm), "first docFreq"); Assert.AreEqual(100, reader.DocFreq(searchTerm2), "first docFreq"); AssertTermDocsCount("first reader", reader, searchTerm, 100); AssertTermDocsCount("first reader", reader, searchTerm2, 100); deleted = reader.DeleteDocuments(searchTerm); Assert.AreEqual(100, deleted, "deleted count"); Assert.AreEqual(100, reader.DocFreq(searchTerm), "deleted docFreq"); Assert.AreEqual(100, reader.DocFreq(searchTerm2), "deleted docFreq"); AssertTermDocsCount("deleted termDocs", reader, searchTerm, 0); AssertTermDocsCount("deleted termDocs", reader, searchTerm2, 100); reader.Close(); // CREATE A NEW READER and re-test reader = IndexReader.Open(dir); Assert.AreEqual(100, reader.DocFreq(searchTerm), "deleted docFreq"); Assert.AreEqual(100, reader.DocFreq(searchTerm2), "deleted docFreq"); AssertTermDocsCount("deleted termDocs", reader, searchTerm, 0); AssertTermDocsCount("deleted termDocs", reader, searchTerm2, 100); reader.Close(); } private Directory GetDirectory() { return FSDirectory.GetDirectory(new System.IO.FileInfo(System.IO.Path.Combine(SupportClass.AppSettings.Get("tempDir", ""), "testIndex"))); } [Test] public virtual void TestFilesOpenClose() { // Create initial data set System.IO.FileInfo dirFile = new System.IO.FileInfo(System.IO.Path.Combine("tempDir", "testIndex")); Directory dir = GetDirectory(); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDoc(writer, "test"); writer.Close(); dir.Close(); // Try to erase the data - this ensures that the writer closed all files _TestUtil.RmDir(dirFile); dir = GetDirectory(); // Now create the data set again, just as before writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDoc(writer, "test"); writer.Close(); dir.Close(); // Now open existing directory and test that reader closes all files dir = GetDirectory(); IndexReader reader1 = IndexReader.Open(dir); reader1.Close(); dir.Close(); // The following will fail if reader did not Close // all files _TestUtil.RmDir(dirFile); } [Test] public virtual void TestLastModified() { Assert.IsFalse(IndexReader.IndexExists("there_is_no_such_index")); System.IO.FileInfo fileDir = new System.IO.FileInfo(System.IO.Path.Combine(SupportClass.AppSettings.Get("tempDir", ""), "testIndex")); // can't do the filesystem version of this test, as a system level process lock prevents deletion of the index file //for (int i = 0; i < 2; i++) for (int i = 0; i < 1; i++) { try { Directory dir; if (0 == i) dir = new MockRAMDirectory(); else dir = GetDirectory(); Assert.IsFalse(IndexReader.IndexExists(dir)); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); Assert.IsTrue(IndexReader.IsLocked(dir)); // writer open, so dir is locked writer.Close(); Assert.IsTrue(IndexReader.IndexExists(dir)); IndexReader reader = IndexReader.Open(dir); Assert.IsFalse(IndexReader.IsLocked(dir)); // reader only, no lock long version = IndexReader.LastModified(dir); if (i == 1) { long version2 = IndexReader.LastModified(fileDir); Assert.AreEqual(version, version2); } reader.Close(); // modify index and check version has been // incremented: while (true) { try { System.Threading.Thread.Sleep(new System.TimeSpan((System.Int64) 10000 * 1000)); break; } catch (System.Threading.ThreadInterruptedException) { SupportClass.ThreadClass.Current().Interrupt(); } } writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); reader = IndexReader.Open(dir); Assert.IsTrue( version <= IndexReader.LastModified(dir), "old lastModified is " + version + "; new lastModified is " + IndexReader.LastModified(dir) ); reader.Close(); dir.Close(); } finally { if (i == 1) _TestUtil.RmDir(fileDir); } } } [Test] public virtual void TestVersion() { Assert.IsFalse(IndexReader.IndexExists("there_is_no_such_index")); Directory dir = new MockRAMDirectory(); Assert.IsFalse(IndexReader.IndexExists(dir)); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); Assert.IsTrue(IndexReader.IsLocked(dir)); // writer open, so dir is locked writer.Close(); Assert.IsTrue(IndexReader.IndexExists(dir)); IndexReader reader = IndexReader.Open(dir); Assert.IsFalse(IndexReader.IsLocked(dir)); // reader only, no lock long version = IndexReader.GetCurrentVersion(dir); reader.Close(); // modify index and check version has been // incremented: writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); reader = IndexReader.Open(dir); Assert.IsTrue(version < IndexReader.GetCurrentVersion(dir), "old version is " + version + "; new version is " + IndexReader.GetCurrentVersion(dir)); reader.Close(); dir.Close(); } [Test] public virtual void TestLock() { Directory dir = new MockRAMDirectory(); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); writer.Close(); writer = new IndexWriter(dir, new WhitespaceAnalyzer(), false); IndexReader reader = IndexReader.Open(dir); try { reader.DeleteDocument(0); Assert.Fail("expected lock"); } catch (System.IO.IOException) { // expected exception } IndexReader.Unlock(dir); // this should not be done in the real world! reader.DeleteDocument(0); reader.Close(); writer.Close(); dir.Close(); } [Test] public virtual void TestUndeleteAll() { Directory dir = new MockRAMDirectory(); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); AddDocumentWithFields(writer); writer.Close(); IndexReader reader = IndexReader.Open(dir); reader.DeleteDocument(0); reader.DeleteDocument(1); reader.UndeleteAll(); reader.Close(); reader = IndexReader.Open(dir); Assert.AreEqual(2, reader.NumDocs()); // nothing has really been deleted thanks to undeleteAll() reader.Close(); dir.Close(); } [Test] public virtual void TestUndeleteAllAfterClose() { Directory dir = new RAMDirectory(); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); AddDocumentWithFields(writer); writer.Close(); IndexReader reader = IndexReader.Open(dir); reader.DeleteDocument(0); reader.DeleteDocument(1); reader.Close(); reader = IndexReader.Open(dir); reader.UndeleteAll(); Assert.AreEqual(2, reader.NumDocs()); // nothing has really been deleted thanks to undeleteAll() reader.Close(); dir.Close(); } [Test] public virtual void TestUndeleteAllAfterCloseThenReopen() { Directory dir = new MockRAMDirectory(); IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); AddDocumentWithFields(writer); AddDocumentWithFields(writer); writer.Close(); IndexReader reader = IndexReader.Open(dir); reader.DeleteDocument(0); reader.DeleteDocument(1); reader.Close(); reader = IndexReader.Open(dir); reader.UndeleteAll(); reader.Close(); reader = IndexReader.Open(dir); Assert.AreEqual(2, reader.NumDocs()); // nothing has really been deleted thanks to undeleteAll() reader.Close(); dir.Close(); } [Test] public virtual void TestDeleteReaderReaderConflictUnoptimized() { DeleteReaderReaderConflict(false); } [Test] public virtual void TestDeleteReaderReaderConflictOptimized() { DeleteReaderReaderConflict(true); } /// Make sure if reader tries to commit but hits disk /// full that reader remains consistent and usable. /// [Test] public virtual void TestDiskFull() { bool debug = false; Term searchTerm = new Term("content", "aaa"); int START_COUNT = 157; int END_COUNT = 144; // First build up a starting index: RAMDirectory startDir = new MockRAMDirectory(); IndexWriter writer = new IndexWriter(startDir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 157; i++) { Lucene.Net.Documents.Document d = new Lucene.Net.Documents.Document(); d.Add(new Field("id", System.Convert.ToString(i), Field.Store.YES, Field.Index.UN_TOKENIZED)); d.Add(new Field("content", "aaa " + i, Field.Store.NO, Field.Index.TOKENIZED)); writer.AddDocument(d); } writer.Close(); long diskUsage = startDir.SizeInBytes(); long diskFree = diskUsage + 100; System.IO.IOException err = null; bool done = false; // Iterate w/ ever increasing free disk space: while (!done) { MockRAMDirectory dir = new MockRAMDirectory(startDir); IndexReader reader = IndexReader.Open(dir); // For each disk size, first try to commit against // dir that will hit random IOExceptions & disk // full; after, give it infinite disk space & turn // off random IOExceptions & retry w/ same reader: bool success = false; for (int x = 0; x < 2; x++) { double rate = 0.05; double diskRatio = ((double) diskFree) / diskUsage; long thisDiskFree; System.String testName; if (0 == x) { thisDiskFree = diskFree; if (diskRatio >= 2.0) { rate /= 2; } if (diskRatio >= 4.0) { rate /= 2; } if (diskRatio >= 6.0) { rate = 0.0; } if (debug) { System.Console.Out.WriteLine("\ncycle: " + diskFree + " bytes"); } testName = "disk full during reader.Close() @ " + thisDiskFree + " bytes"; } else { thisDiskFree = 0; rate = 0.0; if (debug) { System.Console.Out.WriteLine("\ncycle: same writer: unlimited disk space"); } testName = "reader re-use after disk full"; } dir.SetMaxSizeInBytes(thisDiskFree); dir.SetRandomIOExceptionRate(rate, diskFree); try { if (0 == x) { int docId = 12; for (int i = 0; i < 13; i++) { reader.DeleteDocument(docId); reader.SetNorm(docId, "contents", (float) 2.0); docId += 12; } } reader.Close(); success = true; if (0 == x) { done = true; } } catch (System.IO.IOException e) { if (debug) { System.Console.Out.WriteLine(" hit IOException: " + e); } err = e; if (1 == x) { System.Console.Error.WriteLine(e.StackTrace); Assert.Fail(testName + " hit IOException after disk space was freed up"); } } // Whether we succeeded or failed, check that all // un-referenced files were in fact deleted (ie, // we did not create garbage). Just create a // new IndexFileDeleter, have it delete // unreferenced files, then verify that in fact // no files were deleted: System.String[] startFiles = dir.List(); SegmentInfos infos = new SegmentInfos(); infos.Read(dir); IndexFileDeleter d = new IndexFileDeleter(dir, new KeepOnlyLastCommitDeletionPolicy(), infos, null, null); System.String[] endFiles = dir.List(); System.Array.Sort(startFiles); System.Array.Sort(endFiles); //for(int i=0;i 0) { s += "\n "; } s += l[i]; } return s; } [Test] public virtual void TestOpenReaderAfterDelete() { System.IO.FileInfo dirFile = new System.IO.FileInfo(System.IO.Path.Combine(SupportClass.AppSettings.Get("tempDir", ""), "deletetest")); Directory dir = FSDirectory.GetDirectory(dirFile); try { IndexReader reader = IndexReader.Open(dir); Assert.Fail("expected FileNotFoundException"); } catch (System.IO.FileNotFoundException) { // expected } System.IO.Directory.Delete(dirFile.FullName); // Make sure we still get a CorruptIndexException (not NPE): try { IndexReader reader = IndexReader.Open(dir); Assert.Fail("expected FileNotFoundException"); } catch (System.IO.DirectoryNotFoundException) { // expected } } private void DeleteReaderReaderConflict(bool optimize) { Directory dir = GetDirectory(); Term searchTerm1 = new Term("content", "aaa"); Term searchTerm2 = new Term("content", "bbb"); Term searchTerm3 = new Term("content", "ccc"); // add 100 documents with term : aaa // add 100 documents with term : bbb // add 100 documents with term : ccc IndexWriter writer = new IndexWriter(dir, new WhitespaceAnalyzer(), true); for (int i = 0; i < 100; i++) { AddDoc(writer, searchTerm1.Text()); AddDoc(writer, searchTerm2.Text()); AddDoc(writer, searchTerm3.Text()); } if (optimize) writer.Optimize(); writer.Close(); // OPEN TWO READERS // Both readers get segment info as exists at this time IndexReader reader1 = IndexReader.Open(dir); Assert.AreEqual(100, reader1.DocFreq(searchTerm1), "first opened"); Assert.AreEqual(100, reader1.DocFreq(searchTerm2), "first opened"); Assert.AreEqual(100, reader1.DocFreq(searchTerm3), "first opened"); AssertTermDocsCount("first opened", reader1, searchTerm1, 100); AssertTermDocsCount("first opened", reader1, searchTerm2, 100); AssertTermDocsCount("first opened", reader1, searchTerm3, 100); IndexReader reader2 = IndexReader.Open(dir); Assert.AreEqual(100, reader2.DocFreq(searchTerm1), "first opened"); Assert.AreEqual(100, reader2.DocFreq(searchTerm2), "first opened"); Assert.AreEqual(100, reader2.DocFreq(searchTerm3), "first opened"); AssertTermDocsCount("first opened", reader2, searchTerm1, 100); AssertTermDocsCount("first opened", reader2, searchTerm2, 100); AssertTermDocsCount("first opened", reader2, searchTerm3, 100); // DELETE DOCS FROM READER 2 and CLOSE IT // delete documents containing term: aaa // when the reader is closed, the segment info is updated and // the first reader is now stale reader2.DeleteDocuments(searchTerm1); Assert.AreEqual(100, reader2.DocFreq(searchTerm1), "after delete 1"); Assert.AreEqual(100, reader2.DocFreq(searchTerm2), "after delete 1"); Assert.AreEqual(100, reader2.DocFreq(searchTerm3), "after delete 1"); AssertTermDocsCount("after delete 1", reader2, searchTerm1, 0); AssertTermDocsCount("after delete 1", reader2, searchTerm2, 100); AssertTermDocsCount("after delete 1", reader2, searchTerm3, 100); reader2.Close(); // Make sure reader 1 is unchanged since it was open earlier Assert.AreEqual(100, reader1.DocFreq(searchTerm1), "after delete 1"); Assert.AreEqual(100, reader1.DocFreq(searchTerm2), "after delete 1"); Assert.AreEqual(100, reader1.DocFreq(searchTerm3), "after delete 1"); AssertTermDocsCount("after delete 1", reader1, searchTerm1, 100); AssertTermDocsCount("after delete 1", reader1, searchTerm2, 100); AssertTermDocsCount("after delete 1", reader1, searchTerm3, 100); // ATTEMPT TO DELETE FROM STALE READER // delete documents containing term: bbb try { reader1.DeleteDocuments(searchTerm2); Assert.Fail("Delete allowed from a stale index reader"); } catch (System.IO.IOException) { /* success */ } // RECREATE READER AND TRY AGAIN reader1.Close(); reader1 = IndexReader.Open(dir); Assert.AreEqual(100, reader1.DocFreq(searchTerm1), "reopened"); Assert.AreEqual(100, reader1.DocFreq(searchTerm2), "reopened"); Assert.AreEqual(100, reader1.DocFreq(searchTerm3), "reopened"); AssertTermDocsCount("reopened", reader1, searchTerm1, 0); AssertTermDocsCount("reopened", reader1, searchTerm2, 100); AssertTermDocsCount("reopened", reader1, searchTerm3, 100); reader1.DeleteDocuments(searchTerm2); Assert.AreEqual(100, reader1.DocFreq(searchTerm1), "deleted 2"); Assert.AreEqual(100, reader1.DocFreq(searchTerm2), "deleted 2"); Assert.AreEqual(100, reader1.DocFreq(searchTerm3), "deleted 2"); AssertTermDocsCount("deleted 2", reader1, searchTerm1, 0); AssertTermDocsCount("deleted 2", reader1, searchTerm2, 0); AssertTermDocsCount("deleted 2", reader1, searchTerm3, 100); reader1.Close(); // Open another reader to confirm that everything is deleted reader2 = IndexReader.Open(dir); Assert.AreEqual(100, reader2.DocFreq(searchTerm1), "reopened 2"); Assert.AreEqual(100, reader2.DocFreq(searchTerm2), "reopened 2"); Assert.AreEqual(100, reader2.DocFreq(searchTerm3), "reopened 2"); AssertTermDocsCount("reopened 2", reader2, searchTerm1, 0); AssertTermDocsCount("reopened 2", reader2, searchTerm2, 0); AssertTermDocsCount("reopened 2", reader2, searchTerm3, 100); reader2.Close(); dir.Close(); } private void AddDocumentWithFields(IndexWriter writer) { Lucene.Net.Documents.Document doc = new Lucene.Net.Documents.Document(); doc.Add(new Field("keyword", "test1", Field.Store.YES, Field.Index.UN_TOKENIZED)); doc.Add(new Field("text", "test1", Field.Store.YES, Field.Index.TOKENIZED)); doc.Add(new Field("unindexed", "test1", Field.Store.YES, Field.Index.NO)); doc.Add(new Field("unstored", "test1", Field.Store.NO, Field.Index.TOKENIZED)); writer.AddDocument(doc); } private void AddDocumentWithDifferentFields(IndexWriter writer) { Lucene.Net.Documents.Document doc = new Lucene.Net.Documents.Document(); doc.Add(new Field("keyword2", "test1", Field.Store.YES, Field.Index.UN_TOKENIZED)); doc.Add(new Field("text2", "test1", Field.Store.YES, Field.Index.TOKENIZED)); doc.Add(new Field("unindexed2", "test1", Field.Store.YES, Field.Index.NO)); doc.Add(new Field("unstored2", "test1", Field.Store.NO, Field.Index.TOKENIZED)); writer.AddDocument(doc); } private void AddDocumentWithTermVectorFields(IndexWriter writer) { Lucene.Net.Documents.Document doc = new Lucene.Net.Documents.Document(); doc.Add(new Field("tvnot", "tvnot", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.NO)); doc.Add(new Field("termvector", "termvector", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.YES)); doc.Add(new Field("tvoffset", "tvoffset", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_OFFSETS)); doc.Add(new Field("tvposition", "tvposition", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_POSITIONS)); doc.Add(new Field("tvpositionoffset", "tvpositionoffset", Field.Store.YES, Field.Index.TOKENIZED, Field.TermVector.WITH_POSITIONS_OFFSETS)); writer.AddDocument(doc); } private void AddDoc(IndexWriter writer, System.String value_Renamed) { Lucene.Net.Documents.Document doc = new Lucene.Net.Documents.Document(); doc.Add(new Field("content", value_Renamed, Field.Store.NO, Field.Index.TOKENIZED)); writer.AddDocument(doc); } private void RmDir(System.IO.FileInfo dir) { System.IO.FileInfo[] files = SupportClass.FileSupport.GetFiles(dir); for (int i = 0; i < files.Length; i++) { bool tmpBool; if (System.IO.File.Exists(files[i].FullName)) { System.IO.File.Delete(files[i].FullName); tmpBool = true; } else if (System.IO.Directory.Exists(files[i].FullName)) { System.IO.Directory.Delete(files[i].FullName); tmpBool = true; } else tmpBool = false; bool generatedAux = tmpBool; } bool tmpBool2; if (System.IO.File.Exists(dir.FullName)) { System.IO.File.Delete(dir.FullName); tmpBool2 = true; } else if (System.IO.Directory.Exists(dir.FullName)) { System.IO.Directory.Delete(dir.FullName); tmpBool2 = true; } else tmpBool2 = false; bool generatedAux2 = tmpBool2; } public static void AssertIndexEquals(IndexReader index1, IndexReader index2) { Assert.AreEqual(index1.NumDocs(), index2.NumDocs(), "IndexReaders have different values for numDocs."); Assert.AreEqual(index1.MaxDoc(), index2.MaxDoc(), "IndexReaders have different values for maxDoc."); Assert.AreEqual(index1.HasDeletions(), index2.HasDeletions(), "Only one IndexReader has deletions."); Assert.AreEqual(index1.IsOptimized(), index2.IsOptimized(), "Only one index is optimized."); // check field names System.Collections.ICollection fields1 = index1.GetFieldNames(FieldOption.ALL); System.Collections.ICollection fields2 = index2.GetFieldNames(FieldOption.ALL); Assert.AreEqual(fields1.Count, fields2.Count, "IndexReaders have different numbers of fields."); System.Collections.IEnumerator it1 = fields1.GetEnumerator(); System.Collections.IEnumerator it2 = fields2.GetEnumerator(); while (it1.MoveNext()) { Assert.IsTrue(it2.MoveNext()); Assert.AreEqual((System.String) it1.Current, (System.String) it2.Current, "Different field names."); } // check norms it1 = fields1.GetEnumerator(); while (it1.MoveNext()) { System.String curField = (System.String) it1.Current; byte[] norms1 = index1.Norms(curField); byte[] norms2 = index2.Norms(curField); Assert.AreEqual(norms1.Length, norms2.Length); for (int i = 0; i < norms1.Length; i++) { Assert.AreEqual(norms1[i], norms2[i], "Norm different for doc " + i + " and field '" + curField + "'."); } } // check deletions for (int i = 0; i < index1.MaxDoc(); i++) { Assert.AreEqual(index1.IsDeleted(i), index2.IsDeleted(i), "Doc " + i + " only deleted in one index."); } // check stored fields for (int i = 0; i < index1.MaxDoc(); i++) { if (!index1.IsDeleted(i)) { Document doc1 = index1.Document(i); Document doc2 = index2.Document(i); fields1 = doc1.GetFields(); fields2 = doc2.GetFields(); Assert.AreEqual(fields1.Count, fields2.Count, "Different numbers of fields for doc " + i + "."); it1 = fields1.GetEnumerator(); it2 = fields2.GetEnumerator(); while (it1.MoveNext()) { Assert.IsTrue(it2.MoveNext()); Field curField1 = (Field) it1.Current; Field curField2 = (Field) it2.Current; Assert.AreEqual(curField1.Name(), curField2.Name(), "Different fields names for doc " + i + "."); Assert.AreEqual(curField1.StringValue(), curField2.StringValue(), "Different field values for doc " + i + "."); } } } // check dictionary and posting lists TermEnum enum1 = index1.Terms(); TermEnum enum2 = index2.Terms(); TermPositions tp1 = index1.TermPositions(); TermPositions tp2 = index2.TermPositions(); while (enum1.Next()) { Assert.IsTrue(enum2.Next()); Assert.AreEqual(enum1.Term(), enum2.Term(), "Different term in dictionary."); tp1.Seek(enum1.Term()); tp2.Seek(enum1.Term()); while (tp1.Next()) { Assert.IsTrue(tp2.Next()); Assert.AreEqual(tp1.Doc(), tp2.Doc(), "Different doc id in postinglist of term " + enum1.Term() + "."); Assert.AreEqual(tp1.Freq(), tp2.Freq(), "Different term frequence in postinglist of term " + enum1.Term() + "."); for (int i = 0; i < tp1.Freq(); i++) { Assert.AreEqual(tp1.NextPosition(), tp2.NextPosition(), "Different positions in postinglist of term " + enum1.Term() + "."); } } } } public static bool CollectionContains(System.Collections.ICollection col, System.String val) { for (System.Collections.IEnumerator iterator = col.GetEnumerator(); iterator.MoveNext(); ) { System.Collections.DictionaryEntry fi = (System.Collections.DictionaryEntry)iterator.Current; System.String s = fi.Key.ToString(); if (s == val) return true; } return false; } } }