Merge branch 'consolidate-tests-2' into client

Author: adamtheturtle

tests/test_get_database_summary_report.py

@@ -111,6 +111,7 @@ def test_custom_base_url(self, high_quality_image: io.BytesIO) -> None:
                 image=high_quality_image,
             )
 
+
 class TestListTargets:
     """
     Tests for listing targets.
@@ -128,6 +129,7 @@ def test_list_targets(
         id_2 = client.add_target(name='a', width=1, image=high_quality_image)
         assert sorted(client.list_targets()) == sorted([id_1, id_2])
 
+
 class TestDelete:
     """
     Test for deleting a target.
@@ -151,3 +153,125 @@ def test_delete_target(
         client.delete_target(target_id=target_id)
         with pytest.raises(UnknownTarget):
             client.get_target_record(target_id=target_id)
+
+
+class TestGetTargetSummaryReport:
+    """
+    Tests for getting a summary report for a target.
+    """
+
+    def test_get_target_summary_report(
+        self,
+        client: VWS,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        Details of a target are returned by ``get_target_summary_report``.
+        """
+        target_id = client.add_target(
+            name='x',
+            width=1,
+            image=high_quality_image,
+        )
+
+        result = client.get_target_summary_report(target_id=target_id)
+        expected_keys = {
+            'status',
+            'result_code',
+            'transaction_id',
+            'database_name',
+            'target_name',
+            'upload_date',
+            'active_flag',
+            'tracking_rating',
+            'total_recos',
+            'current_month_recos',
+            'previous_month_recos',
+        }
+        assert result.keys() == expected_keys
+
+
+class TestGetDatabaseSummaryReport:
+    """
+    Tests for getting a summary report for a database.
+    """
+
+    def test_get_target(self, client: VWS) -> None:
+        """
+        Details of a database are returned by ``get_database_summary_report``.
+        """
+        report = client.get_database_summary_report()
+        expected_keys = {
+            'active_images',
+            'current_month_recos',
+            'failed_images',
+            'inactive_images',
+            'name',
+            'previous_month_recos',
+            'processing_images',
+            'reco_threshold',
+            'request_quota',
+            'request_usage',
+            'result_code',
+            'target_quota',
+            'total_recos',
+            'transaction_id',
+        }
+        assert report.keys() == expected_keys
+
+
+class TestGetTargetRecord:
+    """
+    Tests for getting a record of a target.
+    """
+
+    def test_get_target_record(
+        self,
+        client: VWS,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        Details of a target are returned by ``get_target_record``.
+        """
+        target_id = client.add_target(
+            name='x',
+            width=1,
+            image=high_quality_image,
+        )
+
+        result = client.get_target_record(target_id=target_id)
+
+        expected_keys = {
+            'target_id',
+            'active_flag',
+            'name',
+            'width',
+            'tracking_rating',
+            'reco_rating',
+        }
+        assert result.keys() == expected_keys
+
+
+class TestWaitForTargetProcessed:
+    """
+    Tests for waiting for a target to be processed.
+    """
+
+    def test_wait_for_target_processed(
+        self,
+        client: VWS,
+        high_quality_image: io.BytesIO,
+    ) -> None:
+        """
+        It is possible to wait until a target is processed.
+        """
+        target_id = client.add_target(
+            name='x',
+            width=1,
+            image=high_quality_image,
+        )
+        report = client.get_target_summary_report(target_id=target_id)
+        assert report['status'] == 'processing'
+        client.wait_for_target_processed(target_id=target_id)
+        report = client.get_target_summary_report(target_id=target_id)
+        assert report['status'] != 'processing'