2
2
import logging
3
3
import time
4
4
import collections
5
- import gzip
6
- import pytest
7
5
8
6
from stp_core .common .logging .TimeAndSizeRotatingFileHandler \
9
7
import TimeAndSizeRotatingFileHandler
10
8
from stp_core .common .log import Logger
11
9
12
10
13
- @pytest .fixture (params = [False , True ], ids = ["plain" , "compressed" ])
14
- def log_compression (request ):
15
- return request .param
16
-
17
- def test_default_log_rotation_config_is_correct (tdir_for_func ):
18
- logDirPath = tdir_for_func
19
- logFile = os .path .join (logDirPath , "log" )
20
- logger = Logger ()
21
-
22
- # Assert this doesn't fail
23
- logger .enableFileLogging (logFile )
24
-
25
-
26
- def test_time_log_rotation (tdir_for_func , log_compression ):
11
+ def test_time_log_rotation (tdir_for_func ):
27
12
logDirPath = tdir_for_func
28
13
logFile = os .path .join (logDirPath , "log" )
29
14
logger = logging .getLogger ('test_time_log_rotation-logger' )
30
15
31
16
logger .setLevel (logging .DEBUG )
32
- handler = TimeAndSizeRotatingFileHandler (
33
- logFile , interval = 1 , when = 's' , compress = log_compression )
17
+ handler = TimeAndSizeRotatingFileHandler (logFile , interval = 1 , when = 's' )
34
18
logger .addHandler (handler )
35
19
for i in range (3 ):
36
20
time .sleep (1 )
37
21
logger .debug ("line" )
38
22
assert len (os .listdir (logDirPath )) == 4 # initial + 3 new
39
23
40
24
41
- def test_size_log_rotation (tdir_for_func , log_compression ):
25
+ def test_size_log_rotation (tdir_for_func ):
42
26
logDirPath = tdir_for_func
43
27
logFile = os .path .join (logDirPath , "log" )
44
28
logger = logging .getLogger ('test_time_log_rotation-logger' )
45
29
46
30
logger .setLevel (logging .DEBUG )
47
31
handler = TimeAndSizeRotatingFileHandler (
48
- logFile , maxBytes = (4 + len (os .linesep )) * 4 + 1 , compress = log_compression )
32
+ logFile , maxBytes = (4 + len (os .linesep )) * 4 + 1 )
49
33
logger .addHandler (handler )
50
34
for i in range (20 ):
51
35
logger .debug ("line" )
@@ -54,14 +38,14 @@ def test_size_log_rotation(tdir_for_func, log_compression):
54
38
assert len (os .listdir (logDirPath )) == 5
55
39
56
40
57
- def test_time_and_size_log_rotation (tdir_for_func , log_compression ):
41
+ def test_time_and_size_log_rotation (tdir_for_func ):
58
42
logDirPath = tdir_for_func
59
43
logFile = os .path .join (logDirPath , "log" )
60
44
logger = logging .getLogger ('test_time_and_size_log_rotation-logger' )
61
45
62
46
logger .setLevel (logging .DEBUG )
63
47
handler = TimeAndSizeRotatingFileHandler (
64
- logFile , maxBytes = (4 + len (os .linesep )) * 4 + 1 , interval = 1 , when = "s" , compress = log_compression )
48
+ logFile , maxBytes = (4 + len (os .linesep )) * 4 + 1 , interval = 1 , when = "s" )
65
49
logger .addHandler (handler )
66
50
67
51
for i in range (20 ):
@@ -74,7 +58,7 @@ def test_time_and_size_log_rotation(tdir_for_func, log_compression):
74
58
assert len (os .listdir (logDirPath )) == 8
75
59
76
60
77
- def test_time_and_size_log_rotation1 (tdir_for_func , log_compression ):
61
+ def test_time_and_size_log_rotation1 (tdir_for_func ):
78
62
log_dir_path = tdir_for_func
79
63
logFile = os .path .join (log_dir_path , "log" )
80
64
logger = logging .getLogger ('test_time_and_size_log_rotation-logger1' )
@@ -90,7 +74,7 @@ def test_time_and_size_log_rotation1(tdir_for_func, log_compression):
90
74
handler = TimeAndSizeRotatingFileHandler (
91
75
logFile ,
92
76
maxBytes = (record_length + len (os .linesep )) * record_per_file + 1 ,
93
- interval = 1 , when = "h" , backupCount = backup_count , utc = True , compress = log_compression )
77
+ interval = 1 , when = "h" , backupCount = backup_count , utc = True )
94
78
logger .addHandler (handler )
95
79
96
80
for i in range (1 , record_count + 1 ):
@@ -107,7 +91,6 @@ def test_time_and_size_log_rotation1(tdir_for_func, log_compression):
107
91
assert len (cir_buffer ) == len (circ_buffer_set )
108
92
assert len (os .listdir (log_dir_path )) == (backup_count + 1 )
109
93
for file_name in os .listdir (log_dir_path ):
110
- open_fn = gzip .open if file_name .endswith (".gz" ) else open
111
- with open_fn (os .path .join (log_dir_path , file_name ), "rt" ) as file :
94
+ with open (os .path .join (log_dir_path , file_name )) as file :
112
95
for line in file .readlines ():
113
96
assert line .strip () in circ_buffer_set
0 commit comments