t/psgi_search.t

   1 # Copyright (C) 2017-2018 all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3 use strict;
   4 use warnings;
   5 use Test::More;
   6 use File::Temp qw/tempdir/;
   7 use Email::MIME;
   8 use PublicInbox::Config;
   9 use PublicInbox::Inbox;
  10 use PublicInbox::WWW;
  11 use bytes (); # only for bytes::length
  12 my @mods = qw(DBD::SQLite Search::Xapian HTTP::Request::Common Plack::Test
  13                 URI::Escape Plack::Builder);
  14 foreach my $mod (@mods) {
  15         eval "require $mod";
  16         plan skip_all => "$mod missing for psgi_search.t" if $@;
  17 }
  18 use_ok $_ foreach (@mods, qw(PublicInbox::SearchIdx));
  19 my $tmpdir = tempdir('pi-psgi-search.XXXXXX', TMPDIR => 1, CLEANUP => 1);
  20 my $git_dir = "$tmpdir/a.git";
  21
  22 is(0, system(qw(git init -q --bare), $git_dir), "git init (main)");
  23 my $ibx = PublicInbox::Inbox->new({mainrepo => $git_dir});
  24 my $rw = PublicInbox::SearchIdx->new($ibx, 1);
  25 ok($rw, "search indexer created");
  26 my $digits = '10010260936330';
  27 my $ua = 'Pine.LNX.4.10';
  28 my $mid = "$ua.$digits.2460-100000\@penguin.transmeta.com";
  29 my $data = <<"EOF";
  30 Subject: test
  31 Message-ID: <$mid>
  32 From: Ævar Arnfjörð Bjarmason <avarab\@example>
  33 To: git\@vger.kernel.org
  34
  35 EOF
  36
  37 my $num = 0;
  38 # nb. using internal API, fragile!
  39 $rw->begin_txn_lazy;
  40
  41 foreach (reverse split(/\n\n/, $data)) {
  42         $_ .= "\n";
  43         my $mime = Email::MIME->new(\$_);
  44         my $bytes = bytes::length($mime->as_string);
  45         my $doc_id = $rw->add_message($mime, $bytes, ++$num, 'ignored');
  46         ok($doc_id, 'message added');
  47 }
  48
  49 $rw->commit_txn_lazy;
  50
  51 my $cfgpfx = "publicinbox.test";
  52 my $config = PublicInbox::Config->new({
  53         "$cfgpfx.address" => 'git@vger.kernel.org',
  54         "$cfgpfx.mainrepo" => $git_dir,
  55 });
  56 my $www = PublicInbox::WWW->new($config);
  57 test_psgi(sub { $www->call(@_) }, sub {
  58         my ($cb) = @_;
  59         my $res;
  60         $res = $cb->(GET('/test/?q=%C3%86var'));
  61         my $html = $res->content;
  62         like($html, qr/<title>&#198;var - /, 'HTML escaped in title');
  63         my @res = ($html =~ m/\?q=(.+var)\b/g);
  64         ok(scalar(@res), 'saw query strings');
  65         my %uniq = map { $_ => 1 } @res;
  66         is(1, scalar keys %uniq, 'all query values identical in HTML');
  67         is('%C3%86var', (keys %uniq)[0], 'matches original query');
  68         ok(index($html, 'by &#198;var Arnfj&#246;r&#240; Bjarmason') >= 0,
  69                 "displayed Ævar's name properly in HTML");
  70
  71         my $warn = [];
  72         local $SIG{__WARN__} = sub { push @$warn, @_ };
  73         $res = $cb->(GET('/test/?q=s:test&l=5e'));
  74         is($res->code, 200, 'successful search result');
  75         is_deeply([], $warn, 'no warnings from non-numeric comparison');
  76
  77         $res = $cb->(POST('/test/?q=s:bogus&x=m'));
  78         is($res->code, 404, 'failed search result gives 404');
  79         is_deeply([], $warn, 'no warnings');
  80
  81         my $mid_re = qr/\Q$mid\E/o;
  82         while (length($digits) > 8) {
  83                 $res = $cb->(GET("/test/$ua.$digits/"));
  84                 is($res->code, 300, 'partial match found while truncated');
  85                 like($res->content, qr/\b1 partial match found\b/);
  86                 like($res->content, $mid_re, 'found mid in response');
  87                 chop($digits);
  88         }
  89 });
  90
  91 done_testing();
  92
  93 1;