"""Unit tests for the config of datadog_caching_monitor.""" from swf_monitoring.flows.datadog_caching_monitor import config def test_get_snowflake_users_to_monitor(monkeypatch): """Test get_snowflake_users_to_monitor.""" assert config.get_snowflake_users_to_monitor() == [ 'PROD_OWS_ANALYTICS', 'PROD_BFF_MOBILE', 'PROD_VAPI'] def test_get_snowflake_warehouses_to_monitor(monkeypatch): """Test get_snowflake_warehouses_to_monitor.""" monkeypatch.setenv('SNOWFLAKE_WAREHOUSES_TO_MONITOR', 'WH1,WH2') assert config.get_snowflake_warehouses_to_monitor() == ['WH1', 'WH2']