File tree Expand file tree Collapse file tree 2 files changed +33
-0
lines changed Expand file tree Collapse file tree 2 files changed +33
-0
lines changed Original file line number Diff line number Diff line change 1+ import logging
2+
3+
4+ class BaseLogger :
5+
6+ def __init__ (self , log_level = logging .CRITICAL ):
7+ self ._logger = logging .getLogger ("google" )
8+ self ._logger .setLevel (log_level )
9+ console_handler = logging .StreamHandler ()
10+ console_handler .setLevel (log_level )
11+ formatter = logging .Formatter ("%(asctime)s - %(name)s - %(levelname)s - %(message)s" )
12+ console_handler .setFormatter (formatter )
13+ self ._logger .addHandler (console_handler )
14+
15+ def get_logger (self ):
16+ return self ._logger
Original file line number Diff line number Diff line change 1+ import logging
2+ import pytest
3+
4+ from google .api_core .client_logging import BaseLogger
5+
6+
7+ def test_base_logger (caplog ):
8+
9+ logger = BaseLogger ().get_logger ()
10+
11+ with caplog .at_level (logging .INFO , logger = "google" ):
12+ logger .info ("This is a test message." )
13+
14+ assert "This is a test message." in caplog .text
15+ assert caplog .records [0 ].name == "google"
16+ assert caplog .records [0 ].levelname == "INFO"
17+ assert caplog .records [0 ].message == "This is a test message."
You can’t perform that action at this time.
0 commit comments