diff --git a/tests/tests_app/utilities/test_cloud.py b/tests/tests_app/utilities/test_cloud.py index 08196b06fca07..529e1c8520260 100644 --- a/tests/tests_app/utilities/test_cloud.py +++ b/tests/tests_app/utilities/test_cloud.py @@ -9,7 +9,7 @@ @mock.patch("lightning.app.core.constants.LIGHTNING_CLOUD_ORGANIZATION_ID", "organization_id") def test_get_project_picks_up_organization_id(): """Uses organization_id from `LIGHTNING_CLOUD_ORGANIZATION_ID` config var if none passed""" - lightning_client = LightningClient() + lightning_client = mock.MagicMock() lightning_client.projects_service_list_memberships = mock.MagicMock( return_value=V1ListMembershipsResponse(memberships=[V1Membership(project_id="project_id")]), ) @@ -18,7 +18,7 @@ def test_get_project_picks_up_organization_id(): def test_get_project_doesnt_pass_organization_id_if_its_not_set(): - lightning_client = LightningClient() + lightning_client = mock.MagicMock() lightning_client.projects_service_list_memberships = mock.MagicMock( return_value=V1ListMembershipsResponse(memberships=[V1Membership(project_id="project_id")]), )