1855Spluknet@nginx.comimport unittest 21019Szelenkov@nginx.comfrom unit.applications.lang.node import TestApplicationNode 3855Spluknet@nginx.com 41017Szelenkov@nginx.com 51019Szelenkov@nginx.comclass TestNodeApplication(TestApplicationNode): 6*1467Szelenkov@nginx.com prerequisites = {'modules': {'node': 'all'}} 7855Spluknet@nginx.com 8855Spluknet@nginx.com def test_node_application_basic(self): 9855Spluknet@nginx.com self.load('basic') 10855Spluknet@nginx.com 11855Spluknet@nginx.com resp = self.get() 121017Szelenkov@nginx.com self.assertEqual( 131017Szelenkov@nginx.com resp['headers']['Content-Type'], 'text/plain', 'basic header' 141017Szelenkov@nginx.com ) 15855Spluknet@nginx.com self.assertEqual(resp['body'], 'Hello World\n', 'basic body') 16855Spluknet@nginx.com 17855Spluknet@nginx.com def test_node_application_seq(self): 18855Spluknet@nginx.com self.load('basic') 19855Spluknet@nginx.com 20855Spluknet@nginx.com self.assertEqual(self.get()['status'], 200, 'seq') 21855Spluknet@nginx.com self.assertEqual(self.get()['status'], 200, 'seq 2') 22855Spluknet@nginx.com 23855Spluknet@nginx.com def test_node_application_variables(self): 24855Spluknet@nginx.com self.load('variables') 25855Spluknet@nginx.com 26855Spluknet@nginx.com body = 'Test body string.' 27855Spluknet@nginx.com 281017Szelenkov@nginx.com resp = self.post( 291017Szelenkov@nginx.com headers={ 301017Szelenkov@nginx.com 'Host': 'localhost', 311017Szelenkov@nginx.com 'Content-Type': 'text/html', 321017Szelenkov@nginx.com 'Custom-Header': 'blah', 331017Szelenkov@nginx.com 'Connection': 'close', 341017Szelenkov@nginx.com }, 351017Szelenkov@nginx.com body=body, 361017Szelenkov@nginx.com ) 37855Spluknet@nginx.com 38855Spluknet@nginx.com self.assertEqual(resp['status'], 200, 'status') 39855Spluknet@nginx.com headers = resp['headers'] 40855Spluknet@nginx.com header_server = headers.pop('Server') 41855Spluknet@nginx.com self.assertRegex(header_server, r'Unit/[\d\.]+', 'server header') 42855Spluknet@nginx.com 43855Spluknet@nginx.com date = headers.pop('Date') 44855Spluknet@nginx.com self.assertEqual(date[-4:], ' GMT', 'date header timezone') 451017Szelenkov@nginx.com self.assertLess( 461017Szelenkov@nginx.com abs(self.date_to_sec_epoch(date) - self.sec_epoch()), 471017Szelenkov@nginx.com 5, 481017Szelenkov@nginx.com 'date header', 491017Szelenkov@nginx.com ) 50855Spluknet@nginx.com 51855Spluknet@nginx.com raw_headers = headers.pop('Request-Raw-Headers') 521017Szelenkov@nginx.com self.assertRegex( 531017Szelenkov@nginx.com raw_headers, 541017Szelenkov@nginx.com r'^(?:Host|localhost|Content-Type|' 551017Szelenkov@nginx.com 'text\/html|Custom-Header|blah|Content-Length|17|Connection|' 561017Szelenkov@nginx.com 'close|,)+$', 571017Szelenkov@nginx.com 'raw headers', 581017Szelenkov@nginx.com ) 59855Spluknet@nginx.com 601017Szelenkov@nginx.com self.assertDictEqual( 611017Szelenkov@nginx.com headers, 621017Szelenkov@nginx.com { 631017Szelenkov@nginx.com 'Connection': 'close', 641017Szelenkov@nginx.com 'Content-Length': str(len(body)), 651017Szelenkov@nginx.com 'Content-Type': 'text/html', 661017Szelenkov@nginx.com 'Request-Method': 'POST', 671017Szelenkov@nginx.com 'Request-Uri': '/', 681017Szelenkov@nginx.com 'Http-Host': 'localhost', 691017Szelenkov@nginx.com 'Server-Protocol': 'HTTP/1.1', 701017Szelenkov@nginx.com 'Custom-Header': 'blah', 711017Szelenkov@nginx.com }, 721017Szelenkov@nginx.com 'headers', 731017Szelenkov@nginx.com ) 74855Spluknet@nginx.com self.assertEqual(resp['body'], body, 'body') 75855Spluknet@nginx.com 76855Spluknet@nginx.com def test_node_application_get_variables(self): 77855Spluknet@nginx.com self.load('get_variables') 78855Spluknet@nginx.com 79855Spluknet@nginx.com resp = self.get(url='/?var1=val1&var2=&var3') 80855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-1'], 'val1', 'GET variables') 81855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-2'], '', 'GET variables 2') 82855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-3'], '', 'GET variables 3') 83855Spluknet@nginx.com 84855Spluknet@nginx.com def test_node_application_post_variables(self): 85855Spluknet@nginx.com self.load('post_variables') 86855Spluknet@nginx.com 871017Szelenkov@nginx.com resp = self.post( 881017Szelenkov@nginx.com headers={ 891017Szelenkov@nginx.com 'Content-Type': 'application/x-www-form-urlencoded', 901017Szelenkov@nginx.com 'Host': 'localhost', 911017Szelenkov@nginx.com 'Connection': 'close', 921017Szelenkov@nginx.com }, 931017Szelenkov@nginx.com body='var1=val1&var2=&var3', 941017Szelenkov@nginx.com ) 95855Spluknet@nginx.com 96855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-1'], 'val1', 'POST variables') 97855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-2'], '', 'POST variables 2') 98855Spluknet@nginx.com self.assertEqual(resp['headers']['X-Var-3'], '', 'POST variables 3') 99855Spluknet@nginx.com 100855Spluknet@nginx.com def test_node_application_404(self): 101855Spluknet@nginx.com self.load('404') 102855Spluknet@nginx.com 103855Spluknet@nginx.com resp = self.get() 104855Spluknet@nginx.com 105855Spluknet@nginx.com self.assertEqual(resp['status'], 404, '404 status') 1061017Szelenkov@nginx.com self.assertRegex( 1071017Szelenkov@nginx.com resp['body'], r'<title>404 Not Found</title>', '404 body' 1081017Szelenkov@nginx.com ) 109855Spluknet@nginx.com 110855Spluknet@nginx.com def test_node_keepalive_body(self): 111855Spluknet@nginx.com self.load('mirror') 112855Spluknet@nginx.com 1131029Szelenkov@nginx.com self.assertEqual(self.get()['status'], 200, 'init') 1141029Szelenkov@nginx.com 1151453Szelenkov@nginx.com body = '0123456789' * 500 1161017Szelenkov@nginx.com (resp, sock) = self.post( 1171017Szelenkov@nginx.com headers={ 1181017Szelenkov@nginx.com 'Host': 'localhost', 1191017Szelenkov@nginx.com 'Connection': 'keep-alive', 1201017Szelenkov@nginx.com 'Content-Type': 'text/html', 1211017Szelenkov@nginx.com }, 1221017Szelenkov@nginx.com start=True, 1231453Szelenkov@nginx.com body=body, 1241029Szelenkov@nginx.com read_timeout=1, 1251017Szelenkov@nginx.com ) 126855Spluknet@nginx.com 127855Spluknet@nginx.com self.assertEqual(resp['body'], '0123456789' * 500, 'keep-alive 1') 128855Spluknet@nginx.com 1291453Szelenkov@nginx.com body = '0123456789' 1301017Szelenkov@nginx.com resp = self.post( 1311017Szelenkov@nginx.com headers={ 1321017Szelenkov@nginx.com 'Host': 'localhost', 1331017Szelenkov@nginx.com 'Connection': 'close', 1341017Szelenkov@nginx.com 'Content-Type': 'text/html', 1351017Szelenkov@nginx.com }, 1361017Szelenkov@nginx.com sock=sock, 1371453Szelenkov@nginx.com body=body, 1381017Szelenkov@nginx.com ) 139855Spluknet@nginx.com 1401453Szelenkov@nginx.com self.assertEqual(resp['body'], body, 'keep-alive 2') 141855Spluknet@nginx.com 142855Spluknet@nginx.com def test_node_application_write_buffer(self): 143855Spluknet@nginx.com self.load('write_buffer') 144855Spluknet@nginx.com 1451017Szelenkov@nginx.com self.assertEqual( 1461295St.nateldemoura@f5.com self.get()['body'], 'buffer', 'write buffer' 1471017Szelenkov@nginx.com ) 148855Spluknet@nginx.com 149855Spluknet@nginx.com def test_node_application_write_callback(self): 150855Spluknet@nginx.com self.load('write_callback') 151855Spluknet@nginx.com 1521017Szelenkov@nginx.com self.assertEqual( 1531017Szelenkov@nginx.com self.get()['body'], 1541295St.nateldemoura@f5.com 'helloworld', 1551017Szelenkov@nginx.com 'write callback order', 1561017Szelenkov@nginx.com ) 1571017Szelenkov@nginx.com self.assertTrue( 1581017Szelenkov@nginx.com self.waitforfiles(self.testdir + '/node/callback'), 1591017Szelenkov@nginx.com 'write callback', 1601017Szelenkov@nginx.com ) 161855Spluknet@nginx.com 162896Szelenkov@nginx.com def test_node_application_write_before_write_head(self): 163855Spluknet@nginx.com self.load('write_before_write_head') 164855Spluknet@nginx.com 165896Szelenkov@nginx.com self.assertEqual(self.get()['status'], 200, 'write before writeHead') 166855Spluknet@nginx.com 167866Szelenkov@nginx.com def test_node_application_double_end(self): 168866Szelenkov@nginx.com self.load('double_end') 169866Szelenkov@nginx.com 170866Szelenkov@nginx.com self.assertEqual(self.get()['status'], 200, 'double end') 171866Szelenkov@nginx.com self.assertEqual(self.get()['status'], 200, 'double end 2') 172866Szelenkov@nginx.com 173855Spluknet@nginx.com def test_node_application_write_return(self): 174855Spluknet@nginx.com self.load('write_return') 175855Spluknet@nginx.com 1761017Szelenkov@nginx.com self.assertEqual( 1771017Szelenkov@nginx.com self.get()['body'], 1781295St.nateldemoura@f5.com 'bodytrue', 1791017Szelenkov@nginx.com 'write return', 1801017Szelenkov@nginx.com ) 181855Spluknet@nginx.com 182855Spluknet@nginx.com def test_node_application_remove_header(self): 183855Spluknet@nginx.com self.load('remove_header') 184855Spluknet@nginx.com 1851017Szelenkov@nginx.com resp = self.get( 1861017Szelenkov@nginx.com headers={ 1871017Szelenkov@nginx.com 'Host': 'localhost', 1881017Szelenkov@nginx.com 'X-Remove': 'X-Header', 1891017Szelenkov@nginx.com 'Connection': 'close', 1901017Szelenkov@nginx.com } 1911017Szelenkov@nginx.com ) 192855Spluknet@nginx.com self.assertEqual(resp['headers']['Was-Header'], 'true', 'was header') 193855Spluknet@nginx.com self.assertEqual(resp['headers']['Has-Header'], 'false', 'has header') 194855Spluknet@nginx.com self.assertFalse('X-Header' in resp['headers'], 'remove header') 195855Spluknet@nginx.com 196866Szelenkov@nginx.com def test_node_application_remove_header_nonexisting(self): 197866Szelenkov@nginx.com self.load('remove_header') 198866Szelenkov@nginx.com 1991017Szelenkov@nginx.com self.assertEqual( 2001017Szelenkov@nginx.com self.get( 2011017Szelenkov@nginx.com headers={ 2021017Szelenkov@nginx.com 'Host': 'localhost', 2031017Szelenkov@nginx.com 'X-Remove': 'blah', 2041017Szelenkov@nginx.com 'Connection': 'close', 2051017Szelenkov@nginx.com } 2061017Szelenkov@nginx.com )['headers']['Has-Header'], 2071017Szelenkov@nginx.com 'true', 2081017Szelenkov@nginx.com 'remove header nonexisting', 2091017Szelenkov@nginx.com ) 210866Szelenkov@nginx.com 211855Spluknet@nginx.com def test_node_application_update_header(self): 212855Spluknet@nginx.com self.load('update_header') 213855Spluknet@nginx.com 2141017Szelenkov@nginx.com self.assertEqual( 2151017Szelenkov@nginx.com self.get()['headers']['X-Header'], 'new', 'update header' 2161017Szelenkov@nginx.com ) 217855Spluknet@nginx.com 218855Spluknet@nginx.com def test_node_application_set_header_array(self): 219855Spluknet@nginx.com self.load('set_header_array') 220855Spluknet@nginx.com 2211017Szelenkov@nginx.com self.assertListEqual( 2221017Szelenkov@nginx.com self.get()['headers']['Set-Cookie'], 2231017Szelenkov@nginx.com ['tc=one,two,three', 'tc=four,five,six'], 2241017Szelenkov@nginx.com 'set header array', 2251017Szelenkov@nginx.com ) 226855Spluknet@nginx.com 2271064Szelenkov@nginx.com @unittest.skip('not yet') 228855Spluknet@nginx.com def test_node_application_status_message(self): 229855Spluknet@nginx.com self.load('status_message') 230855Spluknet@nginx.com 2311017Szelenkov@nginx.com self.assertRegex( 2321017Szelenkov@nginx.com self.get(raw_resp=True), r'200 blah', 'status message' 2331017Szelenkov@nginx.com ) 234855Spluknet@nginx.com 235855Spluknet@nginx.com def test_node_application_get_header_type(self): 236855Spluknet@nginx.com self.load('get_header_type') 237855Spluknet@nginx.com 2381017Szelenkov@nginx.com self.assertEqual( 2391017Szelenkov@nginx.com self.get()['headers']['X-Type'], 'number', 'get header type' 2401017Szelenkov@nginx.com ) 241855Spluknet@nginx.com 242866Szelenkov@nginx.com def test_node_application_header_name_case(self): 243866Szelenkov@nginx.com self.load('header_name_case') 244866Szelenkov@nginx.com 245866Szelenkov@nginx.com headers = self.get()['headers'] 246866Szelenkov@nginx.com 247866Szelenkov@nginx.com self.assertEqual(headers['X-HEADER'], '3', 'header value') 248866Szelenkov@nginx.com self.assertNotIn('X-Header', headers, 'insensitive') 249866Szelenkov@nginx.com self.assertNotIn('X-header', headers, 'insensitive 2') 250866Szelenkov@nginx.com 251866Szelenkov@nginx.com def test_node_application_promise_handler(self): 252866Szelenkov@nginx.com self.load('promise_handler') 253866Szelenkov@nginx.com 2541017Szelenkov@nginx.com self.assertEqual( 2551017Szelenkov@nginx.com self.post( 2561017Szelenkov@nginx.com headers={ 2571017Szelenkov@nginx.com 'Host': 'localhost', 2581017Szelenkov@nginx.com 'Content-Type': 'text/html', 2591017Szelenkov@nginx.com 'Connection': 'close', 2601017Szelenkov@nginx.com }, 2611017Szelenkov@nginx.com body='callback', 2621017Szelenkov@nginx.com )['status'], 2631017Szelenkov@nginx.com 200, 2641017Szelenkov@nginx.com 'promise handler request', 2651017Szelenkov@nginx.com ) 2661017Szelenkov@nginx.com self.assertTrue( 2671017Szelenkov@nginx.com self.waitforfiles(self.testdir + '/node/callback'), 2681017Szelenkov@nginx.com 'promise handler', 2691017Szelenkov@nginx.com ) 270866Szelenkov@nginx.com 271866Szelenkov@nginx.com def test_node_application_promise_handler_write_after_end(self): 272866Szelenkov@nginx.com self.load('promise_handler') 273866Szelenkov@nginx.com 2741017Szelenkov@nginx.com self.assertEqual( 2751017Szelenkov@nginx.com self.post( 2761017Szelenkov@nginx.com headers={ 2771017Szelenkov@nginx.com 'Host': 'localhost', 2781017Szelenkov@nginx.com 'Content-Type': 'text/html', 2791017Szelenkov@nginx.com 'X-Write-Call': '1', 2801017Szelenkov@nginx.com 'Connection': 'close', 2811017Szelenkov@nginx.com }, 2821017Szelenkov@nginx.com body='callback', 2831017Szelenkov@nginx.com )['status'], 2841017Szelenkov@nginx.com 200, 2851017Szelenkov@nginx.com 'promise handler request write after end', 2861017Szelenkov@nginx.com ) 287866Szelenkov@nginx.com 288866Szelenkov@nginx.com def test_node_application_promise_end(self): 289866Szelenkov@nginx.com self.load('promise_end') 290866Szelenkov@nginx.com 2911017Szelenkov@nginx.com self.assertEqual( 2921017Szelenkov@nginx.com self.post( 2931017Szelenkov@nginx.com headers={ 2941017Szelenkov@nginx.com 'Host': 'localhost', 2951017Szelenkov@nginx.com 'Content-Type': 'text/html', 2961017Szelenkov@nginx.com 'Connection': 'close', 2971017Szelenkov@nginx.com }, 2981017Szelenkov@nginx.com body='end', 2991017Szelenkov@nginx.com )['status'], 3001017Szelenkov@nginx.com 200, 3011017Szelenkov@nginx.com 'promise end request', 3021017Szelenkov@nginx.com ) 3031017Szelenkov@nginx.com self.assertTrue( 3041017Szelenkov@nginx.com self.waitforfiles(self.testdir + '/node/callback'), 'promise end' 3051017Szelenkov@nginx.com ) 306866Szelenkov@nginx.com 307866Szelenkov@nginx.com def test_node_application_promise_multiple_calls(self): 308866Szelenkov@nginx.com self.load('promise_handler') 309866Szelenkov@nginx.com 3101017Szelenkov@nginx.com self.post( 3111017Szelenkov@nginx.com headers={ 3121017Szelenkov@nginx.com 'Host': 'localhost', 3131017Szelenkov@nginx.com 'Content-Type': 'text/html', 3141017Szelenkov@nginx.com 'Connection': 'close', 3151017Szelenkov@nginx.com }, 3161017Szelenkov@nginx.com body='callback1', 3171017Szelenkov@nginx.com ) 318866Szelenkov@nginx.com 3191017Szelenkov@nginx.com self.assertTrue( 3201017Szelenkov@nginx.com self.waitforfiles(self.testdir + '/node/callback1'), 3211017Szelenkov@nginx.com 'promise first call', 3221017Szelenkov@nginx.com ) 323866Szelenkov@nginx.com 3241017Szelenkov@nginx.com self.post( 3251017Szelenkov@nginx.com headers={ 3261017Szelenkov@nginx.com 'Host': 'localhost', 3271017Szelenkov@nginx.com 'Content-Type': 'text/html', 3281017Szelenkov@nginx.com 'Connection': 'close', 3291017Szelenkov@nginx.com }, 3301017Szelenkov@nginx.com body='callback2', 3311017Szelenkov@nginx.com ) 332866Szelenkov@nginx.com 3331017Szelenkov@nginx.com self.assertTrue( 3341017Szelenkov@nginx.com self.waitforfiles(self.testdir + '/node/callback2'), 3351017Szelenkov@nginx.com 'promise second call', 3361017Szelenkov@nginx.com ) 337866Szelenkov@nginx.com 3381064Szelenkov@nginx.com @unittest.skip('not yet') 339866Szelenkov@nginx.com def test_node_application_header_name_valid(self): 340866Szelenkov@nginx.com self.load('header_name_valid') 341866Szelenkov@nginx.com 342866Szelenkov@nginx.com self.assertNotIn('status', self.get(), 'header name valid') 343866Szelenkov@nginx.com 344866Szelenkov@nginx.com def test_node_application_header_value_object(self): 345866Szelenkov@nginx.com self.load('header_value_object') 346866Szelenkov@nginx.com 347866Szelenkov@nginx.com self.assertIn('X-Header', self.get()['headers'], 'header value object') 348866Szelenkov@nginx.com 349866Szelenkov@nginx.com def test_node_application_get_header_names(self): 350866Szelenkov@nginx.com self.load('get_header_names') 351866Szelenkov@nginx.com 3521017Szelenkov@nginx.com self.assertListEqual( 3531017Szelenkov@nginx.com self.get()['headers']['X-Names'], 3541017Szelenkov@nginx.com ['date', 'x-header'], 3551017Szelenkov@nginx.com 'get header names', 3561017Szelenkov@nginx.com ) 357866Szelenkov@nginx.com 358866Szelenkov@nginx.com def test_node_application_has_header(self): 359866Szelenkov@nginx.com self.load('has_header') 360866Szelenkov@nginx.com 3611017Szelenkov@nginx.com self.assertEqual( 3621017Szelenkov@nginx.com self.get( 3631017Szelenkov@nginx.com headers={ 3641017Szelenkov@nginx.com 'Host': 'localhost', 3651017Szelenkov@nginx.com 'X-Header': 'length', 3661017Szelenkov@nginx.com 'Connection': 'close', 3671017Szelenkov@nginx.com } 3681017Szelenkov@nginx.com )['headers']['X-Has-Header'], 3691017Szelenkov@nginx.com 'false', 3701017Szelenkov@nginx.com 'has header length', 3711017Szelenkov@nginx.com ) 372866Szelenkov@nginx.com 3731017Szelenkov@nginx.com self.assertEqual( 3741017Szelenkov@nginx.com self.get( 3751017Szelenkov@nginx.com headers={ 3761017Szelenkov@nginx.com 'Host': 'localhost', 3771017Szelenkov@nginx.com 'X-Header': 'Date', 3781017Szelenkov@nginx.com 'Connection': 'close', 3791017Szelenkov@nginx.com } 3801017Szelenkov@nginx.com )['headers']['X-Has-Header'], 3811017Szelenkov@nginx.com 'false', 3821017Szelenkov@nginx.com 'has header date', 3831017Szelenkov@nginx.com ) 384866Szelenkov@nginx.com 385866Szelenkov@nginx.com def test_node_application_write_multiple(self): 386866Szelenkov@nginx.com self.load('write_multiple') 387866Szelenkov@nginx.com 3881017Szelenkov@nginx.com self.assertEqual( 3891017Szelenkov@nginx.com self.get()['body'], 'writewrite2end', 'write multiple' 3901017Szelenkov@nginx.com ) 3911017Szelenkov@nginx.com 392866Szelenkov@nginx.com 393855Spluknet@nginx.comif __name__ == '__main__': 3941019Szelenkov@nginx.com TestNodeApplication.main() 395