Merge pull request #834 from adamtheturtle/project-inactive

Test projectinactive

Author: adamtheturtle

src/vws/exceptions.py

@@ -65,6 +65,21 @@ def __init__(self, response: Response) -> None:
         self.response = response
 
 
+class ProjectInactive(Exception):
+    """
+    Exception raised when Vuforia returns a response with a result code
+    'ProjectInactive'.
+    """
+
+    def __init__(self, response: Response) -> None:
+        """
+        Args:
+            response: The response to a request to Vuforia.
+        """
+        super().__init__()
+        self.response = response
+
+
 class MetadataTooLarge(Exception):
     """
     Exception raised when Vuforia returns a response with a result code

src/vws/vws.py

@@ -19,6 +19,7 @@
     Fail,
     ImageTooLarge,
     MetadataTooLarge,
+    ProjectInactive,
     TargetNameExist,
     TargetStatusProcessing,
     UnknownTarget,
@@ -105,6 +106,7 @@ def _raise_for_result_code(
         'Fail': Fail,
         'ImageTooLarge': ImageTooLarge,
         'MetadataTooLarge': MetadataTooLarge,
+        'ProjectInactive': ProjectInactive,
         'TargetNameExist': TargetNameExist,
         'TargetStatusProcessing': TargetStatusProcessing,
         'UnknownTarget': UnknownTarget,

tests/test_exceptions.py

@@ -6,7 +6,7 @@
 import random
 
 import pytest
-from mock_vws import MockVWS
+from mock_vws import MockVWS, States
 from PIL import Image
 from requests import codes
 
@@ -16,6 +16,7 @@
     Fail,
     ImageTooLarge,
     MetadataTooLarge,
+    ProjectInactive,
     TargetNameExist,
     TargetStatusProcessing,
     UnknownTarget,
@@ -138,6 +139,27 @@ def test_target_name_exist(
     assert exc.value.response.status_code == codes.FORBIDDEN
 
 
+def test_project_inactive(client: VWS, high_quality_image: io.BytesIO) -> None:
+    """
+    A ``ProjectInactive`` exception is raised if adding a target to an
+    inactive database.
+    """
+    with MockVWS(state=States.PROJECT_INACTIVE) as mock:
+        client = VWS(
+            server_access_key=mock.server_access_key,
+            server_secret_key=mock.server_secret_key,
+        )
+
+        with pytest.raises(ProjectInactive) as exc:
+            client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )
+
+    assert exc.value.response.status_code == codes.FORBIDDEN
+
+
 def test_target_status_processing(
     client: VWS,
     high_quality_image: io.BytesIO,