11248Szelenkov@nginx.comimport os 21248Szelenkov@nginx.comfrom subprocess import call 31477Szelenkov@nginx.com 41654Szelenkov@nginx.comfrom conftest import unit_stop 51248Szelenkov@nginx.comfrom unit.applications.lang.python import TestApplicationPython 6*1735Szelenkov@nginx.comfrom unit.utils import waitforfiles 71248Szelenkov@nginx.com 81248Szelenkov@nginx.com 91248Szelenkov@nginx.comclass TestUSR1(TestApplicationPython): 101467Szelenkov@nginx.com prerequisites = {'modules': {'python': 'any'}} 111248Szelenkov@nginx.com 121654Szelenkov@nginx.com def test_usr1_access_log(self, temp_dir): 131248Szelenkov@nginx.com self.load('empty') 141248Szelenkov@nginx.com 151397Szelenkov@nginx.com log = 'access.log' 161397Szelenkov@nginx.com log_new = 'new.log' 171654Szelenkov@nginx.com log_path = temp_dir + '/' + log 181248Szelenkov@nginx.com 191596Szelenkov@nginx.com assert 'success' in self.conf( 201596Szelenkov@nginx.com '"' + log_path + '"', 'access_log' 211596Szelenkov@nginx.com ), 'access log configure' 221248Szelenkov@nginx.com 231596Szelenkov@nginx.com assert waitforfiles(log_path), 'open' 241596Szelenkov@nginx.com 251654Szelenkov@nginx.com os.rename(log_path, temp_dir + '/' + log_new) 261248Szelenkov@nginx.com 271596Szelenkov@nginx.com assert self.get()['status'] == 200 281248Szelenkov@nginx.com 291596Szelenkov@nginx.com assert ( 301596Szelenkov@nginx.com self.wait_for_record(r'"GET / HTTP/1.1" 200 0 "-" "-"', log_new) 311596Szelenkov@nginx.com is not None 321596Szelenkov@nginx.com ), 'rename new' 331596Szelenkov@nginx.com assert not os.path.isfile(log_path), 'rename old' 341248Szelenkov@nginx.com 351654Szelenkov@nginx.com with open(temp_dir + '/unit.pid', 'r') as f: 361248Szelenkov@nginx.com pid = f.read().rstrip() 371248Szelenkov@nginx.com 381248Szelenkov@nginx.com call(['kill', '-s', 'USR1', pid]) 391248Szelenkov@nginx.com 401596Szelenkov@nginx.com assert waitforfiles(log_path), 'reopen' 411248Szelenkov@nginx.com 421596Szelenkov@nginx.com assert self.get(url='/usr1')['status'] == 200 431397Szelenkov@nginx.com 441654Szelenkov@nginx.com unit_stop() 451248Szelenkov@nginx.com 461596Szelenkov@nginx.com assert ( 471596Szelenkov@nginx.com self.wait_for_record(r'"GET /usr1 HTTP/1.1" 200 0 "-" "-"', log) 481596Szelenkov@nginx.com is not None 491596Szelenkov@nginx.com ), 'reopen 2' 501596Szelenkov@nginx.com assert self.search_in_log(r'/usr1', log_new) is None, 'rename new 2' 511248Szelenkov@nginx.com 521654Szelenkov@nginx.com def test_usr1_unit_log(self, temp_dir): 531248Szelenkov@nginx.com self.load('log_body') 541248Szelenkov@nginx.com 551397Szelenkov@nginx.com log_new = 'new.log' 561654Szelenkov@nginx.com log_path = temp_dir + '/unit.log' 571654Szelenkov@nginx.com log_path_new = temp_dir + '/' + log_new 581248Szelenkov@nginx.com 591248Szelenkov@nginx.com os.rename(log_path, log_path_new) 601248Szelenkov@nginx.com 611248Szelenkov@nginx.com body = 'body_for_a_log_new' 621596Szelenkov@nginx.com assert self.post(body=body)['status'] == 200 631248Szelenkov@nginx.com 641596Szelenkov@nginx.com assert self.wait_for_record(body, log_new) is not None, 'rename new' 651596Szelenkov@nginx.com assert not os.path.isfile(log_path), 'rename old' 661248Szelenkov@nginx.com 671654Szelenkov@nginx.com with open(temp_dir + '/unit.pid', 'r') as f: 681248Szelenkov@nginx.com pid = f.read().rstrip() 691248Szelenkov@nginx.com 701248Szelenkov@nginx.com call(['kill', '-s', 'USR1', pid]) 711248Szelenkov@nginx.com 721596Szelenkov@nginx.com assert waitforfiles(log_path), 'reopen' 731248Szelenkov@nginx.com 741248Szelenkov@nginx.com body = 'body_for_a_log_unit' 751596Szelenkov@nginx.com assert self.post(body=body)['status'] == 200 761397Szelenkov@nginx.com 771654Szelenkov@nginx.com unit_stop() 781248Szelenkov@nginx.com 791596Szelenkov@nginx.com assert self.wait_for_record(body) is not None, 'rename new' 801596Szelenkov@nginx.com assert self.search_in_log(body, log_new) is None, 'rename new 2' 811248Szelenkov@nginx.com 821248Szelenkov@nginx.com # merge two log files into unit.log to check alerts 831248Szelenkov@nginx.com 841596Szelenkov@nginx.com with open(log_path, 'w') as unit_log, open( 851596Szelenkov@nginx.com log_path_new, 'r' 861596Szelenkov@nginx.com ) as unit_log_new: 871397Szelenkov@nginx.com unit_log.write(unit_log_new.read()) 88