Skip to content

Commit

Permalink
Code coverage
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrew Greenburg committed Nov 6, 2023
1 parent 416f4fd commit a919598
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tests/test_log.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import logging

from cosmos import get_provider_info
from cosmos.log import get_logger
from airflow.configuration import conf

Expand All @@ -21,3 +22,11 @@ def test_propagate_logs_conf():
conf.set("cosmos", "propagate_logs", "False")
custom_logger = get_logger("cosmos-log")
assert custom_logger.propagate is False


def test_get_provider_info():
provider_info = get_provider_info()
assert "cosmos" in provider_info.get("config").keys()
assert "options" in provider_info.get("config").get("cosmos").keys()
assert "propagate_logs" in provider_info.get("config").get("cosmos").get("options").keys()
assert provider_info["config"]["cosmos"]["options"]["propagate_logs"]["type"] == "boolean"

0 comments on commit a919598

Please sign in to comment.