Skip to content

Commit

Permalink
address PR feedback
Browse files Browse the repository at this point in the history
  • Loading branch information
ohmayr committed Nov 26, 2024
1 parent 0f1a4bd commit b5088af
Showing 1 changed file with 18 additions and 8 deletions.
26 changes: 18 additions & 8 deletions tests/unit/test_client_logging.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@

from google.api_core.client_logging import (
setup_logging,
parse_logging_scopes,
initialize_logging,
)

Expand All @@ -24,7 +23,7 @@ def test_setup_logging_w_no_scopes():
assert base_logger.propagate == False
assert base_logger.level == logging.NOTSET

reset_logger("google")
reset_logger("foo")


def test_setup_logging_w_base_scope():
Expand All @@ -35,7 +34,7 @@ def test_setup_logging_w_base_scope():
assert base_logger.propagate == False
assert base_logger.level == logging.DEBUG

reset_logger("google")
reset_logger("foo")


def test_setup_logging_w_module_scope():
Expand All @@ -52,8 +51,8 @@ def test_setup_logging_w_module_scope():
assert module_logger.propagate == False
assert module_logger.level == logging.DEBUG

reset_logger("google")
reset_logger("google.foo")
reset_logger("foo")
reset_logger("foo.bar")


def test_setup_logging_w_incorrect_scope():
Expand All @@ -71,8 +70,8 @@ def test_setup_logging_w_incorrect_scope():
assert logger.propagate == False
assert logger.level == logging.DEBUG

reset_logger("google")
reset_logger("foo")
reset_logger("abc")


def test_initialize_logging():
Expand All @@ -91,5 +90,16 @@ def test_initialize_logging():
assert module_logger.propagate == False
assert module_logger.level == logging.DEBUG

reset_logger("google")
reset_logger("google.foo")
base_logger.propagate = True
module_logger.propagate = True

with mock.patch("os.getenv", return_value="foo.bar"):
with mock.patch("google.api_core.client_logging._BASE_LOGGER_NAME", "foo"):
initialize_logging()

assert base_logger.propagate == True
assert module_logger.propagate == True


reset_logger("foo")
reset_logger("foo.bar")

0 comments on commit b5088af

Please sign in to comment.