Bug 18374: (QA follow-up) Tidy up code
[koha.git] / t / db_dependent / Koha_SearchEngine_Elasticsearch_Search.t
blobf547fcbc17a6e2aee009af4762c8bda94095ce23
1 # Copyright 2015 Catalyst IT
3 # This file is part of Koha.
5 # Koha is free software; you can redistribute it and/or modify it
6 # under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
10 # Koha is distributed in the hope that it will be useful, but
11 # WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
15 # You should have received a copy of the GNU General Public License
16 # along with Koha; if not, see <http://www.gnu.org/licenses>.
18 use Modern::Perl;
20 use Test::More tests => 12;
21 use t::lib::Mocks;
23 use Koha::SearchEngine::Elasticsearch::QueryBuilder;
25 my $builder = Koha::SearchEngine::Elasticsearch::QueryBuilder->new( { index => 'mydb' } );
27 use_ok('Koha::SearchEngine::Elasticsearch::Search');
29 ok(
30     my $searcher = Koha::SearchEngine::Elasticsearch::Search->new(
31         { 'nodes' => ['localhost:9200'], 'index' => 'mydb' }
32     ),
33     'Creating a Koha::SearchEngine::Elasticsearch::Search object'
36 is( $searcher->index, 'mydb', 'Testing basic accessor' );
38 ok( my $query = $builder->build_query('easy'), 'Build a search query');
40 SKIP: {
42     eval { $builder->get_elasticsearch_params; };
44     skip 'ElasticSeatch configuration not available', 6
45         if $@;
47     ok( my $results = $searcher->search( $query) , 'Do a search ' );
49     ok( my $marc = $searcher->json2marc( $results->first ), 'Convert JSON to MARC');
51     is (my $count = $searcher->count( $query ), 0 , 'Get a count of the results, without returning results ');
53     ok ($results = $searcher->search_compat( $query ), 'Test search_compat' );
55     ok (($results,$count) = $searcher->search_auth_compat ( $query ), 'Test search_auth_compat' );
57     is ( $count = $searcher->count_auth_use($searcher,1), 0, 'Testing count_auth_use');
61 subtest 'json2marc' => sub {
62     plan tests => 4;
63     my $leader = '00626nam a2200193   4500';
64     my $_001 = 42;
65     my $_010a = '123456789';
66     my $_010d = 145;
67     my $_200a = 'a title';
68     my $json = [ # It's not a JSON, see the POD of json2marc
69         [ 'LDR', undef, undef, '_', $leader ],
70         [ '001', undef, undef, '_', $_001 ],
71         [ '010', ' ', ' ', 'a', $_010a, 'd', $_010d ],
72         [ '200', '1', ' ', 'a', $_200a, ], # Yes UNIMARC but we don't mind here
73     ];
75     my $marc = $searcher->json2marc( $json );
76     is( $marc->leader, $leader, );
77     is( $marc->field('001')->data, $_001, );
78     is( $marc->subfield('010', 'a'), $_010a, );
79     is( $marc->subfield('200', 'a'), $_200a, );
83 subtest 'build_query tests' => sub {
84     plan tests => 10;
86     t::lib::Mocks::mock_preference('DisplayLibraryFacets','both');
87     my $query = $builder->build_query();
88     ok( defined $query->{aggregations}{homebranch},
89         'homebranch added to facets if DisplayLibraryFacets=both' );
90     ok( defined $query->{aggregations}{holdingbranch},
91         'holdingbranch added to facets if DisplayLibraryFacets=both' );
92     t::lib::Mocks::mock_preference('DisplayLibraryFacets','holding');
93     $query = $builder->build_query();
94     ok( !defined $query->{aggregations}{homebranch},
95         'homebranch not added to facets if DisplayLibraryFacets=holding' );
96     ok( defined $query->{aggregations}{holdingbranch},
97         'holdingbranch added to facets if DisplayLibraryFacets=holding' );
98     t::lib::Mocks::mock_preference('DisplayLibraryFacets','home');
99     $query = $builder->build_query();
100     ok( defined $query->{aggregations}{homebranch},
101         'homebranch added to facets if DisplayLibraryFacets=home' );
102     ok( !defined $query->{aggregations}{holdingbranch},
103         'holdingbranch not added to facets if DisplayLibraryFacets=home' );
105     t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '' );
107     ( undef, $query ) = $builder->build_query_compat( undef, ['donald duck'] );
108     is(
109         $query->{query}{query_string}{query},
110         "(donald duck)",
111         "query not altered if QueryAutoTruncate disabled"
112     );
114     t::lib::Mocks::mock_preference( 'QueryAutoTruncate', '1' );
116     ( undef, $query ) = $builder->build_query_compat( undef, ['donald duck'] );
117     is(
118         $query->{query}{query_string}{query},
119         "(donald* duck*)",
120         "simple query is auto truncated when QueryAutoTruncate enabled"
121     );
123     # Ensure reserved words are not truncated
124     ( undef, $query ) = $builder->build_query_compat( undef,
125         ['donald or duck and mickey not mouse'] );
126     is(
127         $query->{query}{query_string}{query},
128         "(donald* or duck* and mickey* not mouse*)",
129         "reserved words are not affected by QueryAutoTruncate"
130     );
132     ( undef, $query ) = $builder->build_query_compat( undef, ['donald* duck*'] );
133     is(
134         $query->{query}{query_string}{query},
135         "(donald* duck*)",
136         "query with '*' is unaltered when QueryAutoTruncate is enabled"
137     );