Mercurial > hg > nginx-tests
comparison proxy_variables.t @ 520:4bcf8bc2bafe
Tests: $upstream_header_time and $upstream_response_time variables.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Thu, 15 Jan 2015 18:23:16 +0300 |
parents | |
children | 907e89fba9c3 |
comparison
equal
deleted
inserted
replaced
519:089c4174e269 | 520:4bcf8bc2bafe |
---|---|
1 #!/usr/bin/perl | |
2 | |
3 # (C) Sergey Kandaurov | |
4 # (C) Nginx, Inc. | |
5 | |
6 # Tests for http proxy module with upstream variables. | |
7 | |
8 ############################################################################### | |
9 | |
10 use warnings; | |
11 use strict; | |
12 | |
13 use Test::More; | |
14 | |
15 BEGIN { use FindBin; chdir($FindBin::Bin); } | |
16 | |
17 use lib 'lib'; | |
18 use Test::Nginx; | |
19 | |
20 ############################################################################### | |
21 | |
22 select STDERR; $| = 1; | |
23 select STDOUT; $| = 1; | |
24 | |
25 my $t = Test::Nginx->new()->has(qw/http proxy/) | |
26 ->write_file_expand('nginx.conf', <<'EOF'); | |
27 | |
28 %%TEST_GLOBALS%% | |
29 | |
30 daemon off; | |
31 | |
32 events { | |
33 } | |
34 | |
35 http { | |
36 %%TEST_GLOBALS_HTTP%% | |
37 | |
38 upstream u { | |
39 server 127.0.0.1:8081; | |
40 server 127.0.0.1:8081; | |
41 } | |
42 | |
43 log_format time $upstream_header_time:$upstream_response_time; | |
44 | |
45 server { | |
46 listen 127.0.0.1:8080; | |
47 server_name localhost; | |
48 | |
49 location / { | |
50 add_header X-Header-Time $upstream_header_time; | |
51 proxy_pass http://127.0.0.1:8081; | |
52 access_log %%TESTDIR%%/time.log time; | |
53 } | |
54 | |
55 location /pnu { | |
56 add_header X-Header-Time $upstream_header_time; | |
57 add_header X-Response-Time $upstream_response_time; | |
58 proxy_pass http://u/bad; | |
59 } | |
60 } | |
61 } | |
62 | |
63 EOF | |
64 | |
65 $t->run_daemon(\&http_daemon, 8081); | |
66 $t->try_run('no upstream_header_time')->plan(8); | |
67 | |
68 $t->waitforsocket('127.0.0.1:8081'); | |
69 | |
70 ############################################################################### | |
71 | |
72 my $re = qr/(\d\.\d{3})/; | |
73 my ($ht, $rt, $ht2, $rt2); | |
74 | |
75 ($ht) = http_get('/header') =~ /X-Header-Time: $re/; | |
76 cmp_ok($ht, '>=', 1, 'header time - slow response header'); | |
77 | |
78 ($ht) = http_get('/body') =~ /X-Header-Time: $re/; | |
79 cmp_ok($ht, '<', 1, 'header time - slow response body'); | |
80 | |
81 my $r = http_get('/pnu'); | |
82 ($ht) = $r =~ /X-Header-Time: ($re)/; | |
83 ($rt) = $r =~ /X-Response-Time: ($re)/; | |
84 | |
85 is($ht, $rt, 'header time - bad response'); | |
86 like($r, qr/X-Header-Time: $re, $re/, 'header time - next'); | |
87 | |
88 $t->stop(); | |
89 | |
90 ($ht, $rt, $ht2, $rt2) = $t->read_file('time.log') =~ /^$re:$re\n$re:$re$/; | |
91 | |
92 cmp_ok($ht, '>=', 1, 'header time log - slow response header'); | |
93 cmp_ok($ht2, '<', 1, 'header time log - slow response body'); | |
94 | |
95 cmp_ok($rt, '>=', 1, 'response time log - slow response header'); | |
96 cmp_ok($rt2, '>=', 1, 'response time log - slow response body'); | |
97 | |
98 ############################################################################### | |
99 | |
100 sub http_daemon { | |
101 my ($port) = @_; | |
102 my $once = 1; | |
103 | |
104 my $server = IO::Socket::INET->new( | |
105 Proto => 'tcp', | |
106 LocalHost => '127.0.0.1', | |
107 LocalPort => $port, | |
108 Listen => 5, | |
109 Reuse => 1 | |
110 ) | |
111 or die "Can't create listening socket: $!\n"; | |
112 | |
113 local $SIG{PIPE} = 'IGNORE'; | |
114 | |
115 while (my $client = $server->accept()) { | |
116 $client->autoflush(1); | |
117 | |
118 my $headers = ''; | |
119 my $uri = ''; | |
120 | |
121 while (<$client>) { | |
122 $headers .= $_; | |
123 last if (/^\x0d?\x0a?$/); | |
124 } | |
125 | |
126 $uri = $1 if $headers =~ /^\S+\s+([^ ]+)\s+HTTP/i; | |
127 next unless defined $uri; | |
128 | |
129 if ($uri =~ 'bad' && $once) { | |
130 $once = 0; | |
131 sleep 1; | |
132 next; | |
133 } | |
134 | |
135 if ($uri =~ 'header') { | |
136 sleep 1; | |
137 } | |
138 | |
139 print $client <<EOF; | |
140 HTTP/1.1 200 OK | |
141 Connection: close | |
142 | |
143 SEE-THIS- | |
144 EOF | |
145 | |
146 if ($uri =~ 'body') { | |
147 sleep 1; | |
148 } | |
149 | |
150 print $client 'AND-THIS'; | |
151 } | |
152 } | |
153 | |
154 ############################################################################### |