def test_create_flow_run_requires_flow_id_or_version_group_id():
with set_temporary_config({
"cloud.api": "http://my-cloud.foo",
"cloud.auth_token": "secret_token"
}):
client = Client()
with pytest.raises(ValueError,
match="flow_id or version_group_id must be provided"):
client.create_flow_run()
def run_with_cloud(bucket: str, country: str):
flow = create_flow()
flow_id = flow.register(project_name=PROJECT_NAME)
client = Client()
flow_run_id = client.create_flow_run(flow_id=flow_id,
parameters={
"bucket": bucket,
"country": country
})
print(f"Flow Run id: {flow_run_id}")
flow.run_agent(show_flow_logs=True)
def test_create_flow_run_with_input(patch_post, kwargs):
response = {
"data": {"createFlowRun": {"flow_run": {"id": "FOO"}}},
}
post = patch_post(response)
with set_temporary_config(
{"cloud.api": "http://my-cloud.foo", "cloud.auth_token": "secret_token"}
):
client = Client()
assert client.create_flow_run(**kwargs) == "FOO"
