def test_flow_run_task_with_no_matching_flow(self, client, cloud_api):
# verify a ValueError is raised if the client returns no flows
task = FlowRunTask(flow_name="Test Flow", project_name="Test Project")
client.graphql = MagicMock(return_value=MagicMock(data=MagicMock(
flow=[])))
with pytest.raises(ValueError, match="Flow 'Test Flow' not found."):
task.run()
def test_idempotency_key_uses_map_index_if_present(self, client, cloud_api):
# verify that create_flow_run was called
task = FlowRunTask(project_name="Test Project", flow_name="Test Flow")
# verify that run returns the new flow run ID
with prefect.context(flow_run_id="test-id", map_index=4):
assert task.run() == "xyz890"
# verify create_flow_run was called with the correct arguments
client.create_flow_run.assert_called_once_with(
flow_id="abc123", idempotency_key="test-id-4", parameters=None, context=None
)
def test_flow_run_task(self, client, server_api):
# verify that create_flow_run was called
task = FlowRunTask(flow_name="Test Flow", parameters={"test": "ing"},)
# verify that run returns the new flow run ID
assert task.run() == "xyz890"
# verify the GraphQL query was called with the correct arguments
query_args = list(client.graphql.call_args_list[0][0][0]["query"].keys())[0]
assert "Test Flow" in query_args
# verify create_flow_run was called with the correct arguments
client.create_flow_run.assert_called_once_with(
flow_id="abc123", parameters={"test": "ing"}
)
def test_flow_run_task_with_flow_run_id(self, client, cloud_api):
# verify that create_flow_run was called
task = FlowRunTask(
project_name="Test Project",
flow_name="Test Flow",
parameters={"test": "ing"},
)
# verify that run returns the new flow run ID
with prefect.context(flow_run_id="test-id"):
assert task.run() == "xyz890"
# verify create_flow_run was called with the correct arguments
client.create_flow_run.assert_called_once_with(
flow_id="abc123",
parameters={"test": "ing"},
idempotency_key="test-id")
def test_initialization(self, server_api):
# verify that the task is initialized as expected
task = FlowRunTask(
name="My Flow Run Task",
checkpoint=False,
flow_name="Test Flow",
parameters={"test": "ing"},
)
assert task.name == "My Flow Run Task"
assert task.checkpoint is False
assert task.flow_name == "Test Flow"
assert task.parameters == {"test": "ing"}
def test_flow_run_task(self, client, cloud_api):
# verify that create_flow_run was called
task = FlowRunTask(
project_name="Test Project",
flow_name="Test Flow",
parameters={"test": "ing"},
run_name="test-run",
)
# verify that run returns the new flow run ID
assert task.run() == "xyz890"
# verify the GraphQL query was called with the correct arguments
query_args = list(
client.graphql.call_args_list[0][0][0]["query"].keys())[0]
assert "Test Project" in query_args
assert "Test Flow" in query_args
# verify create_flow_run was called with the correct arguments
client.create_flow_run.assert_called_once_with(
flow_id="abc123",
parameters={"test": "ing"},
idempotency_key=None,
context=None,
run_name="test-run",
)
def test_initialization(self, cloud_api):
# verify that the task is initialized as expected
task = FlowRunTask(
name="My Flow Run Task",
checkpoint=False,
project_name="Test Project",
flow_name="Test Flow",
new_flow_context={"foo": "bar"},
parameters={"test": "ing"},
)
assert task.name == "My Flow Run Task"
assert task.checkpoint is False
assert task.project_name == "Test Project"
assert task.flow_name == "Test Flow"
assert task.new_flow_context == {"foo": "bar"}
assert task.parameters == {"test": "ing"}
def test_flow_run_task_without_project_name(self, cloud_api):
# verify that a ValueError is raised without a project name
task = FlowRunTask(flow_name="Test Flow")
with pytest.raises(ValueError, match="Must provide a project name."):
task.run()
def test_flow_run_task_without_flow_name(self, server_api):
# verify that a ValueError is raised without a flow name
task = FlowRunTask()
with pytest.raises(ValueError, match="Must provide a flow name."):
task.run()