@@ -77,23 +77,21 @@ def test_get_logger_for_server_when_given_cef_format_uses_cef_formatter():
7777 logger = get_logger_for_server (
7878 "example.com" , ServerProtocol .TCP , SendToFileEventsOutputFormat .CEF , None
7979 )
80- assert type (logger .handlers [0 ].formatter ) == FileEventDictToCEFFormatter
80+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToCEFFormatter )
8181
8282
8383def test_get_logger_for_server_when_given_json_format_uses_json_formatter ():
8484 logger = get_logger_for_server (
8585 "example.com" , ServerProtocol .TCP , OutputFormat .JSON , None
8686 )
87- actual = type (logger .handlers [0 ].formatter )
88- assert actual == FileEventDictToJSONFormatter
87+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToJSONFormatter )
8988
9089
9190def test_get_logger_for_server_when_given_raw_json_format_uses_raw_json_formatter ():
9291 logger = get_logger_for_server (
9392 "example.com" , ServerProtocol .TCP , OutputFormat .RAW , None
9493 )
95- actual = type (logger .handlers [0 ].formatter )
96- assert actual == FileEventDictToRawJSONFormatter
94+ assert isinstance (logger .handlers [0 ].formatter , FileEventDictToRawJSONFormatter )
9795
9896
9997def 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():
108106 logger = get_logger_for_server (
109107 "example.com" , ServerProtocol .TCP , SendToFileEventsOutputFormat .CEF , None
110108 )
111- assert type (logger .handlers [0 ]) == NoPrioritySysLogHandler
109+ assert isinstance (logger .handlers [0 ], NoPrioritySysLogHandler )
112110
113111
114112def test_get_logger_for_server_constructs_handler_with_expected_args (
0 commit comments