def test_prefect_logging_level_override_logic(config, agent_env_vars, run_config_env_vars, expected_logging_level, tmpdir): with set_temporary_config(config): agent = LocalAgent(env_vars=agent_env_vars) run = LocalRun(working_dir=str(tmpdir), env=run_config_env_vars) env_vars = agent.populate_env_vars( GraphQLResult({ "id": "id", "name": "name", "flow": { "id": "foo" }, "run_config": run.serialize(), }), run, ) assert env_vars["PREFECT__LOGGING__LEVEL"] == expected_logging_level
def test_populate_env_vars_from_run_config(tmpdir): agent = LocalAgent(env_vars={"KEY1": "VAL1", "KEY2": "VAL2"}) working_dir = str(tmpdir) run = LocalRun( env={"KEY2": "OVERRIDE", "PREFECT__LOGGING__LEVEL": "TEST"}, working_dir=working_dir, ) env_vars = agent.populate_env_vars( GraphQLResult( { "id": "id", "name": "name", "flow": {"id": "foo"}, "run_config": run.serialize(), } ), run, ) assert env_vars["KEY1"] == "VAL1" assert env_vars["KEY2"] == "OVERRIDE" assert env_vars["PREFECT__LOGGING__LEVEL"] == "TEST" assert working_dir in env_vars["PYTHONPATH"]