1#!perl
2
3use strict;
4use warnings;
5
6use File::Basename;
7use Test::More 0.88;
8use lib 't';
9use Util qw[tmpfile rewind slurp monkey_patch dir_list parse_case
10  hashify connect_args clear_socket_source set_socket_source sort_headers
11  $CRLF $LF];
12
13use HTTP::Tiny;
14BEGIN { monkey_patch() }
15
16for my $file ( dir_list("corpus", qr/^redirect/ ) ) {
17  my $label = basename($file);
18  my $data = do { local (@ARGV,$/) = $file; <> };
19  my ($params, @case_pairs) = split /--+\n/, $data;
20  my $case = parse_case($params);
21  my $number_of_requests = @case_pairs / 2;
22
23  my $url = $case->{url}[0];
24  my $method = $case->{method}[0] || 'GET';
25  my %headers = hashify( $case->{headers} );
26  my %new_args = hashify( $case->{new_args} );
27
28  my %options;
29  $options{headers} = \%headers if %headers;
30  my $call_args = %options ? [$method, $url, \%options] : [$method, $url];
31
32  my $version = HTTP::Tiny->VERSION || 0;
33  my $agent = $new_args{agent} || "HTTP-Tiny/$version";
34
35  my (@socket_pairs);
36  while ( @case_pairs ) {
37    my ($expect_req, $give_res) = splice( @case_pairs, 0, 2 );
38    # cleanup source data
39    $expect_req =~ s{HTTP-Tiny/VERSION}{$agent};
40    s{\n}{$CRLF}g for ($expect_req, $give_res);
41
42    # setup mocking and test
43    my $req_fh = tmpfile();
44    my $res_fh = tmpfile($give_res);
45
46    push @socket_pairs, [$req_fh, $res_fh, $expect_req];
47  }
48
49  clear_socket_source();
50  set_socket_source(@$_) for @socket_pairs;
51
52  my $http = HTTP::Tiny->new(keep_alive => 0, %new_args);
53  my $response  = $http->request(@$call_args);
54  my $max_redirects = defined($new_args{max_redirect}) ? $new_args{max_redirect} : 5;
55  my $calls = 0 + $max_redirects;
56
57  for my $i ( 0 .. $calls ) {
58    last unless @socket_pairs;
59    my ($req_fh, $res_fh, $expect_req) = @{ shift @socket_pairs };
60    my $got_req = slurp($req_fh);
61    is( sort_headers($got_req), sort_headers($expect_req), "$label request ($i)");
62    $i++;
63  }
64
65  my $exp_content = $case->{expected}
66                  ? join("$CRLF", @{$case->{expected}}) : '';
67
68  is ( $response->{content}, $exp_content, "$label content" );
69
70  my $number_of_redirects =
71    $max_redirects < $number_of_requests ? $max_redirects : $number_of_requests - 1;
72  is ( @{ $response->{redirects} || [] }, $number_of_redirects,
73      "$label redirects array size"
74  ) or diag explain $response->{redirects};
75
76  if ( $case->{expected_url} ) {
77    is ( $response->{url}, $case->{expected_url}[0], "$label response URL" );
78  }
79
80}
81
82done_testing;
83