def test_nomad_agent_replace_yaml(runner_token):
with set_temporary_config({"cloud.agent.auth_token": "token"}):
flow_run = GraphQLResult({
"flow":
GraphQLResult({
"storage":
Docker(registry_url="test", image_name="name",
image_tag="tag").serialize(),
"id":
"id",
}),
"id":
"id",
})
agent = NomadAgent()
job = agent.replace_job_spec_json(flow_run)
assert job["Job"]["TaskGroups"][0]["Tasks"][0]["Name"] == "id"
assert (job["Job"]["TaskGroups"][0]["Tasks"][0]["Config"]["image"] ==
"test/name:tag")
env = job["Job"]["TaskGroups"][0]["Tasks"][0]["Env"]
assert env["PREFECT__CLOUD__API"] == "https://api.prefect.io"
assert env["PREFECT__CLOUD__AGENT__AUTH_TOKEN"] == "token"
assert env["PREFECT__CONTEXT__FLOW_RUN_ID"] == "id"
assert env["PREFECT__CONTEXT__NAMESPACE"] == "default"
def test_nomad_agent_replace_yaml(runner_token, flag, cloud_api):
with set_temporary_config({"cloud.agent.auth_token": "token"}):
flow_run = GraphQLResult(
{
"flow": GraphQLResult(
{
"storage": Docker(
registry_url="test", image_name="name", image_tag="tag"
).serialize(),
"id": "id",
}
),
"id": "id",
}
)
with set_temporary_config({"logging.log_to_cloud": flag}):
agent = NomadAgent(env_vars=dict(AUTH_THING="foo", PKG_SETTING="bar"))
job = agent.replace_job_spec_json(flow_run)
assert job["Job"]["TaskGroups"][0]["Tasks"][0]["Name"] == "id"
assert (
job["Job"]["TaskGroups"][0]["Tasks"][0]["Config"]["image"]
== "test/name:tag"
)
env = job["Job"]["TaskGroups"][0]["Tasks"][0]["Env"]
assert env["PREFECT__CLOUD__API"] == "https://api.prefect.io"
assert env["PREFECT__CLOUD__AGENT__AUTH_TOKEN"] == "token"
assert env["PREFECT__CONTEXT__FLOW_RUN_ID"] == "id"
assert env["PREFECT__CONTEXT__NAMESPACE"] == "default"
assert env["PREFECT__LOGGING__LOG_TO_CLOUD"] == str(flag).lower()
assert env["AUTH_THING"] == "foo"
assert env["PKG_SETTING"] == "bar"