fixed test

Author: tamirse

tests/unit_tests/containers/test_containers.py

@@ -8,12 +8,10 @@
     SECRETS_ENDPOINT,
     SERVERLESS_COMPUTE_RESOURCES_ENDPOINT,
     Container,
-    ContainerInfo,
     ContainerDeploymentStatus,
     ContainerRegistrySettings,
     ContainersService,
     Deployment,
-    DeploymentInfo,
     EnvVar,
     EnvVarType,
     EntrypointOverridesSettings,
@@ -215,10 +213,10 @@ def test_get_deployments(self, containers_service, deployments_endpoint):
         # assert
         assert type(deployments) == list
         assert len(deployments) == 1
-        assert type(deployment) == DeploymentInfo
+        assert type(deployment) == Deployment
         assert deployment.name == DEPLOYMENT_NAME
         assert len(deployment.containers) == 1
-        assert type(deployment.containers[0]) == ContainerInfo
+        assert type(deployment.containers[0]) == Container
         assert type(deployment.compute) == ComputeResource
         assert deployment.compute.name == COMPUTE_RESOURCE_NAME
         assert responses.assert_call_count(deployments_endpoint, 1) is True
@@ -238,7 +236,7 @@ def test_get_deployment_by_name(self, containers_service, deployments_endpoint):
         deployment = containers_service.get_deployment_by_name(DEPLOYMENT_NAME)
 
         # assert
-        assert type(deployment) == DeploymentInfo
+        assert type(deployment) == Deployment
         assert deployment.name == DEPLOYMENT_NAME
         assert len(deployment.containers) == 1
         assert deployment.containers[0].name == CONTAINER_NAME
@@ -305,7 +303,7 @@ def test_create_deployment(self, containers_service, deployments_endpoint):
         created_deployment = containers_service.create_deployment(deployment)
 
         # assert
-        assert type(created_deployment) == DeploymentInfo
+        assert type(created_deployment) == Deployment
         assert created_deployment.name == DEPLOYMENT_NAME
         assert len(created_deployment.containers) == 1
         assert created_deployment.containers[0].name == CONTAINER_NAME
@@ -324,7 +322,7 @@ def test_update_deployment(self, containers_service, deployments_endpoint):
         )
 
         # create deployment object
-        container = ContainerInfo(
+        container = Container(
             name=CONTAINER_NAME,
             image="nginx:latest",
             exposed_port=80
@@ -335,7 +333,7 @@ def test_update_deployment(self, containers_service, deployments_endpoint):
 
         compute = ComputeResource(name=COMPUTE_RESOURCE_NAME, size=1)
 
-        deployment = DeploymentInfo(
+        deployment = Deployment(
             name=DEPLOYMENT_NAME,
             container_registry_settings=container_registry_settings,
             containers=[container],
@@ -347,7 +345,7 @@ def test_update_deployment(self, containers_service, deployments_endpoint):
             DEPLOYMENT_NAME, deployment)
 
         # assert
-        assert type(updated_deployment) == DeploymentInfo
+        assert type(updated_deployment) == Deployment
         assert updated_deployment.name == DEPLOYMENT_NAME
         assert len(updated_deployment.containers) == 1
         assert updated_deployment.containers[0].name == CONTAINER_NAME