proximl 0.5.8__py3-none-any.whl → 0.5.9__py3-none-any.whl

proximl/projects/services.py

@@ -0,0 +1,63 @@
+import json
+import logging
+
+
+class ProjectServices(object):
+    def __init__(self, proximl, project_id):
+        self.proximl = proximl
+        self.project_id = project_id
+
+    async def list(self, **kwargs):
+        resp = await self.proximl._query(
+            f"/project/{self.project_id}/services", "GET", kwargs
+        )
+        services = [ProjectService(self.proximl, **service) for service in resp]
+        return services
+
+    async def refresh(self):
+        await self.proximl._query(f"/project/{self.project_id}/services", "PATCH")
+
+
+class ProjectService:
+    def __init__(self, proximl, **kwargs):
+        self.proximl = proximl
+        self._entity = kwargs
+        self._id = self._entity.get("id")
+        self._project_uuid = self._entity.get("project_uuid")
+        self._name = self._entity.get("name")
+        self._hostname = self._entity.get("hostname")
+        self._public = self._entity.get("public")
+        self._region_uuid = self._entity.get("region_uuid")
+
+    @property
+    def id(self) -> str:
+        return self._id
+
+    @property
+    def project_uuid(self) -> str:
+        return self._project_uuid
+
+    @property
+    def name(self) -> str:
+        return self._name
+
+    @property
+    def hostname(self) -> str:
+        return self._hostname
+
+    @property
+    def public(self) -> bool:
+        return self._public
+
+    @property
+    def region_uuid(self) -> str:
+        return self._region_uuid
+
+    def __str__(self):
+        return json.dumps({k: v for k, v in self._entity.items()})
+
+    def __repr__(self):
+        return f"ProjectService( proximl , **{self._entity.__repr__()})"
+
+    def __bool__(self):
+        return bool(self._id)

{proximl-0.5.8.dist-info → proximl-0.5.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: proximl
-Version: 0.5.8
+Version: 0.5.9
 Summary: proxiML client SDK and command line utilities
 Home-page: https://github.com/proxiML/python-sdk
 Author: proxiML

{proximl-0.5.8.dist-info → proximl-0.5.9.dist-info}/RECORD

@@ -36,6 +36,9 @@ proximl/cli/cloudbender/reservation.py,sha256=xzHs5l8BbmYgKUq6kfFU-jEtRQY0j_vYnm
 proximl/cli/cloudbender/service.py,sha256=YimwXQml82-PKKQIqu9Cvggo9wOKLVTJPxMvDf4pgn8,2869
 proximl/cli/job/__init__.py,sha256=s8mU2PvCWDcv4gGT3EmjHn8MIZlXBAoayoZKmnKpXnY,6545
 proximl/cli/job/create.py,sha256=sGvbenY0yxvxHo-FZVbdw8FaZx5D4ekTCjD7P4YHG4g,34288
+proximl/cli/project/__init__.py,sha256=VhBZb1WAbe9NCTrYRl4aRNppFn1loDDMQ-cTqBCeV7A,3402
+proximl/cli/project/key.py,sha256=23ugcWOAxNJvC8VzG3FsEUWJ1_FZANnhq66cO04ZEVY,3203
+proximl/cli/project/secret.py,sha256=uys2FsXuO_8_EoolQNuOZ9btX4I0LXBeKG8dGx6StMM,1621
 proximl/cloudbender/__init__.py,sha256=iE29obtC0_9f0IhRvHQcG5aY58fVhVYipTakpjAhdss,64
 proximl/cloudbender/cloudbender.py,sha256=E1_MyOhl-Fq5gr6vitCdI34_f24Pa5cg9YrFXIXPwg8,634
 proximl/cloudbender/data_connectors.py,sha256=urhmaAi4TBR-6L1CD2qS7SG3Dcwo9hOi8Gx613MzKVU,3297
@@ -47,6 +50,13 @@ proximl/cloudbender/providers.py,sha256=cH5lCew5WCFpXYS93vuoGNWkZyx7T_mdMH6YNWp2
 proximl/cloudbender/regions.py,sha256=Nu1LT6nuLD8Nt-5-7_FLlxDNZoDDAY6QduTdEBqfxJA,3570
 proximl/cloudbender/reservations.py,sha256=14ImJRLWQGG7CXDYhDnOI2W8pnP6CVVG2aVpysQVN0E,3586
 proximl/cloudbender/services.py,sha256=KR0EQ-BUVk1ov9Q195NsnfiSiMb9vOYD8XyxD87RRvw,3222
+proximl/projects/__init__.py,sha256=6NKCcHtQMeGB1IyU-djANphfnDX6MEkrXUM5Fyq9fWg,75
+proximl/projects/data_connectors.py,sha256=559yJ-9qmva2JcNReuCyUQxdrSRIEROcGX5BDVr8ySs,1662
+proximl/projects/datastores.py,sha256=zAmKX9CibeS7PSXVvExk0SVrEo9Zq83kz6CQw0GoHck,1560
+proximl/projects/keys.py,sha256=yT5pgv1mGm50CaORf-HnsbDULad69G9I_QpVMN0zcuE,2157
+proximl/projects/projects.py,sha256=3aEGHxOMYbO9FaLSzSAA94A5ZGzljq-fBJk1QcSidCE,2755
+proximl/projects/secrets.py,sha256=NxjP1_CR9UQUMazeS-esin3zH54xZ4b4R11YNSU4hsI,2157
+proximl/projects/services.py,sha256=Jian2RHV9u3pHQKa7AsHKj25DrR4DstB-JaIDhSu7Tw,1679
 tests/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/integration/conftest.py,sha256=zRWpherX-yfbpk7xqZk9bIZCyJ-dwVeszY_7kekn2M4,1134
 tests/integration/test_checkpoints_integration.py,sha256=DUA6ZlX0gsnKCEvFmWCKbKkfJG3dQG6uD8T3fgo7y0M,3230
@@ -59,9 +69,15 @@ tests/integration/test_projects_integration.py,sha256=XdGGqvOQby77kMfoD_zAukKvN6
 tests/integration/test_volumes_integration.py,sha256=Xo2Whw2U-7jyvESIkyex3f0SMXKlExe3kLmsbpXTHhQ,3270
 tests/integration/cloudbender/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/integration/cloudbender/test_providers_integration.py,sha256=gFqPQom-Cn1iZC50_ChQ2us2_f4tIPATQSAUcWdf7ss,1473
+tests/integration/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/integration/projects/conftest.py,sha256=UW9t62xiBRfU3anBRB-sdURD2eqo7xVAUsfTGsgWpcQ,208
+tests/integration/projects/test_projects_integration.py,sha256=wlRLxZkMnJZHlNPaeRFjAeyUlMfbgdrrFEI2eMBcoUI,1241
+tests/integration/projects/test_projects_keys_integration.py,sha256=cLvka_XlQHCYAFX4-gsH57KzhnpQZOGRvk9Q-HZEYbI,1423
+tests/integration/projects/test_projects_secrets_integration.py,sha256=Q8GEJkBE4X38F36sJza3VRo1pWdF9qEEr2kuADqgBB8,1478
 tests/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/unit/conftest.py,sha256=dilDp7ZSmyg9tJbkJoGcqUF07JBTdkNoULOR6qK7NsU,31197
 tests/unit/test_auth.py,sha256=IJZHT5CZLNfu3MybTdEWIsKlvxNfFYpZ6oWYzbS856g,858
+tests/unit/test_auth_unit.py,sha256=IJZHT5CZLNfu3MybTdEWIsKlvxNfFYpZ6oWYzbS856g,858
 tests/unit/test_checkpoints_unit.py,sha256=WfFE6HvQHglBKfcJ3B0IYXdXpaNmcL2BDurRQmIbFAU,16008
 tests/unit/test_connections_unit.py,sha256=LFAZzlrvL9oM8rZJTiC1oA9quw1KA2vMUCc3LV6SjXs,5507
 tests/unit/test_datasets_unit.py,sha256=Bw6bw2Dhe3BPi2KAM-2S3PKBJO9hMRdTWMECA2lfhJU,15879
@@ -72,6 +88,7 @@ tests/unit/test_jobs_unit.py,sha256=kCrm2FdogqYgji0Coe9KGaODYEsXmvuV46FN2mUE1Ts,
 tests/unit/test_models_unit.py,sha256=wgwdvJLt6QZ_IKCvp6Kmt2Z4b_CvIN8KCuFqlBaxDe8,15064
 tests/unit/test_projects_unit.py,sha256=_SwPFhy2Xl6HRt6w5_HYrqtZJHAzFzoCZ3kqAS1Zn3c,9530
 tests/unit/test_proximl.py,sha256=E-er5V1O-4ZVfTO2R-2wGsKN80q0_BVG5ho9Ab9lnE4,1701
+tests/unit/test_proximl_unit.py,sha256=E-er5V1O-4ZVfTO2R-2wGsKN80q0_BVG5ho9Ab9lnE4,1701
 tests/unit/test_volumes_unit.py,sha256=YcyrGPCq7XfAtCdKdp44723o7mLGIu5_hKOdhAFVYV4,15305
 tests/unit/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/unit/cli/conftest.py,sha256=w6p_2URicywJKUCtY79tSD_mx8cwJtxHbK_Lu3grOYs,236
@@ -101,9 +118,16 @@ tests/unit/cloudbender/test_providers_unit.py,sha256=y63VCqHXb4Yu8sh0kW30-ojRvv9
 tests/unit/cloudbender/test_regions_unit.py,sha256=9bvP268gpNyygjh1IEpSSiUt2aP6okv7QOsV1XoaIS0,6299
 tests/unit/cloudbender/test_reservations_unit.py,sha256=ICuFT5sexnLvS7taoC18yQYuDZHpBRrNuCj3Uq_Arwo,5624
 tests/unit/cloudbender/test_services_unit.py,sha256=iYaQpyCXDg77GQEIhmgiVwKX83jyvIf-4-4oya5WA_o,5043
-proximl-0.5.8.dist-info/LICENSE,sha256=ADFxLEZDxKY0j4MdyUd5GNuhQ18rnWH5rOz1ZG7yiOA,1069
-proximl-0.5.8.dist-info/METADATA,sha256=VeQ5LeqASFJQY6mCToDYKwkANOEsKozBy066X_psG6A,7344
-proximl-0.5.8.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-proximl-0.5.8.dist-info/entry_points.txt,sha256=HmI311IIabkZReMCXu-nGbvIEW-KfaduAOyfiSqt5SY,63
-proximl-0.5.8.dist-info/top_level.txt,sha256=-TWqc9tAaxmWmW4c7uYsmzPEYUIoh6z02xxqPbv7Kys,23
-proximl-0.5.8.dist-info/RECORD,,
+tests/unit/projects/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/unit/projects/test_project_data_connectors_unit.py,sha256=Yx2yCUgcuN_KfTKynZxse0Nx8jRrSlMygI7R8a-NjE4,3385
+tests/unit/projects/test_project_datastores_unit.py,sha256=WZMKkhpEg2dcevJqT7k6QRcYJF6FJfq177BDk2GWOBk,3129
+tests/unit/projects/test_project_keys_unit.py,sha256=n5Y2rKaenHA6JQFlViE1_dZ00FZ0PhM3C7rQtkuiRGM,3161
+tests/unit/projects/test_project_secrets_unit.py,sha256=VE9L91FJodcwVGizfF65WYMiHZaF0s2AdW1aiJ3z7xA,3276
+tests/unit/projects/test_project_services_unit.py,sha256=PzeNuJRuAG7RkrPWX0FfgFTt6-63FviecrDY06rLQ6A,3331
+tests/unit/projects/test_projects_unit.py,sha256=lTGdKOCN2bjzGORhaDVVlnzJBuEItUDGN36MsyQjDo4,3823
+proximl-0.5.9.dist-info/LICENSE,sha256=ADFxLEZDxKY0j4MdyUd5GNuhQ18rnWH5rOz1ZG7yiOA,1069
+proximl-0.5.9.dist-info/METADATA,sha256=Gaq8RcfBkBF3J70Pf7Q60Sf6Fm2Yv91phU46RIinlqA,7344
+proximl-0.5.9.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+proximl-0.5.9.dist-info/entry_points.txt,sha256=HmI311IIabkZReMCXu-nGbvIEW-KfaduAOyfiSqt5SY,63
+proximl-0.5.9.dist-info/top_level.txt,sha256=-TWqc9tAaxmWmW4c7uYsmzPEYUIoh6z02xxqPbv7Kys,23
+proximl-0.5.9.dist-info/RECORD,,

tests/integration/projects/conftest.py

@@ -0,0 +1,8 @@
+from pytest import fixture
+
+
+@fixture(scope="module")
+async def project(proximl):
+    project = await proximl.projects.create(name="New Project", copy_keys=False)
+    yield project
+    await project.remove()

tests/integration/projects/test_projects_integration.py

@@ -0,0 +1,38 @@
+import re
+import sys
+import asyncio
+from pytest import mark, fixture
+
+pytestmark = [mark.sdk, mark.integration, mark.projects]
+
+
+@mark.create
+@mark.asyncio
+class GetProjectsTests:
+    async def test_get_projects(self, proximl):
+        projects = await proximl.projects.list()
+        assert len(projects) > 0
+
+    async def test_get_project(self, proximl, project):
+        response = await proximl.projects.get(project.id)
+        assert response.id == project.id
+
+    async def test_project_properties(self, project):
+        assert isinstance(project.id, str)
+        assert isinstance(project.name, str)
+        assert isinstance(project.owner_name, str)
+        assert isinstance(project.is_owner, bool)
+        assert project.name == "New Project"
+        assert project.is_owner
+
+    async def test_project_str(self, project):
+        string = str(project)
+        regex = r"^{.*\"id\": \"" + project.id + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    async def test_project_repr(self, project):
+        string = repr(project)
+        regex = r"^Project\( proximl , \*\*{.*'id': '" + project.id + r"'.*}\)$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)

tests/integration/projects/test_projects_keys_integration.py

@@ -0,0 +1,43 @@
+import re
+import sys
+import asyncio
+from pytest import mark, fixture
+
+pytestmark = [mark.sdk, mark.integration, mark.projects]
+
+
+@mark.create
+@mark.asyncio
+class ProjectKeysTests:
+    @fixture(scope="class")
+    async def project_key(self, project):
+        project_key = await project.keys.put(
+            type="aws", key_id="ASFHALKF", secret="IUHKLHKAHF"
+        )
+        yield project_key
+        await project.keys.remove(type="aws")
+
+    async def test_list_project_keys(self, project, project_key):
+        keys = await project.keys.list()
+        assert len(keys) > 0
+
+    async def test_project_key_properties(self, project, project_key):
+        assert isinstance(project_key.project_uuid, str)
+        assert isinstance(project_key.type, str)
+        assert isinstance(project_key.key_id, str)
+        assert project_key.type == "aws"
+        assert project.id == project_key.project_uuid
+
+    async def test_project_key_str(self, project_key):
+        string = str(project_key)
+        regex = r"^{.*\"type\": \"" + project_key.type + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    async def test_project_key_repr(self, project_key):
+        string = repr(project_key)
+        regex = (
+            r"^ProjectKey\( proximl , \*\*{.*'type': '" + project_key.type + r"'.*}\)$"
+        )
+        assert isinstance(string, str)
+        assert re.match(regex, string)

tests/integration/projects/test_projects_secrets_integration.py

@@ -0,0 +1,44 @@
+import re
+import sys
+import asyncio
+from pytest import mark, fixture
+
+pytestmark = [mark.sdk, mark.integration, mark.projects]
+
+
+@mark.create
+@mark.asyncio
+class ProjectSecretsTests:
+    @fixture(scope="class")
+    async def project_secret(self, project):
+        project_secret = await project.secrets.put(
+            name="secret_value", value="IUHKLHKAHF"
+        )
+        yield project_secret
+        await project.secrets.remove(name="secret_value")
+
+    async def test_list_project_keys(self, project, project_secret):
+        secrets = await project.secrets.list()
+        assert len(secrets) > 0
+
+    async def test_project_secret_properties(self, project, project_secret):
+        assert isinstance(project_secret.project_uuid, str)
+        assert isinstance(project_secret.name, str)
+        assert project_secret.name == "secret_value"
+        assert project.id == project_secret.project_uuid
+
+    async def test_project_secret_str(self, project_secret):
+        string = str(project_secret)
+        regex = r"^{.*\"name\": \"" + project_secret.name + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    async def test_project_secret_repr(self, project_secret):
+        string = repr(project_secret)
+        regex = (
+            r"^ProjectSecret\( proximl , \*\*{.*'name': '"
+            + project_secret.name
+            + r"'.*}\)$"
+        )
+        assert isinstance(string, str)
+        assert re.match(regex, string)

tests/unit/projects/test_project_data_connectors_unit.py

@@ -0,0 +1,102 @@
+import re
+import json
+import logging
+from unittest.mock import AsyncMock, patch
+from pytest import mark, fixture, raises
+from aiohttp import WSMessage, WSMsgType
+
+import proximl.projects.data_connectors as specimen
+from proximl.exceptions import (
+    ApiError,
+    SpecificationError,
+    ProxiMLException,
+)
+
+pytestmark = [mark.sdk, mark.unit, mark.projects]
+
+
+@fixture
+def project_data_connectors(mock_proximl):
+    yield specimen.ProjectDataConnectors(mock_proximl, project_id="1")
+
+
+@fixture
+def project_data_connector(mock_proximl):
+    yield specimen.ProjectDataConnector(
+        mock_proximl,
+        id="ds-id-1",
+        name="connector 1",
+        project_uuid="proj-id-1",
+        type="custom",
+        region_uuid="reg-id-1",
+    )
+
+
+class ProjectDataConnectorsTests:
+    @mark.asyncio
+    async def test_project_data_connectors_refresh(
+        self, project_data_connectors, mock_proximl
+    ):
+        api_response = dict()
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        await project_data_connectors.refresh()
+        mock_proximl._query.assert_called_once_with(
+            "/project/1/data_connectors", "PATCH"
+        )
+
+    @mark.asyncio
+    async def test_project_data_connectors_list(
+        self, project_data_connectors, mock_proximl
+    ):
+        api_response = [
+            {
+                "project_uuid": "proj-id-1",
+                "region_uuid": "reg-id-1",
+                "id": "store-id-1",
+                "type": "custom",
+                "name": "On-Prem Connection A",
+            },
+            {
+                "project_uuid": "proj-id-1",
+                "region_uuid": "reg-id-2",
+                "id": "store-id-2",
+                "type": "custom",
+                "name": "Cloud Connection B",
+            },
+        ]
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        resp = await project_data_connectors.list()
+        mock_proximl._query.assert_called_once_with(
+            "/project/1/data_connectors", "GET", dict()
+        )
+        assert len(resp) == 2
+
+
+class ProjectDataConnectorTests:
+    def test_project_data_connector_properties(self, project_data_connector):
+        assert isinstance(project_data_connector.id, str)
+        assert isinstance(project_data_connector.name, str)
+        assert isinstance(project_data_connector.project_uuid, str)
+        assert isinstance(project_data_connector.type, str)
+        assert isinstance(project_data_connector.region_uuid, str)
+
+    def test_project_data_connector_str(self, project_data_connector):
+        string = str(project_data_connector)
+        regex = r"^{.*\"id\": \"" + project_data_connector.id + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_data_connector_repr(self, project_data_connector):
+        string = repr(project_data_connector)
+        regex = (
+            r"^ProjectDataConnector\( proximl , \*\*{.*'id': '"
+            + project_data_connector.id
+            + r"'.*}\)$"
+        )
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_data_connector_bool(self, project_data_connector, mock_proximl):
+        empty_project_data_connector = specimen.ProjectDataConnector(mock_proximl)
+        assert bool(project_data_connector)
+        assert not bool(empty_project_data_connector)

tests/unit/projects/test_project_datastores_unit.py

@@ -0,0 +1,96 @@
+import re
+import json
+import logging
+from unittest.mock import AsyncMock, patch
+from pytest import mark, fixture, raises
+from aiohttp import WSMessage, WSMsgType
+
+import proximl.projects.datastores as specimen
+from proximl.exceptions import (
+    ApiError,
+    SpecificationError,
+    ProxiMLException,
+)
+
+pytestmark = [mark.sdk, mark.unit, mark.projects]
+
+
+@fixture
+def project_datastores(mock_proximl):
+    yield specimen.ProjectDatastores(mock_proximl, project_id="1")
+
+
+@fixture
+def project_datastore(mock_proximl):
+    yield specimen.ProjectDatastore(
+        mock_proximl,
+        id="ds-id-1",
+        name="datastore 1",
+        project_uuid="proj-id-1",
+        type="nfs",
+        region_uuid="reg-id-1",
+    )
+
+
+class ProjectDatastoresTests:
+    @mark.asyncio
+    async def test_project_datastores_refresh(self, project_datastores, mock_proximl):
+        api_response = dict()
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        await project_datastores.refresh()
+        mock_proximl._query.assert_called_once_with("/project/1/datastores", "PATCH")
+
+    @mark.asyncio
+    async def test_project_datastores_list(self, project_datastores, mock_proximl):
+        api_response = [
+            {
+                "project_uuid": "proj-id-1",
+                "region_uuid": "reg-id-1",
+                "id": "store-id-1",
+                "type": "nfs",
+                "name": "On-prem NFS",
+            },
+            {
+                "project_uuid": "proj-id-1",
+                "region_uuid": "reg-id-2",
+                "id": "store-id-2",
+                "type": "smb",
+                "name": "GCP Samba",
+            },
+        ]
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        resp = await project_datastores.list()
+        mock_proximl._query.assert_called_once_with(
+            "/project/1/datastores", "GET", dict()
+        )
+        assert len(resp) == 2
+
+
+class ProjectDatastoreTests:
+    def test_project_datastore_properties(self, project_datastore):
+        assert isinstance(project_datastore.id, str)
+        assert isinstance(project_datastore.name, str)
+        assert isinstance(project_datastore.project_uuid, str)
+        assert isinstance(project_datastore.type, str)
+        assert isinstance(project_datastore.region_uuid, str)
+
+    def test_project_datastore_str(self, project_datastore):
+        string = str(project_datastore)
+        regex = r"^{.*\"id\": \"" + project_datastore.id + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_datastore_repr(self, project_datastore):
+        string = repr(project_datastore)
+        regex = (
+            r"^ProjectDatastore\( proximl , \*\*{.*'id': '"
+            + project_datastore.id
+            + r"'.*}\)$"
+        )
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_datastore_bool(self, project_datastore, mock_proximl):
+        empty_project_datastore = specimen.ProjectDatastore(mock_proximl)
+        assert bool(project_datastore)
+        assert not bool(empty_project_datastore)

tests/unit/projects/test_project_keys_unit.py

@@ -0,0 +1,96 @@
+import re
+import json
+import logging
+from unittest.mock import AsyncMock, patch
+from pytest import mark, fixture, raises
+from aiohttp import WSMessage, WSMsgType
+
+import proximl.projects.keys as specimen
+from proximl.exceptions import (
+    ApiError,
+    SpecificationError,
+    ProxiMLException,
+)
+
+pytestmark = [mark.sdk, mark.unit, mark.projects]
+
+
+@fixture
+def project_keys(mock_proximl):
+    yield specimen.ProjectKeys(mock_proximl, project_id="1")
+
+
+@fixture
+def project_key(mock_proximl):
+    yield specimen.ProjectKey(
+        mock_proximl, project_uuid="proj-id-1", type="aws", key_id="AIYHGFSDLK"
+    )
+
+
+class ProjectKeysTests:
+    @mark.asyncio
+    async def test_project_keys_list(self, project_keys, mock_proximl):
+        api_response = [
+            {"project_uuid": "proj-id-1", "type": "aws", "key_id": "AIYHGFSDLK"},
+            {"project_uuid": "proj-id-1", "type": "gcp", "key_id": "credentials.json"},
+        ]
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        resp = await project_keys.list()
+        mock_proximl._query.assert_called_once_with("/project/1/keys", "GET", dict())
+        assert len(resp) == 2
+
+    @mark.asyncio
+    async def test_remove_project_key(
+        self,
+        project_keys,
+        mock_proximl,
+    ):
+        api_response = dict()
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        await project_keys.remove("aws")
+        mock_proximl._query.assert_called_once_with(
+            "/project/1/key/aws", "DELETE", dict()
+        )
+
+    @mark.asyncio
+    async def test_put_project_key(self, project_keys, mock_proximl):
+        requested_config = dict(type="aws", key_id="AIUDHADA", secret="ASKHJSLKF")
+        expected_payload = dict(key_id="AIUDHADA", secret="ASKHJSLKF")
+        api_response = {
+            "project_uuid": "project-id-1",
+            "type": "aws",
+            "key_id": "AIUDHADA",
+        }
+
+        mock_proximl._query = AsyncMock(return_value=api_response)
+        response = await project_keys.put(**requested_config)
+        mock_proximl._query.assert_called_once_with(
+            "/project/1/key/aws", "PUT", None, expected_payload
+        )
+        assert response.type == "aws"
+
+
+class ProjectKeyTests:
+    def test_project_key_properties(self, project_key):
+        assert isinstance(project_key.type, str)
+        assert isinstance(project_key.key_id, str)
+        assert isinstance(project_key.project_uuid, str)
+
+    def test_project_key_str(self, project_key):
+        string = str(project_key)
+        regex = r"^{.*\"type\": \"" + project_key.type + r"\".*}$"
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_key_repr(self, project_key):
+        string = repr(project_key)
+        regex = (
+            r"^ProjectKey\( proximl , \*\*{.*'type': '" + project_key.type + r"'.*}\)$"
+        )
+        assert isinstance(string, str)
+        assert re.match(regex, string)
+
+    def test_project_key_bool(self, project_key, mock_proximl):
+        empty_project_key = specimen.ProjectKey(mock_proximl)
+        assert bool(project_key)
+        assert not bool(empty_project_key)