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'