Mercurial > hg > nginx-tests

comparison js_body_filter.t @ 1656:759efa4abc76

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Tests: added tests for js_body_filter directive.
author Dmitry Volyntsev <xeioex@nginx.com>
date Fri, 19 Feb 2021 12:20:57 +0000
parents
children
comparison
equal deleted inserted replaced
1655:666d54ab5036 1656:759efa4abc76
1 #!/usr/bin/perl
2
3 # (C) Dmitry Volyntsev
4 # (C) Nginx, Inc.
5
6 # Tests for http njs module, body filter.
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 js_import test.js;
39
40 server {
41 listen 127.0.0.1:8080;
42 server_name localhost;
43
44 location /njs {
45 js_content test.njs;
46 }
47
48 location /append {
49 js_body_filter test.append;
50 proxy_pass http://127.0.0.1:8081/source;
51 }
52
53 location /buffer_type {
54 js_body_filter test.buffer_type buffer_type=buffer;
55 proxy_pass http://127.0.0.1:8081/source;
56 }
57
58 location /forward {
59 js_body_filter test.forward buffer_type=string;
60 proxy_pass http://127.0.0.1:8081/source;
61 }
62
63 location /filter {
64 proxy_buffering off;
65 js_body_filter test.filter;
66 proxy_pass http://127.0.0.1:8081/source;
67 }
68
69 location /prepend {
70 js_body_filter test.prepend;
71 proxy_pass http://127.0.0.1:8081/source;
72 }
73 }
74
75 server {
76 listen 127.0.0.1:8081;
77 server_name localhost;
78
79 location /source {
80 postpone_output 1;
81 js_content test.source;
82 }
83 }
84 }
85
86 EOF
87
88 $t->write_file('test.js', <<EOF);
89 function test_njs(r) {
90 r.return(200, njs.version);
91 }
92
93 function append(r, data, flags) {
94 r.sendBuffer(data, {last:false});
95
96 if (flags.last) {
97 r.sendBuffer("XXX", flags);
98 }
99 }
100
101 var collect = Buffer.from([]);
102 function buffer_type(r, data, flags) {
103 collect = Buffer.concat([collect, data]);
104
105 if (flags.last) {
106 r.sendBuffer(collect, flags);
107 }
108 }
109
110 function chain(chunks, i) {
111 if (i < chunks.length) {
112 chunks.r.send(chunks[i++]);
113 setTimeout(chunks.chain, chunks.delay, chunks, i);
114
115 } else {
116 chunks.r.finish();
117 }
118 }
119
120 function source(r) {
121 var chunks = ['AAA', 'BB', 'C', 'DDDD'];
122 chunks.delay = 5;
123 chunks.r = r;
124 chunks.chain = chain;
125
126 r.status = 200;
127 r.sendHeader();
128 chain(chunks, 0);
129 }
130
131 function filter(r, data, flags) {
132 if (flags.last || data.length >= Number(r.args.len)) {
133 r.sendBuffer(`\${data}|`, flags);
134
135 if (r.args.dup && !flags.last) {
136 r.sendBuffer(data, flags);
137 }
138 }
139 }
140
141 function forward(r, data, flags) {
142 r.sendBuffer(data, flags);
143 }
144
145 function prepend(r, data, flags) {
146 r.sendBuffer("XXX");
147 r.sendBuffer(data, flags);
148 r.done();
149 }
150
151 export default {njs: test_njs, append, buffer_type, filter, forward,
152 prepend, source};
153
154 EOF
155
156 $t->try_run('no njs body filter')->plan(6);
157
158 ###############################################################################
159
160 like(http_get('/append'), qr/AAABBCDDDDXXX/, 'append');
161 like(http_get('/buffer_type'), qr/AAABBCDDDD/, 'buffer type');
162 like(http_get('/forward'), qr/AAABBCDDDD/, 'forward');
163 like(http_get('/filter?len=3'), qr/AAA|DDDD|/, 'filter 3');
164 like(http_get('/filter?len=2&dup=1'), qr/AAA|AAABB|BBDDDD|DDDD/,
165 'filter 2 dup');
166 like(http_get('/prepend'), qr/XXXAAABBCDDDD/, 'prepend');
167
168 ###############################################################################