Add: Another unit test

greenbone · Mar 6, 2025 · 6dee935 · 6dee935
1 parent 1f718d8
commit 6dee935
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/nvd/source/test_api.py b/tests/nvd/source/test_api.py
@@ -183,3 +183,10 @@ async def test_sources_source_identifier(self):
 
         with self.assertRaises(StopAsyncIteration):
             await anext(it)
+
+    async def test_context_manager(self):
+        async with self.api:
+            pass
+
+        self.http_client.__aenter__.assert_awaited_once()
+        self.http_client.__aexit__.assert_awaited_once()