www_stream: add response wrapper sub

[public-inbox.git] / t / view.t

diff --git a/t/view.t b/t/view.t
index 6c0859915e89c9ae2d1384d69c2bd35a5c8d5ed0..4fdd15180cc45983630c6f2aa4f2b34f519f4704 100644 (file)
--- a/t/view.t
+++ b/t/view.t
@@ -4,13 +4,38 @@ use strict;
 use warnings;
 use Test::More;
 use Email::MIME;
-use PublicInbox::View;
+use Plack::Util;
+use_ok 'PublicInbox::View';
+
+my @q = (
+       'foo@bar', 'foo@bar',
+       'a b', "'a b'",
+       "a'b", "'a'\\''b'",
+);
+while (@q) {
+       my $input = shift @q;
+       my $expect = shift @q;
+       my $res = PublicInbox::View::squote_maybe($input);
+       is($res, $expect, "quote $input => $res");
+}
+
+# FIXME: make this test less fragile
+my $ctx = {
+       env => { HTTP_HOST => 'example.com', 'psgi.url_scheme' => 'http' },
+       -inbox => Plack::Util::inline_object(
+               name => 'test',
+               search => sub { undef },
+               cloneurl => sub {[]},
+               description => sub { '' }),
+};
+$ctx->{-inbox}->{-primary_address} = 'test@example.com';
 
 sub msg_html ($) {
        my ($mime) = @_;
 
        my $s = '';
-       my $body = PublicInbox::View::msg_html(undef, $mime);
+       my $r = PublicInbox::View::msg_html($ctx, $mime);
+       my $body = $r->[2];
        while (defined(my $buf = $body->getline)) {
                $s .= $buf;
        }