[FLPY-37] Added mock responses for shared account and tests

Author: awkrupka

flow360/cli/app.py

@@ -54,6 +54,7 @@ def flow360():
     is_flag=True,
     help='Always prompt for confirmation for "submit()" when creating new resources while logged into a shared account',
 )
+# pylint: disable=too-many-arguments
 def configure(
     apikey,
     profile,

tests/data/mock_webapi/organization_accounts_resp.json

@@ -0,0 +1,28 @@
+{
+  "data": {
+    "credit": 100000000000.0,
+    "s3Usage": 0,
+    "totalTaskCount": 0,
+    "monthlyTaskCount": 0,
+    "creditExpiration": "2029-05-31T07:28:31.829Z",
+    "accountType": "tenant",
+    "tenantId": "0000000-0000-0000-0000-000000000000",
+    "tenantName": "Test",
+    "clientAdmin": true,
+    "allowanceCurrentCycleAmount": null,
+    "allowanceCurrentCycleTotalAmount": null,
+    "allowanceCurrentCycleStartDate": null,
+    "allowanceCurrentCycleEndDate": null,
+    "allowanceNextCycleStartDate": null,
+    "allowanceAllCycleStartDate": null,
+    "allowanceAllCycleEndDate": null,
+    "tenantMembers": [
+      {"userIdentity": "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa", "userId": "AAAAAAAAAAAAAAAAAAAAA", "userEmail": "user1@test.com"},
+      {"userIdentity": "us-east-1:bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb", "userId": "BBBBBBBBBBBBBBBBBBBBB", "userEmail": "user2@test.com"}],
+    "intraCompanySharingEnabled": true,
+    "taskDeInfo": null,
+    "userId": "ABCDEFGHIJKLMNOPRSTUV",
+    "internal": true
+  }
+}
+

tests/mock_server.py

@@ -75,6 +75,16 @@ def json():
         return res
 
 
+class MockResponseOrganizationAccounts(MockResponse):
+    """response to retrieving shared account list"""
+
+    @staticmethod
+    def json():
+        with open(os.path.join(here, "data/mock_webapi/organization_accounts_resp.json")) as fh:
+            res = json.load(fh)
+        return res
+
+
 class MockResponseCase(MockResponse):
     """response if Case(id="00000000-0000-0000-0000-000000000000")"""
 
@@ -136,6 +146,7 @@ def json():
     "/cases/00112233-4455-6677-8899-bbbbbbbbbbbb/runtimeParams": MockResponseCaseRuntimeParams,
     "/cases/c58e7a75-e349-476a-9020-247af6b2e92b": MockResponseCase,
     "-python-client-v2": MockResponseVersions,
+    "/account": MockResponseOrganizationAccounts,
 }
 
 

tests/test_shared_accounts.py

@@ -1,7 +1,20 @@
 import pytest
 
-import flow360 as fl
+from flow360 import Accounts, Env
 
+from .mock_server import mock_response
 
-def test_shared_account():
-    pass
+
+def test_shared_account(mock_response):
+    Accounts.choose_shared_account("user1@test.com")
+
+    assert Accounts.shared_account_info() == "user1@test.com"
+    assert Env.impersonate == "aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa"
+
+    Accounts.leave_shared_account()
+
+    assert Accounts.shared_account_info() is None
+    assert Env.impersonate is None
+
+    with pytest.raises(ValueError):
+        Accounts.choose_shared_account("user3@test.com")