1*693Szelenkov@nginx.comimport unittest 2*693Szelenkov@nginx.comimport unit 3*693Szelenkov@nginx.com 4*693Szelenkov@nginx.comclass TestUnitGoApplication(unit.TestUnitApplicationGo): 5*693Szelenkov@nginx.com 6*693Szelenkov@nginx.com def setUpClass(): 7*693Szelenkov@nginx.com u = unit.TestUnit() 8*693Szelenkov@nginx.com 9*693Szelenkov@nginx.com if u.architecture == '32bit': 10*693Szelenkov@nginx.com raise unittest.SkipTest('Skip Go tests for x86') 11*693Szelenkov@nginx.com 12*693Szelenkov@nginx.com u.check_modules('go') 13*693Szelenkov@nginx.com 14*693Szelenkov@nginx.com def test_go_application_variables(self): 15*693Szelenkov@nginx.com self.load('variables') 16*693Szelenkov@nginx.com 17*693Szelenkov@nginx.com body = 'Test body string.' 18*693Szelenkov@nginx.com 19*693Szelenkov@nginx.com resp = self.post(headers={ 20*693Szelenkov@nginx.com 'Host': 'localhost', 21*693Szelenkov@nginx.com 'Content-Type': 'text/html', 22*693Szelenkov@nginx.com 'Custom-Header': 'blah' 23*693Szelenkov@nginx.com }, body=body) 24*693Szelenkov@nginx.com 25*693Szelenkov@nginx.com self.assertEqual(resp['status'], 200, 'status') 26*693Szelenkov@nginx.com headers = resp['headers'] 27*693Szelenkov@nginx.com header_server = headers.pop('Server') 28*693Szelenkov@nginx.com self.assertRegex(header_server, r'Unit/[\d\.]+', 'server header') 29*693Szelenkov@nginx.com 30*693Szelenkov@nginx.com date = headers.pop('Date') 31*693Szelenkov@nginx.com self.assertEqual(date[-4:], ' GMT', 'date header timezone') 32*693Szelenkov@nginx.com self.assertLess(abs(self.date_to_sec_epoch(date) - self.sec_epoch()), 5, 33*693Szelenkov@nginx.com 'date header') 34*693Szelenkov@nginx.com 35*693Szelenkov@nginx.com self.assertDictEqual(headers, { 36*693Szelenkov@nginx.com 'Content-Length': str(len(body)), 37*693Szelenkov@nginx.com 'Content-Type': 'text/html', 38*693Szelenkov@nginx.com 'Request-Method': 'POST', 39*693Szelenkov@nginx.com 'Request-Uri': '/', 40*693Szelenkov@nginx.com 'Http-Host': 'localhost', 41*693Szelenkov@nginx.com 'Server-Protocol': 'HTTP/1.1', 42*693Szelenkov@nginx.com 'Server-Protocol-Major': '1', 43*693Szelenkov@nginx.com 'Server-Protocol-Minor': '1', 44*693Szelenkov@nginx.com 'Custom-Header': 'blah' 45*693Szelenkov@nginx.com }, 'headers') 46*693Szelenkov@nginx.com self.assertEqual(resp['body'], body, 'body') 47*693Szelenkov@nginx.com 48*693Szelenkov@nginx.com def test_go_application_get_variables(self): 49*693Szelenkov@nginx.com self.load('get_variables') 50*693Szelenkov@nginx.com 51*693Szelenkov@nginx.com resp = self.get(url='/?var1=val1&var2=&var3') 52*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-1'], 'val1', 'GET variables') 53*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-2'], '', 'GET variables 2') 54*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-3'], '', 'GET variables 3') 55*693Szelenkov@nginx.com 56*693Szelenkov@nginx.com def test_go_application_post_variables(self): 57*693Szelenkov@nginx.com self.load('post_variables') 58*693Szelenkov@nginx.com 59*693Szelenkov@nginx.com resp = self.post(headers={ 60*693Szelenkov@nginx.com 'Content-Type': 'application/x-www-form-urlencoded', 61*693Szelenkov@nginx.com 'Host': 'localhost', 62*693Szelenkov@nginx.com 'Connection': 'close' 63*693Szelenkov@nginx.com }, body='var1=val1&var2=&var3') 64*693Szelenkov@nginx.com 65*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-1'], 'val1', 'POST variables') 66*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-2'], '', 'POST variables 2') 67*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Var-3'], '', 'POST variables 3') 68*693Szelenkov@nginx.com 69*693Szelenkov@nginx.com def test_go_application_404(self): 70*693Szelenkov@nginx.com self.load('404') 71*693Szelenkov@nginx.com 72*693Szelenkov@nginx.com resp = self.get() 73*693Szelenkov@nginx.com 74*693Szelenkov@nginx.com self.assertEqual(resp['status'], 404, '404 status') 75*693Szelenkov@nginx.com self.assertRegex(resp['body'], r'<title>404 Not Found</title>', 76*693Szelenkov@nginx.com '404 body') 77*693Szelenkov@nginx.com 78*693Szelenkov@nginx.com def test_go_keepalive_body(self): 79*693Szelenkov@nginx.com self.load('mirror') 80*693Szelenkov@nginx.com 81*693Szelenkov@nginx.com (resp, sock) = self.post(headers={ 82*693Szelenkov@nginx.com 'Connection': 'keep-alive', 83*693Szelenkov@nginx.com 'Content-Type': 'text/html', 84*693Szelenkov@nginx.com 'Host': 'localhost' 85*693Szelenkov@nginx.com }, start=True, body='0123456789' * 500) 86*693Szelenkov@nginx.com 87*693Szelenkov@nginx.com self.assertEqual(resp['body'], '0123456789' * 500, 'keep-alive 1') 88*693Szelenkov@nginx.com 89*693Szelenkov@nginx.com resp = self.post(headers={ 90*693Szelenkov@nginx.com 'Connection': 'close', 91*693Szelenkov@nginx.com 'Content-Type': 'text/html', 92*693Szelenkov@nginx.com 'Host': 'localhost' 93*693Szelenkov@nginx.com }, sock=sock, body='0123456789') 94*693Szelenkov@nginx.com 95*693Szelenkov@nginx.com self.assertEqual(resp['body'], '0123456789', 'keep-alive 2') 96*693Szelenkov@nginx.com 97*693Szelenkov@nginx.com def test_go_application_cookies(self): 98*693Szelenkov@nginx.com self.load('cookies') 99*693Szelenkov@nginx.com 100*693Szelenkov@nginx.com resp = self.get(headers={ 101*693Szelenkov@nginx.com 'Cookie': 'var1=val1; var2=val2', 102*693Szelenkov@nginx.com 'Host': 'localhost', 103*693Szelenkov@nginx.com 'Connection': 'close' 104*693Szelenkov@nginx.com }) 105*693Szelenkov@nginx.com 106*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Cookie-1'], 'val1', 'cookie 1') 107*693Szelenkov@nginx.com self.assertEqual(resp['headers']['X-Cookie-2'], 'val2', 'cookie 2') 108*693Szelenkov@nginx.com 109*693Szelenkov@nginx.com def test_go_application_command_line_arguments_type(self): 110*693Szelenkov@nginx.com self.load('command_line_arguments') 111*693Szelenkov@nginx.com 112*693Szelenkov@nginx.com self.assertIn('error', self.conf(''"a b c", 113*693Szelenkov@nginx.com '/applications/command_line_arguments/arguments'), 'arguments type') 114*693Szelenkov@nginx.com 115*693Szelenkov@nginx.com def test_go_application_command_line_arguments_0(self): 116*693Szelenkov@nginx.com self.load('command_line_arguments') 117*693Szelenkov@nginx.com 118*693Szelenkov@nginx.com self.assertEqual(self.get()['headers']['X-Arg-0'], 119*693Szelenkov@nginx.com self.conf_get('/applications/command_line_arguments/executable'), 120*693Szelenkov@nginx.com 'argument 0') 121*693Szelenkov@nginx.com 122*693Szelenkov@nginx.com def test_go_application_command_line_arguments(self): 123*693Szelenkov@nginx.com self.load('command_line_arguments') 124*693Szelenkov@nginx.com 125*693Szelenkov@nginx.com arg1 = '--cc=gcc-7.2.0' 126*693Szelenkov@nginx.com arg2 = '--cc-opt=\'-O0 -DNXT_DEBUG_MEMORY=1 -fsanitize=address\'' 127*693Szelenkov@nginx.com arg3 = '--debug' 128*693Szelenkov@nginx.com 129*693Szelenkov@nginx.com self.conf('["' + arg1 + '", "' + arg2 + '", "' + arg3 + '"]', 130*693Szelenkov@nginx.com '/applications/command_line_arguments/arguments') 131*693Szelenkov@nginx.com 132*693Szelenkov@nginx.com self.assertEqual(self.get()['body'], arg1 + ',' + arg2 + ',' + arg3, 133*693Szelenkov@nginx.com 'arguments') 134*693Szelenkov@nginx.com 135*693Szelenkov@nginx.com def test_go_application_command_line_arguments_change(self): 136*693Szelenkov@nginx.com self.load('command_line_arguments') 137*693Szelenkov@nginx.com 138*693Szelenkov@nginx.com args_path = '/applications/command_line_arguments/arguments' 139*693Szelenkov@nginx.com 140*693Szelenkov@nginx.com self.conf('["0", "a", "$", ""]', args_path) 141*693Szelenkov@nginx.com 142*693Szelenkov@nginx.com self.assertEqual(self.get()['body'], '0,a,$,', 'arguments') 143*693Szelenkov@nginx.com 144*693Szelenkov@nginx.com self.conf('["-1", "b", "%"]', args_path) 145*693Szelenkov@nginx.com 146*693Szelenkov@nginx.com self.assertEqual(self.get()['body'], '-1,b,%', 'arguments change') 147*693Szelenkov@nginx.com 148*693Szelenkov@nginx.com self.conf('[]', args_path) 149*693Szelenkov@nginx.com 150*693Szelenkov@nginx.com self.assertEqual(self.get()['headers']['Content-Length'], '0', 151*693Szelenkov@nginx.com 'arguments empty') 152*693Szelenkov@nginx.com 153*693Szelenkov@nginx.comif __name__ == '__main__': 154*693Szelenkov@nginx.com unittest.main() 155