summaryrefslogtreecommitdiff
path: root/tests/test_info.py
blob: ef775d7d3d5f1043e9973c4fc2e09fef414b60e9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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)