From 96fc7da5ae3ef4b71af428e8888cca1ee9308580 Mon Sep 17 00:00:00 2001
From: david grossman <dgrossman@vulncheck.com>
Date: Fri, 16 Jan 2026 15:39:17 -0500
Subject: [PATCH] task: create asyncio integration tests

---
 tests/integration_aio.py      | 155 ++++++++++++++++++++++++++++++++++
 tests/readme_examples_test.py |   1 +
 2 files changed, 156 insertions(+)
 create mode 100644 tests/integration_aio.py

diff --git a/tests/integration_aio.py b/tests/integration_aio.py
new file mode 100644
index 00000000..a1092b91
--- /dev/null
+++ b/tests/integration_aio.py
@@ -0,0 +1,155 @@
+import os
+import asyncio
+import vulncheck_sdk.aio as vcaio
+from vulncheck_sdk.aio.api.endpoints_api import EndpointsApi
+from vulncheck_sdk.aio.api.indices_api import IndicesApi
+from vulncheck_sdk.aio.exceptions import ApiException, UnauthorizedException
+
+DEFAULT_HOST = "https://api.vulncheck.com"
+DEFAULT_API = DEFAULT_HOST + "/v3"
+API_TOKEN = os.environ.get("VULNCHECK_API_TOKEN", "")
+
+# --- Helpers ---
+
+
+async def _get_http_status(func, *args) -> int:
+    """Awaits the SDK call and returns the status code, handling common exceptions."""
+    try:
+        api_response = await func(*args)
+        return api_response.status_code
+    except UnauthorizedException as e:
+        print(f"Not authorized for this endpoint: {e}")
+        return 401
+    except ApiException as e:
+        status = getattr(e, "status", 500)
+        print(f"ApiException (Status {status}) when calling {func.__name__}: {e}")
+        return status
+    except Exception as e:
+        print(f"Unexpected error in {func.__name__}: {e}")
+        return 500
+
+
+# --- EndpointsApi Tests ---
+
+
+async def test_openapi_get(api: EndpointsApi):
+    print(f"Testing test_openapi_get")
+    status = await _get_http_status(api.openapi_get_with_http_info)
+    assert status == 200
+
+
+async def test_entitlements_get(api: EndpointsApi):
+    print(f"Testing test_entitlements_get")
+    status = await _get_http_status(api.entitlements_get_with_http_info)
+    assert status == 200
+
+
+async def test_purl_get(api: EndpointsApi):
+    print(f"Testing test_purl_get")
+    status = await _get_http_status(
+        api.purl_get_with_http_info, "pkg:hex/coherence@0.1.2"
+    )
+    assert status == 200
+
+
+async def test_index_get(api: EndpointsApi):
+    print(f"Testing test_index_get")
+    status = await _get_http_status(api.index_get_with_http_info)
+    assert status == 200
+
+
+async def test_cpe_get(api: EndpointsApi):
+    print(f"Testing test_cpe_get")
+    status = await _get_http_status(
+        api.cpe_get_with_http_info, "cpe:/a:microsoft:internet_explorer:8.0.6001:beta"
+    )
+    assert status == 200
+
+
+async def test_backup_get(api: EndpointsApi):
+    print(f"Testing test_backup_get")
+    status = await _get_http_status(api.backup_get_with_http_info)
+    assert status == 200
+
+
+async def test_backup_index_get(api: EndpointsApi):
+    print(f"Testing test_backup_index_get")
+    status = await _get_http_status(api.backup_index_get_with_http_info, "")
+    assert status == 200
+
+
+async def test_pdns_vulncheck_c2_get(api: EndpointsApi):
+    print(f"Testing test_pdns_vulncheck_c2_get")
+    status = await _get_http_status(api.pdns_vulncheck_c2_get_with_http_info, "")
+    assert status == 200
+
+
+async def test_rules_initial_access_type_get(api: EndpointsApi):
+    print(f"Testing test_rules_initial_access_type_get")
+    status = await _get_http_status(
+        api.rules_initial_access_type_get_with_http_info, "suricata"
+    )
+    assert status == 200
+
+
+# --- IndicesApi Tests ---
+
+
+async def test_all_indices(indices: IndicesApi):
+    targets = [
+        "index_nist_nvd2_get_with_http_info",
+        "index_vulncheck_kev_get_with_http_info",
+        "index_ipintel3d_get_with_http_info",
+        "index_cisa_kev_get_with_http_info",
+    ]
+
+    for name in targets:
+        method = getattr(indices, name)
+        # Passing strings for limit/page to satisfy Pydantic validation
+        status = await _get_http_status(method, 1, 1)
+        print(f"Testing {name}: Result {status}")
+        assert status == 200
+
+
+# --- Orchestrator ---
+
+
+async def main():
+    if not API_TOKEN:
+        print("Warning: VULNCHECK_API_TOKEN is not set.")
+
+    config = vcaio.Configuration(host=DEFAULT_API)
+    config.api_key["Bearer"] = API_TOKEN
+
+    # The 'async with' ensures the ClientSession is closed
+    async with vcaio.ApiClient(config) as client:
+        endpoints_api = vcaio.EndpointsApi(client)
+        indices_api = vcaio.IndicesApi(client)
+
+        print("--- Running Tests ---")
+        asyncio.gather(
+            test_openapi_get(endpoints_api),
+            test_entitlements_get(endpoints_api),
+            test_purl_get(endpoints_api),
+            test_index_get(endpoints_api),
+            test_cpe_get(endpoints_api),
+            test_backup_get(endpoints_api),
+            test_backup_index_get(endpoints_api),
+            test_pdns_vulncheck_c2_get(endpoints_api),
+            test_rules_initial_access_type_get(endpoints_api),
+        )
+
+        await test_all_indices(indices_api)
+
+    # --- CRITICAL FIX FOR "UNCLOSED CONNECTOR" ---
+    # aiohttp requires a small window of time to allow the
+    # underlying SSL/TCP transports to finish closing.
+    await asyncio.sleep(0.250)
+    print("--- All Tests Completed Successfully ---")
+
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        pass
diff --git a/tests/readme_examples_test.py b/tests/readme_examples_test.py
index d922cce0..a77a3445 100644
--- a/tests/readme_examples_test.py
+++ b/tests/readme_examples_test.py
@@ -29,6 +29,7 @@
     "./tests/purl_aio.py",
     "./tests/quickstart.py",
     "./tests/quickstart_aio.py",
+    "./tests/integration_aio.py"
 ]