Mercurial > hg > nginx-tests

comparison proxy_variables.t @ 520:4bcf8bc2bafe

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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 ###############################################################################