Progress towards tests for adding a target - success case

Author: adamtheturtle

src/vws/__init__.py

@@ -1,3 +1,9 @@
 """
 A library for Vuforia Web Services.
 """
+
+from .vws import VWS
+
+__all__ = [
+    'VWS',
+]

tests/fixtures/__init__.py

@@ -0,0 +1,3 @@
+"""
+Common fixtures.
+"""

tests/fixtures/images.py

@@ -0,0 +1,20 @@
+"""
+Fixtures for images.
+"""
+
+import io
+
+import pytest
+
+
+@pytest.fixture()
+def high_quality_image() -> io.BytesIO:
+    """
+    Return an image file which is expected to have a 'success' status when
+    added to a target and a high tracking rating.
+
+    At the time of writing, this image gains a tracking rating of 5.
+    """
+    path = 'tests/data/high_quality_image.jpg'
+    with open(path, 'rb') as high_quality_image_file:
+        return io.BytesIO(high_quality_image_file.read())

tests/test_add_target.py

@@ -0,0 +1,42 @@
+"""
+Tests for helper function for adding a target to a Vuforia database.
+"""
+
+import io
+
+from vws import VWS
+from vws.exceptions import (
+    BadImage,
+    Fail,
+    ImageTooLarge,
+    MetadataTooLarge,
+    ProjectInactive,
+    TargetNameExist,
+)
+
+
+class TestSuccess:
+    """
+    Tests for successfully adding a target.
+    """
+
+    def test_add_target(
+        self,
+        client: VWS,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        No exception is raised when adding one target.
+        """
+        client.add_target(name='x', width=1, image=high_quality_image)
+
+    def test_add_two_targets(
+        self,
+        client: VWS,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        No exception is raised when adding two targets with different names.
+        """
+        client.add_target(name='x', width=1, image=high_quality_image)
+        client.add_target(name='a', width=1, image=high_quality_image)