xref: /unit/test/test_usr1.py (revision 1397:5fd52d94ff40)
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 = '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
25        os.rename(log_path, self.testdir + '/' + log_new)
26
27        self.assertEqual(self.get()['status'], 200)
28
29        self.assertIsNotNone(
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.assertEqual(self.get(url='/usr1')['status'], 200)
43
44        self.stop()
45
46        self.assertIsNotNone(
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', 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_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'
65        self.assertEqual(self.post(body=body)['status'], 200)
66
67        self.assertIsNotNone(
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'
80        self.assertEqual(self.post(body=body)['status'], 200)
81
82        self.stop()
83
84        self.assertIsNotNone(self.wait_for_record(body), 'rename new')
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, \
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()
96