@@ -77,23 +77,21 @@ def test_get_logger_for_server_when_given_cef_format_uses_cef_formatter():
77
77
logger = get_logger_for_server (
78
78
"example.com" , ServerProtocol .TCP , SendToFileEventsOutputFormat .CEF , None
79
79
)
80
- assert type (logger .handlers [0 ].formatter ) == FileEventDictToCEFFormatter
80
+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToCEFFormatter )
81
81
82
82
83
83
def test_get_logger_for_server_when_given_json_format_uses_json_formatter ():
84
84
logger = get_logger_for_server (
85
85
"example.com" , ServerProtocol .TCP , OutputFormat .JSON , None
86
86
)
87
- actual = type (logger .handlers [0 ].formatter )
88
- assert actual == FileEventDictToJSONFormatter
87
+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToJSONFormatter )
89
88
90
89
91
90
def test_get_logger_for_server_when_given_raw_json_format_uses_raw_json_formatter ():
92
91
logger = get_logger_for_server (
93
92
"example.com" , ServerProtocol .TCP , OutputFormat .RAW , None
94
93
)
95
- actual = type (logger .handlers [0 ].formatter )
96
- assert actual == FileEventDictToRawJSONFormatter
94
+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToRawJSONFormatter )
97
95
98
96
99
97
def test_get_logger_for_server_when_called_twice_only_has_one_handler ():
@@ -108,7 +106,7 @@ def test_get_logger_for_server_uses_no_priority_syslog_handler():
108
106
logger = get_logger_for_server (
109
107
"example.com" , ServerProtocol .TCP , SendToFileEventsOutputFormat .CEF , None
110
108
)
111
- assert type (logger .handlers [0 ]) == NoPrioritySysLogHandler
109
+ assert isinstance (logger .handlers [0 ], NoPrioritySysLogHandler )
112
110
113
111
114
112
def test_get_logger_for_server_constructs_handler_with_expected_args (
0 commit comments