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"