ismijnipverweg/tests/test_iplookup.py

'''
Test ismijnverweg geolookup api
'''
import logging
import random
import re
from operator import itemgetter

from faker import Faker
from fastapi.testclient import TestClient

from main import app  # type: ignore

# Set up logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Initialize Faker for generating test data
fake = Faker()

# Create test client
fake_ipv6 = fake.ipv6()
client = TestClient(app, client=(fake_ipv6, 31337))


def test_no_query():
    """Test searching without a query parameter"""

    response = client.get("/")

    assert response.status_code == 200
    results = response.json()
    logging.info(results)
    assert results['ip'] == fake_ipv6
    assert len(results) > 0


def test_single_query():
    """Test searching with an ip address"""
    fake_ipv4 = fake.ipv4_public()

    response = client.get(f"/{fake_ipv4}")

    assert response.status_code == 200
    results = response.json()
    logging.info(results)
    assert results['ip'] == fake_ipv4
    assert len(results) > 0


def test_multi_query():
    """Test searching with an ip address"""
    fake_ips = [{'ip': fake.ipv6() if random.random() > 0.5 else fake.ipv4()}
                for _ in range(16)]

    response = client.post("/", json=fake_ips)

    assert response.status_code == 200
    results = response.json()
    logging.info(results)

    for ip in map(itemgetter('ip'), results):
        assert ip in map(itemgetter('ip'), fake_ips)

    assert len(results) > 0


def test_invalid_query():
    """Test searching with an invalid ip address"""
    invalid_ip = '500.312.77.31337'
    test_pattern = 'Input is not a valid IPv[46] address'

    response = client.get(f"/{invalid_ip}")

    assert response.status_code == 422
    results = response.json()
    logging.info(results)
    assert all(map(lambda x: x == invalid_ip, (
        map(itemgetter('input'), results['detail']))))
    assert all(map(lambda x: re.match(test_pattern, x), (
        map(itemgetter('msg'), results['detail']))))
    assert len(results) > 0


if __name__ == "__main__":
    # Run tests
    test_no_query()
    test_single_query()
    test_invalid_query()
    test_multi_query()
    print("All tests passed!")