1 files changed, 82 insertions, 0 deletions
diff --git a/web/server/h2o/libh2o/misc/p5-net-fastcgi/t/020_protocol/015_build_stream.t b/web/server/h2o/libh2o/misc/p5-net-fastcgi/t/020_protocol/015_build_stream.t
new file mode 100644
index 00000000..233f225f
--- /dev/null
+++ b/web/server/h2o/libh2o/misc/p5-net-fastcgi/t/020_protocol/015_build_stream.t
@@ -0,0 +1,82 @@
+#!perl
+
+use strict;
+use warnings;
+
+use lib 't/lib', 'lib';
+use myconfig;
+
+use Test::More tests => 12;
+use Test::HexString;
+use Test::Exception;
+
+BEGIN {
+    use_ok('Net::FastCGI::Protocol', qw[ build_stream ]);
+}
+
+sub TRUE  () { !!1 }
+sub FALSE () { !!0 }
+
+my @tests = (
+    # expected,                             type, request_id, content, terminate
+    [ "",                                      1,          1,      '',     FALSE ],
+    [ "",                                      1,          1,   undef,     FALSE ],
+    [ "\x01\x01\x00\x01\x00\x00\x00\x00",      1,          1,      '',      TRUE ],
+    [ "\x01\x01\x00\x01\x00\x00\x00\x00",      1,          1,   undef,      TRUE ],
+
+    [ "\x01\x01\x00\x01\x00\x03\x05\x00"
+    . "FOO\x00\x00\x00\x00\x00",               1,          1,   'FOO',     FALSE ],
+
+    [ "\x01\x01\x00\x01\x00\x03\x05\x00"
+    . "FOO\x00\x00\x00\x00\x00"
+    . "\x01\x01\x00\x01\x00\x00\x00\x00",      1,          1,   'FOO',      TRUE ],
+);
+
+foreach my $test (@tests) {
+    my $expected = $test->[0];
+    my $got      = build_stream(@$test[1..4]);
+    is_hexstr($got, $expected, 'build_stream()');
+}
+
+{
+    my $header  = "\x01\x01\x00\x01\x7F\xF8\x00\x00";
+    my $content = "x" x 32760;
+    my $trailer = "\x01\x01\x00\x01\x00\x00\x00\x00";
+
+    {
+        my $expected = $header . $content;
+        my $got      = build_stream(1,1, $content);
+        is_hexstr($got, $expected, 'build_stream(content_length: 32760 terminate:false)');
+    }
+
+    {
+        my $expected = $header . $content . $trailer;
+        my $got      = build_stream(1,1, $content, 1);
+        is_hexstr($got, $expected, 'build_stream(content_length: 32760 terminate:true)');
+    }
+}
+
+{
+    my $records = "\x01\x01\x00\x01\x7F\xF8\x00\x00" #  H1
+                . "x" x 32760                        #  C1
+                . "\x01\x01\x00\x01\x00\x08\x00\x00" #  H2
+                . "x" x 8                            #  C2
+                ;
+    my $content = "x" x 32768;
+    my $trailer = "\x01\x01\x00\x01\x00\x00\x00\x00";
+
+    {
+        my $expected = $records;
+        my $got      = build_stream(1,1, $content);
+        is_hexstr($got, $records, 'build_stream(content_length: 32768 terminate:false)');
+    }
+
+    {
+        my $expected = $records . $trailer;
+        my $got      = build_stream(1,1, $content, 1);
+        is_hexstr($got, $expected, 'build_stream(content_length: 32768 terminate:true)');
+    }
+}
+
+throws_ok { build_stream() } qr/^Usage: /;
+