internal virtual void AssertQuery(Query query, Filter filter, Sort sort)
{
int size = TestUtil.NextInt(Random(), 1, searcher.IndexReader.MaxDoc / 5);
TopDocs expected = searcher.Search(query, filter, size, sort, Random().NextBoolean
(), Random().NextBoolean());
// make our actual sort, mutating original by replacing some of the
// sortfields with equivalent expressions
SortField[] original = sort.GetSort();
SortField[] mutated = new SortField[original.Length];
for (int i = 0; i < mutated.Length; i++)
{
if (Random().Next(3) > 0)
{
SortField s = original[i];
Expression expr = JavascriptCompiler.Compile(s.Field);
SimpleBindings simpleBindings = new SimpleBindings();
simpleBindings.Add(s);
bool reverse = s.Type == SortField.Type_e.SCORE || s.Reverse;
mutated[i] = expr.GetSortField(simpleBindings, reverse);
}
else
{
mutated[i] = original[i];
}
}
Sort mutatedSort = new Sort(mutated);
TopDocs actual = searcher.Search(query, filter, size, mutatedSort, Random().NextBoolean
(), Random().NextBoolean());
CheckHits.CheckEqual(query, expected.ScoreDocs, actual.ScoreDocs);
if (size < actual.TotalHits)
{
expected = searcher.SearchAfter(expected.ScoreDocs[size - 1], query, filter, size
, sort);
actual = searcher.SearchAfter(actual.ScoreDocs[size - 1], query, filter, size, mutatedSort
);
CheckHits.CheckEqual(query, expected.ScoreDocs, actual.ScoreDocs);
}
}
private IEnumerable <long> findContentIds(Query query)
{
const int batchSize = 256;
// ReSharper disable once InconsistentlySynchronizedField
using (var reader = _indexWriter.GetReader(applyAllDeletes: true))
{
var searcher = new IndexSearcher(reader);
TopDocs searchResult = searcher.Search(query, batchSize);
int count = 0;
while (searchResult.ScoreDocs.Length > 0)
{
foreach (var scoreDoc in searchResult.ScoreDocs)
{
if (count >= MaxResultCount)
{
yield break;
}
count++;
var doc = searcher.Doc(scoreDoc.Doc);
long contentId = doc.GetField <StoredField>(IdFieldName).GetInt64Value().Value;
yield return(contentId);
}
if (searchResult.ScoreDocs.Length < batchSize)
{
break;
}
var lastDoc = searchResult.ScoreDocs[searchResult.ScoreDocs.Length - 1];
searchResult = searcher.SearchAfter(lastDoc, query, batchSize);
}
}
}
public void TestSearchAfterWhenSortingByFunctionValues()
{
Directory dir = NewDirectory();
IndexWriterConfig iwc = NewIndexWriterConfig(TEST_VERSION_CURRENT, null);
iwc.SetMergePolicy(NewLogMergePolicy()); // depends on docid order
RandomIndexWriter writer = new RandomIndexWriter(Random, dir, iwc);
Document doc = new Document();
Field field = new StringField("value", "", Field.Store.YES);
doc.Add(field);
// Save docs unsorted (decreasing value n, n-1, ...)
const int NUM_VALS = 5;
for (int val = NUM_VALS; val > 0; val--)
{
field.SetStringValue(Convert.ToString(val));
writer.AddDocument(doc);
}
// Open index
IndexReader reader = writer.GetReader();
writer.Dispose();
IndexSearcher searcher = NewSearcher(reader);
// Get ValueSource from FieldCache
Int32FieldSource src = new Int32FieldSource("value");
// ...and make it a sort criterion
SortField sf = src.GetSortField(false).Rewrite(searcher);
Sort orderBy = new Sort(sf);
// Get hits sorted by our FunctionValues (ascending values)
Query q = new MatchAllDocsQuery();
TopDocs hits = searcher.Search(q, reader.MaxDoc, orderBy);
assertEquals(NUM_VALS, hits.ScoreDocs.Length);
// Verify that sorting works in general
int i = 0;
foreach (ScoreDoc hit in hits.ScoreDocs)
{
int valueFromDoc = Convert.ToInt32(reader.Document(hit.Doc).Get("value"));
assertEquals(++i, valueFromDoc);
}
// Now get hits after hit #2 using IS.searchAfter()
int afterIdx = 1;
FieldDoc afterHit = (FieldDoc)hits.ScoreDocs[afterIdx];
hits = searcher.SearchAfter(afterHit, q, reader.MaxDoc, orderBy);
// Expected # of hits: NUM_VALS - 2
assertEquals(NUM_VALS - (afterIdx + 1), hits.ScoreDocs.Length);
// Verify that hits are actually "after"
int afterValue = (int)((double?)afterHit.Fields[0]);
foreach (ScoreDoc hit in hits.ScoreDocs)
{
int val = Convert.ToInt32(reader.Document(hit.Doc).Get("value"));
assertTrue(afterValue <= val);
assertFalse(hit.Doc == afterHit.Doc);
}
reader.Dispose();
dir.Dispose();
}
