www_stream: add response wrapper sub

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

diff --git a/t/view.t b/t/view.t
index 4ce3c7739018ddf3f5d060471043d729320ef55c..4fdd15180cc45983630c6f2aa4f2b34f519f4704 100644 (file)
--- a/t/view.t
+++ b/t/view.t
@@ -4,8 +4,20 @@ 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 = {
@@ -22,7 +34,8 @@ sub msg_html ($) {
        my ($mime) = @_;
 
        my $s = '';
-       my $body = PublicInbox::View::msg_html($ctx, $mime);
+       my $r = PublicInbox::View::msg_html($ctx, $mime);
+       my $body = $r->[2];
        while (defined(my $buf = $body->getline)) {
                $s .= $buf;
        }