ENG-486: fix. add possibility to remove a user default credits (#201)

neuro-inc · Nov 30, 2024 · 8efddb4 · 8efddb4
1 parent e26c09d
commit 8efddb4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/neuro_admin_client/__init__.py b/neuro_admin_client/__init__.py
@@ -467,7 +467,7 @@ async def update_org_balance_by_delta(
     async def update_org_defaults(
         self,
         org_name: str,
-        user_default_credits: Decimal,
+        user_default_credits: Decimal | None,
     ) -> Org: ...
 
     #  org user
@@ -1818,9 +1818,12 @@ async def update_org_balance_by_delta(
     async def update_org_defaults(
         self,
         org_name: str,
-        user_default_credits: Decimal,
+        user_default_credits: Decimal | None,
     ) -> Org:
-        payload: dict[str, str] = {"credits": str(user_default_credits)}
+        credits = (
+            str(user_default_credits) if user_default_credits is not None else None
+        )
+        payload: dict[str, str | None] = {"credits": credits}
         async with self._request(
             "PATCH",
             f"orgs/{org_name}/defaults",
@@ -3031,7 +3034,7 @@ async def update_org_balance_by_delta(
     async def update_org_defaults(
         self,
         org_name: str,
-        default_credits: Decimal,
+        default_credits: Decimal | None,
     ) -> Org:
         return self.DUMMY_ORG
 

diff --git a/tests/test_admin_client.py b/tests/test_admin_client.py
@@ -311,6 +311,14 @@ async def test_patch_org_defaults(self, mock_admin_server: AdminServer) -> None:
             assert org.user_default_credits == Decimal(200)
             assert org == mock_admin_server.orgs[0]
 
+            # set defaults to `None`
+            org = await client.update_org_defaults(
+                org_name="org",
+                user_default_credits=None,
+            )
+            assert org.user_default_credits is None
+            assert org == mock_admin_server.orgs[0]
+
     async def test_list_orgs(self, mock_admin_server: AdminServer) -> None:
         mock_admin_server.orgs = [
             Org(