added tests

Author: tamirse

tests/unit_tests/containers/test_containers.py

@@ -37,7 +37,8 @@
 
 DEPLOYMENT_NAME = "test-deployment"
 CONTAINER_NAME = "test-container"
-COMPUTE_RESOURCE_NAME = "test-compute"
+COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE = "General Compute"
+COMPUTE_RESOURCE_NAME_H100 = "H100"
 SECRET_NAME = "test-secret"
 SECRET_VALUE = "test-secret-value"
 REGISTRY_CREDENTIAL_NAME = "test-credential"
@@ -82,7 +83,7 @@
         }
     ],
     "compute": {
-        "name": COMPUTE_RESOURCE_NAME,
+        "name": COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE,
         "size": 1,
         "is_available": True
     },
@@ -118,12 +119,12 @@
 # Sample compute resources data
 COMPUTE_RESOURCES_DATA = [
     {
-        "name": COMPUTE_RESOURCE_NAME,
+        "name": COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE,
         "size": 1,
         "is_available": True
     },
     {
-        "name": "large-compute",
+        "name": COMPUTE_RESOURCE_NAME_H100,
         "size": 4,
         "is_available": True
     }
@@ -218,7 +219,7 @@ def test_get_deployments(self, containers_service, deployments_endpoint):
         assert len(deployment.containers) == 1
         assert type(deployment.containers[0]) == Container
         assert type(deployment.compute) == ComputeResource
-        assert deployment.compute.name == COMPUTE_RESOURCE_NAME
+        assert deployment.compute.name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
         assert responses.assert_call_count(deployments_endpoint, 1) is True
 
     @responses.activate
@@ -240,7 +241,7 @@ def test_get_deployment_by_name(self, containers_service, deployments_endpoint):
         assert deployment.name == DEPLOYMENT_NAME
         assert len(deployment.containers) == 1
         assert deployment.containers[0].name == CONTAINER_NAME
-        assert deployment.compute.name == COMPUTE_RESOURCE_NAME
+        assert deployment.compute.name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
         assert responses.assert_call_count(url, 1) is True
 
     @responses.activate
@@ -285,7 +286,8 @@ def test_create_deployment(self, containers_service, deployments_endpoint):
                 type=VolumeMountType.SCRATCH, mount_path="/data")]
         )
 
-        compute = ComputeResource(name=COMPUTE_RESOURCE_NAME, size=1)
+        compute = ComputeResource(
+            name=COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE, size=1)
 
         container_registry_settings = ContainerRegistrySettings(
             is_private=False)
@@ -307,7 +309,7 @@ def test_create_deployment(self, containers_service, deployments_endpoint):
         assert created_deployment.name == DEPLOYMENT_NAME
         assert len(created_deployment.containers) == 1
         assert created_deployment.containers[0].name == CONTAINER_NAME
-        assert created_deployment.compute.name == COMPUTE_RESOURCE_NAME
+        assert created_deployment.compute.name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
         assert responses.assert_call_count(deployments_endpoint, 1) is True
 
     @responses.activate
@@ -331,7 +333,8 @@ def test_update_deployment(self, containers_service, deployments_endpoint):
         container_registry_settings = ContainerRegistrySettings(
             is_private=False)
 
-        compute = ComputeResource(name=COMPUTE_RESOURCE_NAME, size=1)
+        compute = ComputeResource(
+            name=COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE, size=1)
 
         deployment = Deployment(
             name=DEPLOYMENT_NAME,
@@ -349,7 +352,7 @@ def test_update_deployment(self, containers_service, deployments_endpoint):
         assert updated_deployment.name == DEPLOYMENT_NAME
         assert len(updated_deployment.containers) == 1
         assert updated_deployment.containers[0].name == CONTAINER_NAME
-        assert updated_deployment.compute.name == COMPUTE_RESOURCE_NAME
+        assert updated_deployment.compute.name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
         assert responses.assert_call_count(url, 1) is True
 
     @responses.activate
@@ -648,7 +651,73 @@ def test_get_compute_resources(self, containers_service, compute_resources_endpo
         assert type(resources) == list
         assert len(resources) == 2
         assert type(resources[0]) == ComputeResource
-        assert resources[0].name == COMPUTE_RESOURCE_NAME
+        assert resources[0].name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
+        assert resources[0].size == 1
+        assert resources[0].is_available == True
+        assert responses.assert_call_count(
+            compute_resources_endpoint, 1) is True
+
+    @responses.activate
+    def test_get_compute_resources_filter_by_size(self, containers_service, compute_resources_endpoint):
+        # arrange - add response mock
+        responses.add(
+            responses.GET,
+            compute_resources_endpoint,
+            json=[COMPUTE_RESOURCES_DATA],
+            status=200
+        )
+
+        # act
+        resources = containers_service.get_compute_resources(size=4)
+
+        # assert
+        assert type(resources) == list
+        assert len(resources) == 1
+        assert type(resources[0]) == ComputeResource
+        assert resources[0].name == COMPUTE_RESOURCE_NAME_H100
+        assert resources[0].size == 4
+        assert resources[0].is_available == True
+        assert responses.assert_call_count(
+            compute_resources_endpoint, 1) is True
+
+    @responses.activate
+    def test_get_compute_resources_filter_by_availability(self, containers_service, compute_resources_endpoint):
+        # arrange - add response mock
+        responses.add(
+            responses.GET,
+            compute_resources_endpoint,
+            json=[COMPUTE_RESOURCES_DATA],
+            status=200
+        )
+
+        # act
+        resources = containers_service.get_compute_resources(is_available=True)
+
+        # assert
+        assert type(resources) == list
+        assert len(resources) == 2
+        assert all(r.is_available == True for r in resources)
+        assert responses.assert_call_count(
+            compute_resources_endpoint, 1) is True
+
+    @responses.activate
+    def test_get_compute_resources_filter_by_size_and_availability(self, containers_service, compute_resources_endpoint):
+        # arrange - add response mock
+        responses.add(
+            responses.GET,
+            compute_resources_endpoint,
+            json=[COMPUTE_RESOURCES_DATA],
+            status=200
+        )
+
+        # act
+        resources = containers_service.get_compute_resources(
+            size=1, is_available=True)
+
+        # assert
+        assert type(resources) == list
+        assert len(resources) == 1
+        assert resources[0].name == COMPUTE_RESOURCE_NAME_GENERAL_COMPUTE
         assert resources[0].size == 1
         assert resources[0].is_available == True
         assert responses.assert_call_count(