def test_update_project(mock_channel, mock_admin):
client = _RawSynchronousFlyteClient(url="a.b.com", insecure=True)
project = _project_pb2.Project(id="foo",
name="name",
description="description",
state=_project_pb2.Project.ACTIVE)
client.update_project(project)
mock_admin.AdminServiceStub().UpdateProject.assert_called_with(project)
def test_list_projects_paginated(mock_channel, mock_admin):
client = _RawSynchronousFlyteClient(url="a.b.com", insecure=True)
project_list_request = _project_pb2.ProjectListRequest(limit=100,
token="",
filters=None,
sort_by=None)
client.list_projects(project_list_request)
mock_admin.AdminServiceStub().ListProjects.assert_called_with(
project_list_request, metadata=None)
def test_client_set_token(mock_secure_channel, mock_channel, mock_admin,
mock_force):
mock_force.return_value = True
mock_secure_channel.return_value = True
mock_channel.return_value = True
mock_admin.AdminServiceStub.return_value = True
client = _RawSynchronousFlyteClient(url="a.b.com", insecure=True)
client.set_access_token("abc")
assert client._metadata[0][1] == "Bearer abc"
def test_client_set_token(mock_channel, mock_admin):
mock_channel.return_value = True
mock_admin.AdminServiceStub.return_value = True
client = _RawSynchronousFlyteClient(url='a.b.com', insecure=True)
client.set_access_token('abc')
assert client._metadata[0][1] == 'Bearer abc'
