1import os
2import unittest
3from subprocess import call
4from unit.applications.lang.python import TestApplicationPython
5
6
7class TestUSR1(TestApplicationPython):
8 prerequisites = {'modules': ['python']}
9
10 def test_usr1_access_log(self):
11 self.load('empty')
12
13 log_path = self.testdir + '/access.log'
13 log = 'access.log'
14 log_new = 'new.log'
15 log_path = self.testdir + '/' + log
16
17 self.assertIn(
18 'success',
19 self.conf('"' + log_path + '"', 'access_log'),
20 'access log configure',
21 )
22
23 self.assertTrue(self.waitforfiles(log_path), 'open')
24
23 log_path_new = self.testdir + '/new.log'
25 os.rename(log_path, self.testdir + '/' + log_new)
26
25 os.rename(log_path, log_path_new)
27 self.assertEqual(self.get()['status'], 200)
28
27 self.get()
28
29 self.assertIsNotNone(
30 self.wait_for_record(r'"GET / HTTP/1.1" 200 0 "-" "-"', 'new.log'),
30 self.wait_for_record(r'"GET / HTTP/1.1" 200 0 "-" "-"', log_new),
31 'rename new',
32 )
33 self.assertFalse(os.path.isfile(log_path), 'rename old')
34
35 with open(self.testdir + '/unit.pid', 'r') as f:
36 pid = f.read().rstrip()
37
38 call(['kill', '-s', 'USR1', pid])
39
40 self.assertTrue(self.waitforfiles(log_path), 'reopen')
41
42 self.get(url='/usr1')
42 self.assertEqual(self.get(url='/usr1')['status'], 200)
43
44 self.stop()
45
46 self.assertIsNotNone(
45 self.wait_for_record(
46 r'"GET /usr1 HTTP/1.1" 200 0 "-" "-"', 'access.log'
47 ),
47 self.wait_for_record(r'"GET /usr1 HTTP/1.1" 200 0 "-" "-"', log),
48 'reopen 2',
49 )
50 self.assertIsNone(
51 self.search_in_log(r'/usr1', 'new.log'), 'rename new 2'
51 self.search_in_log(r'/usr1', log_new), 'rename new 2'
52 )
53
54 @unittest.skip('not yet')
55 def test_usr1_unit_log(self):
56 self.load('log_body')
57
58 log_path = self.testdir + '/unit.log'
59 log_path_new = self.testdir + '/new.log'
58 log_new = 'new.log'
59 log_path = self.testdir + '/' + 'unit.log'
60 log_path_new = self.testdir + '/' + log_new
61
62 os.rename(log_path, log_path_new)
63
64 body = 'body_for_a_log_new'
64 self.post(body=body)
65 self.assertEqual(self.post(body=body)['status'], 200)
66
67 self.assertIsNotNone(
67 self.wait_for_record(body, 'new.log'), 'rename new'
68 self.wait_for_record(body, log_new), 'rename new'
69 )
70 self.assertFalse(os.path.isfile(log_path), 'rename old')
71
72 with open(self.testdir + '/unit.pid', 'r') as f:
73 pid = f.read().rstrip()
74
75 call(['kill', '-s', 'USR1', pid])
76
77 self.assertTrue(self.waitforfiles(log_path), 'reopen')
78
79 body = 'body_for_a_log_unit'
79 self.post(body=body)
80 self.assertEqual(self.post(body=body)['status'], 200)
81
82 self.stop()
83
84 self.assertIsNotNone(self.wait_for_record(body), 'rename new')
82 self.assertIsNone(self.search_in_log(body, 'new.log'), 'rename new 2')
85 self.assertIsNone(self.search_in_log(body, log_new), 'rename new 2')
86
87 # merge two log files into unit.log to check alerts
88
89 with open(log_path, 'w') as unit_log, \
87 open(log_path_new, 'r') as new_log:
88 unit_log.write(new_log.read())
90 open(log_path_new, 'r') as unit_log_new:
91 unit_log.write(unit_log_new.read())
92
93
94if __name__ == '__main__':
95 TestUSR1.main()