def test_update_project(mock_raw_update_project):
client = _SynchronousFlyteClient(url="a.b.com", insecure=True)
project = _Project("foo",
"name",
"description",
state=_Project.ProjectState.ACTIVE)
client.update_project(project)
mock_raw_update_project.assert_called_with(project.to_flyte_idl())
def test_update_project(mock_raw_update_project):
client = _SynchronousFlyteClient(
PlatformConfig.for_endpoint("a.b.com", True))
project = _Project("foo",
"name",
"description",
state=_Project.ProjectState.ACTIVE)
client.update_project(project)
mock_raw_update_project.assert_called_with(project.to_flyte_idl())
def test_list_projects_paginated(mock_raw_list_projects):
client = _SynchronousFlyteClient(url="a.b.com", insecure=True)
client.list_projects_paginated(limit=100, token="")
project_list_request = _project_pb2.ProjectListRequest(limit=100,
token="",
filters=None,
sort_by=None)
mock_raw_list_projects.assert_called_with(
project_list_request=project_list_request)
def test_list_projects_paginated(mock_raw_list_projects):
client = _SynchronousFlyteClient(
PlatformConfig.for_endpoint("a.b.com", True))
client.list_projects_paginated(limit=100, token="")
project_list_request = _project_pb2.ProjectListRequest(limit=100,
token="",
filters=None,
sort_by=None)
mock_raw_list_projects.assert_called_with(
project_list_request=project_list_request)
def test_create_upload_location(mock_raw_create_upload_location):
client = _SynchronousFlyteClient(
PlatformConfig.for_endpoint("a.b.com", True))
client.get_upload_signed_url("foo", "bar", bytes(), "baz.qux",
timedelta(minutes=42))
duration_pb = Duration()
duration_pb.FromTimedelta(timedelta(minutes=42))
create_upload_location_request = _data_proxy_pb2.CreateUploadLocationRequest(
project="foo",
domain="bar",
filename="baz.qux",
expires_in=duration_pb)
mock_raw_create_upload_location.assert_called_with(
create_upload_location_request)
def __init__(self, *args, **kwargs):
# TODO: React to changing configs. For now this is frozen for the lifetime of the process, which covers most
# TODO: use cases.
if type(self)._CLIENT is None:
c = _SynchronousFlyteClient(*args, **kwargs)
type(self)._CLIENT = c