Add test for custom base URL

Author: adamtheturtle

tests/test_add_target.py

@@ -4,6 +4,8 @@
 
 import io
 
+from mock_vws import MockVWS
+
 from vws import VWS
 
 
@@ -32,3 +34,28 @@ def test_add_two_targets(
         """
         client.add_target(name='x', width=1, image=high_quality_image)
         client.add_target(name='a', width=1, image=high_quality_image)
+
+
+class TestCustomBaseURL:
+    """
+    Tests for adding images to databases under custom VWS URLs.
+    """
+
+    def test_custom_base_url(self, high_quality_image: io.BytesIO) -> None:
+        """
+        It is possible to use add a target to a database under a custom VWS
+        URL.
+        """
+        base_vws_url = 'http://example.com'
+        with MockVWS(base_vws_url=base_vws_url) as mock:
+            client = VWS(
+                server_access_key=mock.server_access_key,
+                server_secret_key=mock.server_secret_key,
+                base_vws_url=base_vws_url,
+            )
+
+            client.add_target(
+                name='x',
+                width=1,
+                image=high_quality_image,
+            )