add test helper for querying ES

Author: haarg

t/lib/MetaCPAN/Server/Test.pm

@@ -4,17 +4,21 @@ use strict;
 use warnings;
 use feature qw(state);
 
+use Carp                      qw( croak );
 use HTTP::Request::Common     qw( DELETE GET POST );    ## no perlimports
+use MetaCPAN::ESConfig        qw( es_doc_path );
 use MetaCPAN::Model           ();
 use MetaCPAN::Server          ();
 use MetaCPAN::Server::Config  ();
 use MetaCPAN::Types::TypeTiny qw( ES );
+use MetaCPAN::Util            qw( hit_total );
 use Plack::Test;                                        ## no perlimports
 
 use base 'Exporter';
 our @EXPORT_OK = qw(
     POST GET DELETE
     es
+    es_result
     model
     test_psgi app
     query
@@ -57,6 +61,28 @@ sub query {
     state $query = MetaCPAN::Query->new( es => es() );
 }
 
+sub es_result {
+    my ( $type, $query, $size ) = @_;
+    $size //= wantarray ? 999 : 1;
+    if ( !wantarray && $size != 1 ) {
+        croak "multiple results requested with scalar return!";
+    }
+    my $res = es()->search(
+        es_doc_path($type),
+        body => {
+            size  => ( wantarray ? 999 : 1 ),
+            query => $query,
+        },
+    );
+    my @hits = map $_->{_source}, @{ $res->{hits}{hits} };
+    if ( !wantarray ) {
+        croak "query did not return a single result"
+            if hit_total($res) != 1;
+        return $hits[0];
+    }
+    return @hits;
+}
+
 1;
 
 =pod