17326615b7
Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
27 lines
689 B
Python
27 lines
689 B
Python
"""Test GeoIP Wrapper"""
|
|
from django.test import TestCase
|
|
|
|
from authentik.events.geo import GeoIPReader
|
|
|
|
|
|
class TestGeoIP(TestCase):
|
|
"""Test GeoIP Wrapper"""
|
|
|
|
def setUp(self) -> None:
|
|
self.reader = GeoIPReader()
|
|
|
|
def test_simple(self):
|
|
"""Test simple city wrapper"""
|
|
# IPs from
|
|
# https://github.com/maxmind/MaxMind-DB/blob/main/source-data/GeoLite2-City-Test.json
|
|
self.assertEqual(
|
|
self.reader.city_dict("2.125.160.216"),
|
|
{
|
|
"city": "Boxford",
|
|
"continent": "EU",
|
|
"country": "GB",
|
|
"lat": 51.75,
|
|
"long": -1.25,
|
|
},
|
|
)
|