Bug 33277: Add comments and missing thesauri
[koha.git] / t / db_dependent / Heading.t
1 #!/usr/bin/perl
2 #
3 # This file is part of Koha.
4 #
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.
9 #
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.
14 #
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>.
17
18 use strict;
19 use warnings;
20
21 use Test::More tests => 4;
22
23 use t::lib::Mocks;
24 use Test::MockModule;
25
26 BEGIN {
27     use_ok('C4::Heading', qw( field valid_heading_subfield ));
28 }
29
30 subtest "MARC21 tests" => sub {
31     plan tests => 8;
32
33     t::lib::Mocks::mock_preference('marcflavour', 'MARC21');
34
35     ok(C4::Heading::valid_heading_subfield('100', 'a'), '100a valid for bib');
36     ok(!C4::Heading::valid_heading_subfield('100', 'e'), '100e not valid for bib');
37
38     ok(C4::Heading::valid_heading_subfield('100', 'a', 1), '100a valid for authority');
39
40     ok(C4::Heading::valid_heading_subfield('110', 'a'), '110a valid for bib');
41     ok(!C4::Heading::valid_heading_subfield('110', 'e'), '110e not valid for bib');
42
43     ok(C4::Heading::valid_heading_subfield('600', 'a'), '600a valid for bib');
44     ok(!C4::Heading::valid_heading_subfield('600', 'e'), '600e not valid for bib');
45
46     ok(!C4::Heading::valid_heading_subfield('012', 'a'), '012a invalid field for bib');
47 };
48
49 subtest "UNIMARC tests" => sub {
50     plan tests => 7;
51
52     t::lib::Mocks::mock_preference('marcflavour', 'UNIMARC');
53
54     ok(C4::Heading::valid_heading_subfield('100', 'a'), '100a valid for bib');
55     ok(!C4::Heading::valid_heading_subfield('100', 'i'), '100i not valid fir bib');
56
57     ok(C4::Heading::valid_heading_subfield('110', 'a'), '110a valid for bib');
58     ok(!C4::Heading::valid_heading_subfield('110', 'i'), '110i not valid for bib');
59
60     ok(C4::Heading::valid_heading_subfield('600', 'a'), '600a valid for bib');
61     ok(!C4::Heading::valid_heading_subfield('600', 'i'), '600i not valid for bib');
62
63     ok(!C4::Heading::valid_heading_subfield('012', 'a'), '012a invalid field for bib');
64 };
65
66 subtest "_search tests" => sub {
67
68     plan tests => 7;
69
70     t::lib::Mocks::mock_preference('marcflavour', 'MARC21');
71     t::lib::Mocks::mock_preference('SearchEngine', 'Elasticsearch');
72     # NOTE: We are testing solely against ES here to make the assumptions simpler while testing
73     # C4/Headings code specifically. The actual query building and searching code should
74     # be covered in other test files
75     my $search = Test::MockModule->new('Koha::SearchEngine::Elasticsearch::Search');
76
77     $search->mock('search_auth_compat', sub {
78         my $self = shift;
79         my $search_query = shift;
80         return ($search_query, 1 );
81     });
82
83
84     my $field = MARC::Field->new( '650', ' ', '0', a => 'Uncles', x => 'Fiction' );
85     my $heading = C4::Heading->new_from_field($field);
86     my ($search_query) = $heading->_search( 'match-heading' );
87     my $terms = $search_query->{query}->{bool}->{must};
88     my $expected_terms = [
89         { term => { 'match-heading.ci_raw' => 'Uncles generalsubdiv Fiction' } },
90         { term => { 'subject-heading-thesaurus.ci_raw' => 'a' } },
91     ];
92     is_deeply( $terms, $expected_terms, "Search formed as expected for a subject with second indicator 0");
93
94     $field = MARC::Field->new( '650', ' ', '3', a => 'Uncles', x => 'Fiction' );
95     $heading = C4::Heading->new_from_field($field);
96     ($search_query) = $heading->_search( 'match-heading' );
97     $terms = $search_query->{query}->{bool}->{must};
98     $expected_terms = [
99         { term => { 'match-heading.ci_raw' => 'Uncles generalsubdiv Fiction' } },
100         { term => { 'subject-heading-thesaurus.ci_raw' => 'd' } },
101     ];
102     is_deeply( $terms, $expected_terms, "Search formed as expected with second indicator 3");
103
104     $field = MARC::Field->new( '650', ' ', '7', a => 'Uncles', x => 'Fiction', 2 => 'special_sauce' );
105     $heading = C4::Heading->new_from_field($field);
106     ($search_query) = $heading->_search( 'match-heading' );
107     $terms = $search_query->{query}->{bool}->{must};
108     $expected_terms = [
109         { term => { 'match-heading.ci_raw' => 'Uncles generalsubdiv Fiction' } },
110         { term => { 'subject-heading-thesaurus.ci_raw' => 'special_sauce' } },
111     ];
112     is_deeply( $terms, $expected_terms, "Search formed as expected with second indicator 7 and subfield 2");
113
114     $field = MARC::Field->new( '100', ' ', '', a => 'Yankovic, Al', d => '1959-,' );
115     $heading = C4::Heading->new_from_field($field);
116     ($search_query) = $heading->_search( 'match-heading' );
117     $terms = $search_query->{query}->{bool}->{must};
118     $expected_terms = [
119         { term => { 'match-heading.ci_raw' => 'Yankovic, Al 1959' } },
120     ];
121     is_deeply( $terms, $expected_terms, "Search formed as expected for a non-subject field with single punctuation mark");
122
123     $field = MARC::Field->new( '100', ' ', '', a => 'Yankovic, Al', d => '1959-,', e => '[author]' );
124     $heading = C4::Heading->new_from_field($field);
125     ($search_query) = $heading->_search( 'match-heading' );
126     $terms = $search_query->{query}->{bool}->{must};
127     $expected_terms = [
128         { term => { 'match-heading.ci_raw' => 'Yankovic, Al 1959' } },
129     ];
130     is_deeply( $terms, $expected_terms, "Search formed as expected for a non-subject field with double punctuation, hyphen+comma");
131
132     $field = MARC::Field->new( '100', ' ', '', a => 'Tolkien, J.R.R.,', e => '[author]' );
133     $heading = C4::Heading->new_from_field($field);
134     ($search_query) = $heading->_search( 'match-heading' );
135     $terms = $search_query->{query}->{bool}->{must};
136     $expected_terms = [
137         { term => { 'match-heading.ci_raw' => 'Tolkien, J.R.R' } },
138     ];
139     is_deeply( $terms, $expected_terms, "Search formed as expected for a non-subject field with double punctuation, period+comma ");
140
141     $search->mock('search_auth_compat', sub {
142         my $self = shift;
143         my $search_query = shift;
144         if( $search_query->{query}->{bool}->{must}[1]->{term}->{'subject-heading-thesaurus.ci_raw'} eq 'special_sauce' ){
145             return;
146         }
147         return ($search_query, 1);
148     });
149
150     # Special case where thesaurus defined in subfield 2 should also match record with no thesaurus
151     $field = MARC::Field->new( '650', ' ', '7', a => 'Uncles', x => 'Fiction', 2 => 'special_sauce' );
152     $heading = C4::Heading->new_from_field($field);
153     ($search_query) = $heading->_search( 'match-heading' );
154     $terms = $search_query->{query}->{bool}->{must};
155     $expected_terms = [
156         { term => { 'match-heading.ci_raw' => 'Uncles generalsubdiv Fiction' } },
157         { term => { 'subject-heading-thesaurus.ci_raw' => 'z' } },
158     ];
159     is_deeply( $terms, $expected_terms, "When thesaurus in subfield 2, and nothing is found, we should search again for 008_11 = z");
160
161 };