Mercurial > hg > nginx-tests

comparison js_fetch_resolver.t @ 1640:67adc5fd0548

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: added js tests for ngx.fetch() method.
author Dmitry Volyntsev <xeioex@nginx.com>
date Thu, 21 Jan 2021 18:19:37 +0000
parents
children f89770df737a
comparison
equal deleted inserted replaced
1639:6c323c672a86 1640:67adc5fd0548
1 #!/usr/bin/perl
2
3 # (C) Dmitry Volyntsev
4 # (C) Nginx, Inc.
5
6 # Tests for http njs module, fetch method, dns support.
7
8 ###############################################################################
9
10 use warnings;
11 use strict;
12
13 use Test::More;
14
15 use IO::Select;
16
17 BEGIN { use FindBin; chdir($FindBin::Bin); }
18
19 use lib 'lib';
20 use Test::Nginx;
21
22 ###############################################################################
23
24 select STDERR; $| = 1;
25 select STDOUT; $| = 1;
26
27 plan(skip_all => '127.0.0.2 local address required')
28 unless defined IO::Socket::INET->new( LocalAddr => '127.0.0.2' );
29
30 my $t = Test::Nginx->new()->has(qw/http/)
31 ->write_file_expand('nginx.conf', <<'EOF');
32
33 %%TEST_GLOBALS%%
34
35 daemon off;
36
37 events {
38 }
39
40 http {
41 %%TEST_GLOBALS_HTTP%%
42
43 js_import test.js;
44
45 server {
46 listen 127.0.0.1:8080;
47 server_name localhost;
48
49 location /njs {
50 js_content test.njs;
51 }
52
53 location /dns {
54 js_content test.dns;
55
56 resolver 127.0.0.1:%%PORT_8981_UDP%%;
57 resolver_timeout 1s;
58 }
59 }
60
61 server {
62 listen 127.0.0.1:8080;
63 server_name aaa;
64
65 location /loc {
66 js_content test.loc;
67 }
68 }
69
70 server {
71 listen 127.0.0.1:8080;
72 server_name many;
73
74 location /loc {
75 js_content test.loc;
76 }
77 }
78 }
79
80 EOF
81
82 my $p0 = port(8080);
83
84 $t->write_file('test.js', <<EOF);
85 function test_njs(r) {
86 r.return(200, njs.version);
87 }
88
89 function dns(r) {
90 var url = `http://\${r.args.domain}:$p0/loc`;
91
92 ngx.fetch(url)
93 .then(reply => reply.text())
94 .then(body => r.return(200, body))
95 .catch(e => r.return(501, e.message))
96 }
97
98 function str(v) { return v ? v : ''};
99
100 function loc(r) {
101 var v = r.variables;
102 var body = str(r.requestText);
103 var foo = str(r.headersIn.foo);
104 var bar = str(r.headersIn.bar);
105 var c = r.headersIn.code ? Number(r.headersIn.code) : 200;
106 r.return(c, `\${v.host}:\${v.request_method}:\${foo}:\${bar}:\${body}`);
107 }
108
109 export default {njs: test_njs, dns, loc};
110 EOF
111
112 $t->try_run('no njs.fetch')->plan(3);
113
114 $t->run_daemon(\&dns_daemon, port(8981), $t);
115 $t->waitforfile($t->testdir . '/' . port(8981));
116
117 ###############################################################################
118
119 local $TODO = 'not yet'
120 unless http_get('/njs') =~ /^([.0-9]+)$/m && $1 ge '0.5.1';
121
122 like(http_get('/dns?domain=aaa'), qr/aaa:GET:::$/s, 'fetch dns aaa');
123 like(http_get('/dns?domain=many'), qr/many:GET:::$/s, 'fetch dns many');
124 like(http_get('/dns?domain=unknown'), qr/"unknown" could not be resolved/s,
125 'fetch dns unknown');
126
127 ###############################################################################
128
129 sub reply_handler {
130 my ($recv_data, $port, %extra) = @_;
131
132 my (@name, @rdata);
133
134 use constant NOERROR => 0;
135 use constant FORMERR => 1;
136 use constant SERVFAIL => 2;
137 use constant NXDOMAIN => 3;
138
139 use constant A => 1;
140
141 use constant IN => 1;
142
143 # default values
144
145 my ($hdr, $rcode, $ttl) = (0x8180, NOERROR, 3600);
146
147 # decode name
148
149 my ($len, $offset) = (undef, 12);
150 while (1) {
151 $len = unpack("\@$offset C", $recv_data);
152 last if $len == 0;
153 $offset++;
154 push @name, unpack("\@$offset A$len", $recv_data);
155 $offset += $len;
156 }
157
158 $offset -= 1;
159 my ($id, $type, $class) = unpack("n x$offset n2", $recv_data);
160
161 my $name = join('.', @name);
162
163 if ($name eq 'aaa' && $type == A) {
164 push @rdata, rd_addr($ttl, '127.0.0.1');
165
166 } elsif ($name eq 'many' && $type == A) {
167 push @rdata, rd_addr($ttl, '127.0.0.2');
168 push @rdata, rd_addr($ttl, '127.0.0.1');
169 }
170
171 $len = @name;
172 pack("n6 (C/a*)$len x n2", $id, $hdr | $rcode, 1, scalar @rdata,
173 0, 0, @name, $type, $class) . join('', @rdata);
174 }
175
176 sub rd_addr {
177 my ($ttl, $addr) = @_;
178
179 my $code = 'split(/\./, $addr)';
180
181 return pack 'n3N', 0xc00c, A, IN, $ttl if $addr eq '';
182
183 pack 'n3N nC4', 0xc00c, A, IN, $ttl, eval "scalar $code", eval($code);
184 }
185
186 sub dns_daemon {
187 my ($port, $t, %extra) = @_;
188
189 my ($data, $recv_data);
190 my $socket = IO::Socket::INET->new(
191 LocalAddr => '127.0.0.1',
192 LocalPort => $port,
193 Proto => 'udp',
194 )
195 or die "Can't create listening socket: $!\n";
196
197 my $sel = IO::Select->new($socket);
198
199 local $SIG{PIPE} = 'IGNORE';
200
201 # signal we are ready
202
203 open my $fh, '>', $t->testdir() . '/' . $port;
204 close $fh;
205
206 while (my @ready = $sel->can_read) {
207 foreach my $fh (@ready) {
208 if ($socket == $fh) {
209 $fh->recv($recv_data, 65536);
210 $data = reply_handler($recv_data, $port);
211 $fh->send($data);
212
213 } else {
214 $fh->recv($recv_data, 65536);
215 unless (length $recv_data) {
216 $sel->remove($fh);
217 $fh->close;
218 next;
219 }
220
221 again:
222 my $len = unpack("n", $recv_data);
223 $data = substr $recv_data, 2, $len;
224 $data = reply_handler($data, $port, tcp => 1);
225 $data = pack("n", length $data) . $data;
226 $fh->send($data);
227 $recv_data = substr $recv_data, 2 + $len;
228 goto again if length $recv_data;
229 }
230 }
231 }
232 }
233
234 ###############################################################################