11274Szelenkov@nginx.comimport re 2*1477Szelenkov@nginx.comimport socket 31274Szelenkov@nginx.comimport time 41274Szelenkov@nginx.comimport unittest 5*1477Szelenkov@nginx.com 61274Szelenkov@nginx.comfrom unit.applications.lang.python import TestApplicationPython 71274Szelenkov@nginx.com 81274Szelenkov@nginx.com 91274Szelenkov@nginx.comclass TestProxy(TestApplicationPython): 101467Szelenkov@nginx.com prerequisites = {'modules': {'python': 'any'}} 111274Szelenkov@nginx.com 121274Szelenkov@nginx.com SERVER_PORT = 7999 131274Szelenkov@nginx.com 141353Smax.romanov@nginx.com @staticmethod 151353Smax.romanov@nginx.com def run_server(server_port): 161274Szelenkov@nginx.com sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 171274Szelenkov@nginx.com sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) 181274Szelenkov@nginx.com 191353Smax.romanov@nginx.com server_address = ('', server_port) 201274Szelenkov@nginx.com sock.bind(server_address) 211274Szelenkov@nginx.com sock.listen(5) 221274Szelenkov@nginx.com 231274Szelenkov@nginx.com def recvall(sock): 241274Szelenkov@nginx.com buff_size = 4096 251274Szelenkov@nginx.com data = b'' 261274Szelenkov@nginx.com while True: 271274Szelenkov@nginx.com part = sock.recv(buff_size) 281274Szelenkov@nginx.com data += part 291274Szelenkov@nginx.com if len(part) < buff_size: 301274Szelenkov@nginx.com break 311274Szelenkov@nginx.com return data 321274Szelenkov@nginx.com 331274Szelenkov@nginx.com req = b"""HTTP/1.1 200 OK 341274Szelenkov@nginx.comContent-Length: 10 351274Szelenkov@nginx.com 361274Szelenkov@nginx.com""" 371274Szelenkov@nginx.com 381274Szelenkov@nginx.com while True: 391274Szelenkov@nginx.com connection, client_address = sock.accept() 401274Szelenkov@nginx.com 411274Szelenkov@nginx.com data = recvall(connection).decode() 421274Szelenkov@nginx.com 431274Szelenkov@nginx.com to_send = req 441274Szelenkov@nginx.com 451274Szelenkov@nginx.com m = re.search('X-Len: (\d+)', data) 461274Szelenkov@nginx.com if m: 471274Szelenkov@nginx.com to_send += b'X' * int(m.group(1)) 481274Szelenkov@nginx.com 491274Szelenkov@nginx.com connection.sendall(to_send) 501274Szelenkov@nginx.com 511274Szelenkov@nginx.com connection.close() 521274Szelenkov@nginx.com 531274Szelenkov@nginx.com def get_http10(self, *args, **kwargs): 541274Szelenkov@nginx.com return self.get(*args, http_10=True, **kwargs) 551274Szelenkov@nginx.com 561274Szelenkov@nginx.com def post_http10(self, *args, **kwargs): 571274Szelenkov@nginx.com return self.post(*args, http_10=True, **kwargs) 581274Szelenkov@nginx.com 591274Szelenkov@nginx.com def setUp(self): 601274Szelenkov@nginx.com super().setUp() 611274Szelenkov@nginx.com 621353Smax.romanov@nginx.com self.run_process(self.run_server, self.SERVER_PORT) 631274Szelenkov@nginx.com self.waitforsocket(self.SERVER_PORT) 641274Szelenkov@nginx.com 651274Szelenkov@nginx.com self.assertIn( 661274Szelenkov@nginx.com 'success', 671274Szelenkov@nginx.com self.conf( 681274Szelenkov@nginx.com { 691274Szelenkov@nginx.com "listeners": { 701274Szelenkov@nginx.com "*:7080": {"pass": "routes"}, 711274Szelenkov@nginx.com "*:7081": {"pass": "applications/mirror"}, 721274Szelenkov@nginx.com }, 731274Szelenkov@nginx.com "routes": [{"action": {"proxy": "http://127.0.0.1:7081"}}], 741274Szelenkov@nginx.com "applications": { 751274Szelenkov@nginx.com "mirror": { 761274Szelenkov@nginx.com "type": "python", 771274Szelenkov@nginx.com "processes": {"spare": 0}, 781274Szelenkov@nginx.com "path": self.current_dir + "/python/mirror", 791274Szelenkov@nginx.com "working_directory": self.current_dir 801274Szelenkov@nginx.com + "/python/mirror", 811274Szelenkov@nginx.com "module": "wsgi", 821274Szelenkov@nginx.com }, 831274Szelenkov@nginx.com "custom_header": { 841274Szelenkov@nginx.com "type": "python", 851274Szelenkov@nginx.com "processes": {"spare": 0}, 861274Szelenkov@nginx.com "path": self.current_dir + "/python/custom_header", 871274Szelenkov@nginx.com "working_directory": self.current_dir 881274Szelenkov@nginx.com + "/python/custom_header", 891274Szelenkov@nginx.com "module": "wsgi", 901274Szelenkov@nginx.com }, 911274Szelenkov@nginx.com "delayed": { 921274Szelenkov@nginx.com "type": "python", 931274Szelenkov@nginx.com "processes": {"spare": 0}, 941274Szelenkov@nginx.com "path": self.current_dir + "/python/delayed", 951274Szelenkov@nginx.com "working_directory": self.current_dir 961274Szelenkov@nginx.com + "/python/delayed", 971274Szelenkov@nginx.com "module": "wsgi", 981274Szelenkov@nginx.com }, 991274Szelenkov@nginx.com }, 1001274Szelenkov@nginx.com } 1011274Szelenkov@nginx.com ), 1021274Szelenkov@nginx.com 'proxy initial configuration', 1031274Szelenkov@nginx.com ) 1041274Szelenkov@nginx.com 1051274Szelenkov@nginx.com def test_proxy_http10(self): 1061274Szelenkov@nginx.com for _ in range(10): 1071274Szelenkov@nginx.com self.assertEqual(self.get_http10()['status'], 200, 'status') 1081274Szelenkov@nginx.com 1091274Szelenkov@nginx.com def test_proxy_chain(self): 1101274Szelenkov@nginx.com self.assertIn( 1111274Szelenkov@nginx.com 'success', 1121274Szelenkov@nginx.com self.conf( 1131274Szelenkov@nginx.com { 1141274Szelenkov@nginx.com "listeners": { 1151274Szelenkov@nginx.com "*:7080": {"pass": "routes/first"}, 1161274Szelenkov@nginx.com "*:7081": {"pass": "routes/second"}, 1171274Szelenkov@nginx.com "*:7082": {"pass": "routes/third"}, 1181274Szelenkov@nginx.com "*:7083": {"pass": "routes/fourth"}, 1191274Szelenkov@nginx.com "*:7084": {"pass": "routes/fifth"}, 1201274Szelenkov@nginx.com "*:7085": {"pass": "applications/mirror"}, 1211274Szelenkov@nginx.com }, 1221274Szelenkov@nginx.com "routes": { 1231274Szelenkov@nginx.com "first": [ 1241274Szelenkov@nginx.com {"action": {"proxy": "http://127.0.0.1:7081"}} 1251274Szelenkov@nginx.com ], 1261274Szelenkov@nginx.com "second": [ 1271274Szelenkov@nginx.com {"action": {"proxy": "http://127.0.0.1:7082"}} 1281274Szelenkov@nginx.com ], 1291274Szelenkov@nginx.com "third": [ 1301274Szelenkov@nginx.com {"action": {"proxy": "http://127.0.0.1:7083"}} 1311274Szelenkov@nginx.com ], 1321274Szelenkov@nginx.com "fourth": [ 1331274Szelenkov@nginx.com {"action": {"proxy": "http://127.0.0.1:7084"}} 1341274Szelenkov@nginx.com ], 1351274Szelenkov@nginx.com "fifth": [ 1361274Szelenkov@nginx.com {"action": {"proxy": "http://127.0.0.1:7085"}} 1371274Szelenkov@nginx.com ], 1381274Szelenkov@nginx.com }, 1391274Szelenkov@nginx.com "applications": { 1401274Szelenkov@nginx.com "mirror": { 1411274Szelenkov@nginx.com "type": "python", 1421274Szelenkov@nginx.com "processes": {"spare": 0}, 1431274Szelenkov@nginx.com "path": self.current_dir + "/python/mirror", 1441274Szelenkov@nginx.com "working_directory": self.current_dir 1451274Szelenkov@nginx.com + "/python/mirror", 1461274Szelenkov@nginx.com "module": "wsgi", 1471274Szelenkov@nginx.com } 1481274Szelenkov@nginx.com }, 1491274Szelenkov@nginx.com } 1501274Szelenkov@nginx.com ), 1511274Szelenkov@nginx.com 'proxy chain configuration', 1521274Szelenkov@nginx.com ) 1531274Szelenkov@nginx.com 1541274Szelenkov@nginx.com self.assertEqual(self.get_http10()['status'], 200, 'status') 1551274Szelenkov@nginx.com 1561274Szelenkov@nginx.com def test_proxy_body(self): 1571274Szelenkov@nginx.com payload = '0123456789' 1581274Szelenkov@nginx.com for _ in range(10): 1591274Szelenkov@nginx.com resp = self.post_http10(body=payload) 1601274Szelenkov@nginx.com 1611274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1621274Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1631274Szelenkov@nginx.com 1641274Szelenkov@nginx.com payload = 'X' * 4096 1651274Szelenkov@nginx.com for _ in range(10): 1661274Szelenkov@nginx.com resp = self.post_http10(body=payload) 1671274Szelenkov@nginx.com 1681274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1691274Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1701274Szelenkov@nginx.com 1711274Szelenkov@nginx.com payload = 'X' * 4097 1721274Szelenkov@nginx.com for _ in range(10): 1731274Szelenkov@nginx.com resp = self.post_http10(body=payload) 1741274Szelenkov@nginx.com 1751274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1761274Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1771274Szelenkov@nginx.com 1781274Szelenkov@nginx.com payload = 'X' * 4096 * 256 1791274Szelenkov@nginx.com for _ in range(10): 1801274Szelenkov@nginx.com resp = self.post_http10(body=payload, read_buffer_size=4096 * 128) 1811274Szelenkov@nginx.com 1821274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1831274Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1841274Szelenkov@nginx.com 1851274Szelenkov@nginx.com payload = 'X' * 4096 * 257 1861274Szelenkov@nginx.com for _ in range(10): 1871274Szelenkov@nginx.com resp = self.post_http10(body=payload, read_buffer_size=4096 * 128) 1881274Szelenkov@nginx.com 1891274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1901274Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1911274Szelenkov@nginx.com 1921366Szelenkov@nginx.com self.conf({'http': {'max_body_size': 32 * 1024 * 1024}}, 'settings') 1931366Szelenkov@nginx.com 1941366Szelenkov@nginx.com payload = '0123456789abcdef' * 32 * 64 * 1024 1951366Szelenkov@nginx.com resp = self.post_http10(body=payload, read_buffer_size=1024 * 1024) 1961366Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 1971366Szelenkov@nginx.com self.assertEqual(resp['body'], payload, 'body') 1981366Szelenkov@nginx.com 1991274Szelenkov@nginx.com def test_proxy_parallel(self): 2001274Szelenkov@nginx.com payload = 'X' * 4096 * 257 2011274Szelenkov@nginx.com buff_size = 4096 * 258 2021274Szelenkov@nginx.com 2031274Szelenkov@nginx.com socks = [] 2041274Szelenkov@nginx.com for i in range(10): 2051274Szelenkov@nginx.com _, sock = self.post_http10( 2061274Szelenkov@nginx.com body=payload + str(i), 2071274Szelenkov@nginx.com start=True, 2081274Szelenkov@nginx.com no_recv=True, 2091274Szelenkov@nginx.com read_buffer_size=buff_size, 2101274Szelenkov@nginx.com ) 2111274Szelenkov@nginx.com socks.append(sock) 2121274Szelenkov@nginx.com 2131274Szelenkov@nginx.com for i in range(10): 2141274Szelenkov@nginx.com resp = self.recvall(socks[i], buff_size=buff_size).decode() 2151274Szelenkov@nginx.com socks[i].close() 2161274Szelenkov@nginx.com 2171274Szelenkov@nginx.com resp = self._resp_to_dict(resp) 2181274Szelenkov@nginx.com 2191274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 2201274Szelenkov@nginx.com self.assertEqual(resp['body'], payload + str(i), 'body') 2211274Szelenkov@nginx.com 2221274Szelenkov@nginx.com def test_proxy_header(self): 2231274Szelenkov@nginx.com self.assertIn( 2241274Szelenkov@nginx.com 'success', 2251274Szelenkov@nginx.com self.conf( 2261274Szelenkov@nginx.com {"pass": "applications/custom_header"}, 'listeners/*:7081' 2271274Szelenkov@nginx.com ), 2281274Szelenkov@nginx.com 'custom_header configure', 2291274Szelenkov@nginx.com ) 2301274Szelenkov@nginx.com 2311274Szelenkov@nginx.com header_value = 'blah' 2321274Szelenkov@nginx.com self.assertEqual( 2331274Szelenkov@nginx.com self.get_http10( 2341274Szelenkov@nginx.com headers={'Host': 'localhost', 'Custom-Header': header_value} 2351274Szelenkov@nginx.com )['headers']['Custom-Header'], 2361274Szelenkov@nginx.com header_value, 2371274Szelenkov@nginx.com 'custom header', 2381274Szelenkov@nginx.com ) 2391274Szelenkov@nginx.com 2401274Szelenkov@nginx.com header_value = '(),/:;<=>?@[\]{}\t !#$%&\'*+-.^_`|~' 2411274Szelenkov@nginx.com self.assertEqual( 2421274Szelenkov@nginx.com self.get_http10( 2431274Szelenkov@nginx.com headers={'Host': 'localhost', 'Custom-Header': header_value} 2441274Szelenkov@nginx.com )['headers']['Custom-Header'], 2451274Szelenkov@nginx.com header_value, 2461274Szelenkov@nginx.com 'custom header 2', 2471274Szelenkov@nginx.com ) 2481274Szelenkov@nginx.com 2491274Szelenkov@nginx.com header_value = 'X' * 4096 2501274Szelenkov@nginx.com self.assertEqual( 2511274Szelenkov@nginx.com self.get_http10( 2521274Szelenkov@nginx.com headers={'Host': 'localhost', 'Custom-Header': header_value} 2531274Szelenkov@nginx.com )['headers']['Custom-Header'], 2541274Szelenkov@nginx.com header_value, 2551274Szelenkov@nginx.com 'custom header 3', 2561274Szelenkov@nginx.com ) 2571274Szelenkov@nginx.com 2581274Szelenkov@nginx.com header_value = 'X' * 8191 2591274Szelenkov@nginx.com self.assertEqual( 2601274Szelenkov@nginx.com self.get_http10( 2611274Szelenkov@nginx.com headers={'Host': 'localhost', 'Custom-Header': header_value} 2621274Szelenkov@nginx.com )['headers']['Custom-Header'], 2631274Szelenkov@nginx.com header_value, 2641274Szelenkov@nginx.com 'custom header 4', 2651274Szelenkov@nginx.com ) 2661274Szelenkov@nginx.com 2671274Szelenkov@nginx.com header_value = 'X' * 8192 2681274Szelenkov@nginx.com self.assertEqual( 2691274Szelenkov@nginx.com self.get_http10( 2701274Szelenkov@nginx.com headers={'Host': 'localhost', 'Custom-Header': header_value} 2711274Szelenkov@nginx.com )['status'], 2721274Szelenkov@nginx.com 431, 2731274Szelenkov@nginx.com 'custom header 5', 2741274Szelenkov@nginx.com ) 2751274Szelenkov@nginx.com 2761274Szelenkov@nginx.com def test_proxy_fragmented(self): 2771274Szelenkov@nginx.com _, sock = self.http( 2781274Szelenkov@nginx.com b"""GET / HTT""", raw=True, start=True, no_recv=True 2791274Szelenkov@nginx.com ) 2801274Szelenkov@nginx.com 2811274Szelenkov@nginx.com time.sleep(1) 2821274Szelenkov@nginx.com 2831274Szelenkov@nginx.com sock.sendall("P/1.0\r\nHost: localhos".encode()) 2841274Szelenkov@nginx.com 2851274Szelenkov@nginx.com time.sleep(1) 2861274Szelenkov@nginx.com 2871274Szelenkov@nginx.com sock.sendall("t\r\n\r\n".encode()) 2881274Szelenkov@nginx.com 2891274Szelenkov@nginx.com self.assertRegex( 2901274Szelenkov@nginx.com self.recvall(sock).decode(), '200 OK', 'fragmented send' 2911274Szelenkov@nginx.com ) 2921274Szelenkov@nginx.com sock.close() 2931274Szelenkov@nginx.com 2941274Szelenkov@nginx.com def test_proxy_fragmented_close(self): 2951274Szelenkov@nginx.com _, sock = self.http( 2961274Szelenkov@nginx.com b"""GET / HTT""", raw=True, start=True, no_recv=True 2971274Szelenkov@nginx.com ) 2981274Szelenkov@nginx.com 2991274Szelenkov@nginx.com time.sleep(1) 3001274Szelenkov@nginx.com 3011274Szelenkov@nginx.com sock.sendall("P/1.0\r\nHo".encode()) 3021274Szelenkov@nginx.com 3031274Szelenkov@nginx.com sock.close() 3041274Szelenkov@nginx.com 3051274Szelenkov@nginx.com def test_proxy_fragmented_body(self): 3061274Szelenkov@nginx.com _, sock = self.http( 3071274Szelenkov@nginx.com b"""GET / HTT""", raw=True, start=True, no_recv=True 3081274Szelenkov@nginx.com ) 3091274Szelenkov@nginx.com 3101274Szelenkov@nginx.com time.sleep(1) 3111274Szelenkov@nginx.com 3121274Szelenkov@nginx.com sock.sendall("P/1.0\r\nHost: localhost\r\n".encode()) 3131274Szelenkov@nginx.com sock.sendall("Content-Length: 30000\r\n".encode()) 3141274Szelenkov@nginx.com 3151274Szelenkov@nginx.com time.sleep(1) 3161274Szelenkov@nginx.com 3171274Szelenkov@nginx.com sock.sendall("\r\n".encode()) 3181274Szelenkov@nginx.com sock.sendall(("X" * 10000).encode()) 3191274Szelenkov@nginx.com 3201274Szelenkov@nginx.com time.sleep(1) 3211274Szelenkov@nginx.com 3221274Szelenkov@nginx.com sock.sendall(("X" * 10000).encode()) 3231274Szelenkov@nginx.com 3241274Szelenkov@nginx.com time.sleep(1) 3251274Szelenkov@nginx.com 3261274Szelenkov@nginx.com sock.sendall(("X" * 10000).encode()) 3271274Szelenkov@nginx.com 3281274Szelenkov@nginx.com resp = self._resp_to_dict(self.recvall(sock).decode()) 3291274Szelenkov@nginx.com sock.close() 3301274Szelenkov@nginx.com 3311274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 3321274Szelenkov@nginx.com self.assertEqual(resp['body'], "X" * 30000, 'body') 3331274Szelenkov@nginx.com 3341274Szelenkov@nginx.com def test_proxy_fragmented_body_close(self): 3351274Szelenkov@nginx.com _, sock = self.http( 3361274Szelenkov@nginx.com b"""GET / HTT""", raw=True, start=True, no_recv=True 3371274Szelenkov@nginx.com ) 3381274Szelenkov@nginx.com 3391274Szelenkov@nginx.com time.sleep(1) 3401274Szelenkov@nginx.com 3411274Szelenkov@nginx.com sock.sendall("P/1.0\r\nHost: localhost\r\n".encode()) 3421274Szelenkov@nginx.com sock.sendall("Content-Length: 30000\r\n".encode()) 3431274Szelenkov@nginx.com 3441274Szelenkov@nginx.com time.sleep(1) 3451274Szelenkov@nginx.com 3461274Szelenkov@nginx.com sock.sendall("\r\n".encode()) 3471274Szelenkov@nginx.com sock.sendall(("X" * 10000).encode()) 3481274Szelenkov@nginx.com 3491274Szelenkov@nginx.com sock.close() 3501274Szelenkov@nginx.com 3511274Szelenkov@nginx.com def test_proxy_nowhere(self): 3521274Szelenkov@nginx.com self.assertIn( 3531274Szelenkov@nginx.com 'success', 3541274Szelenkov@nginx.com self.conf( 3551274Szelenkov@nginx.com [{"action": {"proxy": "http://127.0.0.1:7082"}}], 'routes' 3561274Szelenkov@nginx.com ), 3571274Szelenkov@nginx.com 'proxy path changed', 3581274Szelenkov@nginx.com ) 3591274Szelenkov@nginx.com 3601274Szelenkov@nginx.com self.assertEqual(self.get_http10()['status'], 502, 'status') 3611274Szelenkov@nginx.com 3621274Szelenkov@nginx.com def test_proxy_ipv6(self): 3631274Szelenkov@nginx.com self.assertIn( 3641274Szelenkov@nginx.com 'success', 3651274Szelenkov@nginx.com self.conf( 3661274Szelenkov@nginx.com { 3671274Szelenkov@nginx.com "*:7080": {"pass": "routes"}, 3681274Szelenkov@nginx.com "[::1]:7081": {'application': 'mirror'}, 3691274Szelenkov@nginx.com }, 3701274Szelenkov@nginx.com 'listeners', 3711274Szelenkov@nginx.com ), 3721274Szelenkov@nginx.com 'add ipv6 listener configure', 3731274Szelenkov@nginx.com ) 3741274Szelenkov@nginx.com 3751274Szelenkov@nginx.com self.assertIn( 3761274Szelenkov@nginx.com 'success', 3771274Szelenkov@nginx.com self.conf([{"action": {"proxy": "http://[::1]:7081"}}], 'routes'), 3781274Szelenkov@nginx.com 'proxy ipv6 configure', 3791274Szelenkov@nginx.com ) 3801274Szelenkov@nginx.com 3811274Szelenkov@nginx.com self.assertEqual(self.get_http10()['status'], 200, 'status') 3821274Szelenkov@nginx.com 3831274Szelenkov@nginx.com def test_proxy_unix(self): 3841274Szelenkov@nginx.com addr = self.testdir + '/sock' 3851274Szelenkov@nginx.com 3861274Szelenkov@nginx.com self.assertIn( 3871274Szelenkov@nginx.com 'success', 3881274Szelenkov@nginx.com self.conf( 3891274Szelenkov@nginx.com { 3901274Szelenkov@nginx.com "*:7080": {"pass": "routes"}, 3911274Szelenkov@nginx.com "unix:" + addr: {'application': 'mirror'}, 3921274Szelenkov@nginx.com }, 3931274Szelenkov@nginx.com 'listeners', 3941274Szelenkov@nginx.com ), 3951274Szelenkov@nginx.com 'add unix listener configure', 3961274Szelenkov@nginx.com ) 3971274Szelenkov@nginx.com 3981274Szelenkov@nginx.com self.assertIn( 3991274Szelenkov@nginx.com 'success', 4001274Szelenkov@nginx.com self.conf( 4011274Szelenkov@nginx.com [{"action": {"proxy": 'http://unix:' + addr}}], 'routes' 4021274Szelenkov@nginx.com ), 4031274Szelenkov@nginx.com 'proxy unix configure', 4041274Szelenkov@nginx.com ) 4051274Szelenkov@nginx.com 4061274Szelenkov@nginx.com self.assertEqual(self.get_http10()['status'], 200, 'status') 4071274Szelenkov@nginx.com 4081274Szelenkov@nginx.com def test_proxy_delayed(self): 4091274Szelenkov@nginx.com self.assertIn( 4101274Szelenkov@nginx.com 'success', 4111274Szelenkov@nginx.com self.conf( 4121274Szelenkov@nginx.com {"pass": "applications/delayed"}, 'listeners/*:7081' 4131274Szelenkov@nginx.com ), 4141274Szelenkov@nginx.com 'delayed configure', 4151274Szelenkov@nginx.com ) 4161274Szelenkov@nginx.com 4171274Szelenkov@nginx.com body = '0123456789' * 1000 4181274Szelenkov@nginx.com resp = self.post_http10( 4191274Szelenkov@nginx.com headers={ 4201274Szelenkov@nginx.com 'Host': 'localhost', 4211274Szelenkov@nginx.com 'Content-Type': 'text/html', 4221274Szelenkov@nginx.com 'Content-Length': str(len(body)), 4231274Szelenkov@nginx.com 'X-Parts': '2', 4241274Szelenkov@nginx.com 'X-Delay': '1', 4251274Szelenkov@nginx.com }, 4261274Szelenkov@nginx.com body=body, 4271274Szelenkov@nginx.com ) 4281274Szelenkov@nginx.com 4291274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 4301274Szelenkov@nginx.com self.assertEqual(resp['body'], body, 'body') 4311274Szelenkov@nginx.com 4321274Szelenkov@nginx.com resp = self.post_http10( 4331274Szelenkov@nginx.com headers={ 4341274Szelenkov@nginx.com 'Host': 'localhost', 4351274Szelenkov@nginx.com 'Content-Type': 'text/html', 4361274Szelenkov@nginx.com 'Content-Length': str(len(body)), 4371274Szelenkov@nginx.com 'X-Parts': '2', 4381274Szelenkov@nginx.com 'X-Delay': '1', 4391274Szelenkov@nginx.com }, 4401274Szelenkov@nginx.com body=body, 4411274Szelenkov@nginx.com ) 4421274Szelenkov@nginx.com 4431274Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 4441274Szelenkov@nginx.com self.assertEqual(resp['body'], body, 'body') 4451274Szelenkov@nginx.com 4461274Szelenkov@nginx.com def test_proxy_delayed_close(self): 4471274Szelenkov@nginx.com self.assertIn( 4481274Szelenkov@nginx.com 'success', 4491274Szelenkov@nginx.com self.conf( 4501274Szelenkov@nginx.com {"pass": "applications/delayed"}, 'listeners/*:7081' 4511274Szelenkov@nginx.com ), 4521274Szelenkov@nginx.com 'delayed configure', 4531274Szelenkov@nginx.com ) 4541274Szelenkov@nginx.com 4551274Szelenkov@nginx.com _, sock = self.post_http10( 4561274Szelenkov@nginx.com headers={ 4571274Szelenkov@nginx.com 'Host': 'localhost', 4581274Szelenkov@nginx.com 'Content-Type': 'text/html', 4591274Szelenkov@nginx.com 'Content-Length': '10000', 4601274Szelenkov@nginx.com 'X-Parts': '3', 4611274Szelenkov@nginx.com 'X-Delay': '1', 4621274Szelenkov@nginx.com }, 4631274Szelenkov@nginx.com body='0123456789' * 1000, 4641274Szelenkov@nginx.com start=True, 4651274Szelenkov@nginx.com no_recv=True, 4661274Szelenkov@nginx.com ) 4671274Szelenkov@nginx.com 4681274Szelenkov@nginx.com self.assertRegex( 4691274Szelenkov@nginx.com sock.recv(100).decode(), '200 OK', 'first' 4701274Szelenkov@nginx.com ) 4711274Szelenkov@nginx.com sock.close() 4721274Szelenkov@nginx.com 4731274Szelenkov@nginx.com _, sock = self.post_http10( 4741274Szelenkov@nginx.com headers={ 4751274Szelenkov@nginx.com 'Host': 'localhost', 4761274Szelenkov@nginx.com 'Content-Type': 'text/html', 4771274Szelenkov@nginx.com 'Content-Length': '10000', 4781274Szelenkov@nginx.com 'X-Parts': '3', 4791274Szelenkov@nginx.com 'X-Delay': '1', 4801274Szelenkov@nginx.com }, 4811274Szelenkov@nginx.com body='0123456789' * 1000, 4821274Szelenkov@nginx.com start=True, 4831274Szelenkov@nginx.com no_recv=True, 4841274Szelenkov@nginx.com ) 4851274Szelenkov@nginx.com 4861274Szelenkov@nginx.com self.assertRegex( 4871274Szelenkov@nginx.com sock.recv(100).decode(), '200 OK', 'second' 4881274Szelenkov@nginx.com ) 4891274Szelenkov@nginx.com sock.close() 4901274Szelenkov@nginx.com 4911274Szelenkov@nginx.com @unittest.skip('not yet') 4921274Szelenkov@nginx.com def test_proxy_content_length(self): 4931274Szelenkov@nginx.com self.assertIn( 4941274Szelenkov@nginx.com 'success', 4951274Szelenkov@nginx.com self.conf( 4961274Szelenkov@nginx.com [ 4971274Szelenkov@nginx.com { 4981274Szelenkov@nginx.com "action": { 4991274Szelenkov@nginx.com "proxy": "http://127.0.0.1:" 5001274Szelenkov@nginx.com + str(self.SERVER_PORT) 5011274Szelenkov@nginx.com } 5021274Szelenkov@nginx.com } 5031274Szelenkov@nginx.com ], 5041274Szelenkov@nginx.com 'routes', 5051274Szelenkov@nginx.com ), 5061274Szelenkov@nginx.com 'proxy backend configure', 5071274Szelenkov@nginx.com ) 5081274Szelenkov@nginx.com 5091274Szelenkov@nginx.com resp = self.get_http10() 5101274Szelenkov@nginx.com self.assertEqual(len(resp['body']), 0, 'body lt Content-Length 0') 5111274Szelenkov@nginx.com 5121274Szelenkov@nginx.com resp = self.get_http10(headers={'Host': 'localhost', 'X-Len': '5'}) 5131274Szelenkov@nginx.com self.assertEqual(len(resp['body']), 5, 'body lt Content-Length 5') 5141274Szelenkov@nginx.com 5151274Szelenkov@nginx.com resp = self.get_http10(headers={'Host': 'localhost', 'X-Len': '9'}) 5161274Szelenkov@nginx.com self.assertEqual(len(resp['body']), 9, 'body lt Content-Length 9') 5171274Szelenkov@nginx.com 5181274Szelenkov@nginx.com resp = self.get_http10(headers={'Host': 'localhost', 'X-Len': '11'}) 5191274Szelenkov@nginx.com self.assertEqual(len(resp['body']), 10, 'body gt Content-Length 11') 5201274Szelenkov@nginx.com 5211274Szelenkov@nginx.com resp = self.get_http10(headers={'Host': 'localhost', 'X-Len': '15'}) 5221274Szelenkov@nginx.com self.assertEqual(len(resp['body']), 10, 'body gt Content-Length 15') 5231274Szelenkov@nginx.com 5241274Szelenkov@nginx.com def test_proxy_invalid(self): 5251476Szelenkov@nginx.com def check_proxy(proxy): 5261476Szelenkov@nginx.com self.assertIn( 5271476Szelenkov@nginx.com 'error', 5281476Szelenkov@nginx.com self.conf([{"action": {"proxy": proxy}}], 'routes'), 5291476Szelenkov@nginx.com 'proxy invalid', 5301476Szelenkov@nginx.com ) 5311476Szelenkov@nginx.com 5321476Szelenkov@nginx.com check_proxy('blah') 5331476Szelenkov@nginx.com check_proxy('/blah') 5341476Szelenkov@nginx.com check_proxy('unix:/blah') 5351476Szelenkov@nginx.com check_proxy('http://blah') 5361476Szelenkov@nginx.com check_proxy('http://127.0.0.1') 5371476Szelenkov@nginx.com check_proxy('http://127.0.0.1:') 5381476Szelenkov@nginx.com check_proxy('http://127.0.0.1:blah') 5391476Szelenkov@nginx.com check_proxy('http://127.0.0.1:-1') 5401476Szelenkov@nginx.com check_proxy('http://127.0.0.1:7080b') 5411476Szelenkov@nginx.com check_proxy('http://[]') 5421476Szelenkov@nginx.com check_proxy('http://[]:7080') 5431476Szelenkov@nginx.com check_proxy('http://[:]:7080') 5441476Szelenkov@nginx.com check_proxy('http://[::7080') 5451274Szelenkov@nginx.com 5461274Szelenkov@nginx.com def test_proxy_loop(self): 5471465Szelenkov@nginx.com self.skip_alerts.extend( 5481465Szelenkov@nginx.com [ 5491465Szelenkov@nginx.com r'socket.*failed', 5501465Szelenkov@nginx.com r'accept.*failed', 5511465Szelenkov@nginx.com r'new connections are not accepted', 5521465Szelenkov@nginx.com ] 5531465Szelenkov@nginx.com ) 5541274Szelenkov@nginx.com self.conf( 5551274Szelenkov@nginx.com { 5561274Szelenkov@nginx.com "listeners": { 5571274Szelenkov@nginx.com "*:7080": {"pass": "routes"}, 5581274Szelenkov@nginx.com "*:7081": {"pass": "applications/mirror"}, 5591274Szelenkov@nginx.com "*:7082": {"pass": "routes"}, 5601274Szelenkov@nginx.com }, 5611274Szelenkov@nginx.com "routes": [{"action": {"proxy": "http://127.0.0.1:7082"}}], 5621274Szelenkov@nginx.com "applications": { 5631274Szelenkov@nginx.com "mirror": { 5641274Szelenkov@nginx.com "type": "python", 5651274Szelenkov@nginx.com "processes": {"spare": 0}, 5661274Szelenkov@nginx.com "path": self.current_dir + "/python/mirror", 5671274Szelenkov@nginx.com "working_directory": self.current_dir 5681274Szelenkov@nginx.com + "/python/mirror", 5691274Szelenkov@nginx.com "module": "wsgi", 5701274Szelenkov@nginx.com }, 5711274Szelenkov@nginx.com }, 5721274Szelenkov@nginx.com } 5731274Szelenkov@nginx.com ) 5741274Szelenkov@nginx.com 5751274Szelenkov@nginx.com self.get_http10(no_recv=True) 5761465Szelenkov@nginx.com self.get_http10(read_timeout=1) 5771274Szelenkov@nginx.com 5781274Szelenkov@nginx.comif __name__ == '__main__': 5791274Szelenkov@nginx.com TestProxy.main() 580