Add: Unit test to empty result behavior

greenbone · Aug 13, 2024 · 5f038c5 · 5f038c5
1 parent 26d05ce
commit 5f038c5
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/nvd/test_api.py b/tests/nvd/test_api.py
@@ -186,6 +186,29 @@ async def test_items(self):
         with self.assertRaises(StopAsyncIteration):
             await anext(it)
 
+    async def test_items_no_results(self):
+        response_mock = MagicMock(spec=Response)
+        response_mock.json.side_effect = [
+            {
+                "values": [],
+                "total_results": 0,
+                "results_per_page": 0,
+            },
+        ]
+        api_mock = AsyncMock(spec=NVDApi)
+        api_mock._get.return_value = response_mock
+
+        results: NVDResults[Result] = NVDResults(
+            api_mock,
+            {},
+            result_func,
+        )
+
+        it = aiter(results.items())
+
+        with self.assertRaises(StopAsyncIteration):
+            await anext(it)
+
     async def test_aiter(self):
         response_mock = MagicMock(spec=Response)
         response_mock.json.side_effect = [