perl/t/217-poly_lexicon.t

# 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.

use strict;
use warnings;

use lib 'buildlib';
use Lucy::Test;

use strict;
use warnings;

package MyArchitecture;
use base qw( Lucy::Plan::Architecture );

sub index_interval { confess("should be displaced via local() below") }

package MySchema;
use base qw( Lucy::Plan::Schema );

sub new {
    my $self = shift->SUPER::new(@_);
    my $type = Lucy::Plan::StringType->new( sortable => 1 );
    $self->spec_field( name => 'content', type => $type );
    return $self;
}

sub architecture { MyArchitecture->new }

package main;
use Test::More tests => 35;
use File::Spec::Functions qw( catfile );

my @docs = ( ( ("a") x 100 ), 'j', "Foo" );
my @chars = ( 'a' .. 'z' );
for ( 0 .. 100 ) {
    my $content = '';
    for my $num_chars ( 1 .. int( rand(10) + 1 ) ) {
        $content .= @chars[ rand(@chars) ];
    }
    push @docs, "$content";
}

# Accumulate unique sorted terms.
my @correct = ();
for ( sort @docs ) {
    next if ( @correct and $_ eq $correct[-1] );
    push @correct, $_;
}

# Remember where 'j' exists in our 'correct' list.
my $correct_term_num = 0;
for (@correct) {
    last if $correct[$correct_term_num] eq 'j';
    $correct_term_num++;
}

for my $index_interval ( 1, 2, 3, 4, 7, 128, 1024 ) {

    no warnings 'redefine';
    local *MyArchitecture::index_interval = sub {$index_interval};

    my $folder = Lucy::Store::RAMFolder->new;
    my $schema = MySchema->new;

    my @docs_copy = @docs;
    while (@docs_copy) {
        my $indexer = Lucy::Index::Indexer->new(
            index  => $folder,
            schema => $schema,
        );
        my $docs_this_seg = int( rand(@docs_copy) );
        $docs_this_seg = 10 if $docs_this_seg < 10;

        for ( 0 .. 10 ) {
            last unless @docs_copy;
            my $tick = int( rand(@docs_copy) );
            $indexer->add_doc(
                { content => splice( @docs_copy, $tick, 1 ) } );
        }
        $indexer->commit;
    }

    my $reader = Lucy::Index::IndexReader->open( index => $folder, );

    my $lexicon = $reader->obtain("Lucy::Index::LexiconReader")
        ->lexicon( field => 'content' );
    isa_ok( $lexicon, "Lucy::Index::PolyLexicon" );

    $lexicon->next;
    is( $lexicon->get_term, $correct[0],
        "calling lexicon() without term returns Lexicon with iterator reset"
    );
    $lexicon->reset;

    my @got;
    while ( $lexicon->next ) {
        push @got, $lexicon->get_term;
    }

    is_deeply( \@got, \@correct,
        "correct order for random strings (interval: $index_interval)" );

    $lexicon->seek('j');
    is( $lexicon->get_term, 'j', "seek (interval: $index_interval)" );

    $lexicon->seek(undef);
    ok( !defined $lexicon->get_term, "seek to undef resets" );
}