fix tests

Lightning-AI · May 12, 2023 · 7eeabc9 · 7eeabc9
1 parent bad6e32
commit 7eeabc9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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")]),
  )