added tests

1 files changed, 37 insertions, 0 deletions

diff --git a/tests/test_info.py b/tests/test_info.py
new file mode 100644
index 0000000..ef775d7
--- /dev/null
+++ b/tests/test_info.py
@@ -0,0 +1,37 @@
+"""
+Test our info
+"""
+import unittest
+import json
+
+import requests
+
+from src.soc_collector.auth import load_api_keys
+
+BASE_URL = "https://localhost:8000"
+
+
+class TestAddress(unittest.TestCase):
+    """
+    Test our info
+    """
+
+    def test_info(self) -> None:
+        """
+        Test info
+        """
+
+        api_keys = load_api_keys("data/api_keys.txt")
+        request_headers = {"API-KEY": api_keys[-1]}
+
+        req = requests.get(
+            f"{BASE_URL}/info",
+            headers=request_headers,
+            timeout=4,
+            verify="./data/collector_root_ca.crt",
+        )
+        self.assertTrue(req.status_code == 200)
+        data = json.loads(req.text)
+        self.assertTrue("Estimated document count" in data)
+        self.assertTrue(isinstance(data["Estimated document count"], int))
+        self.assertTrue(data["Estimated document count"] >= 0)