mirror of
https://github.com/rommapp/romm.git
synced 2026-04-23 06:54:40 +00:00
Georges-Antoine Assi
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
from dataclasses import replace
|
|
from unittest.mock import MagicMock
|
|
|
|
from ua_parser import parse as parse_ua
|
|
|
|
from handler.database import db_device_handler
|
|
from models.user import User
|
|
from utils.auth import _get_device_name, create_or_find_web_device
|
|
|
|
CHROME_MAC_UA = (
|
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
|
|
)
|
|
|
|
|
|
def _make_request(user_agent: str = CHROME_MAC_UA, forwarded_for: str = "1.2.3.4"):
|
|
request = MagicMock()
|
|
request.headers = {"user-agent": user_agent, "x-forwarded-for": forwarded_for}
|
|
request.client.host = "127.0.0.1"
|
|
return request
|
|
|
|
|
|
class TestGetDeviceName:
|
|
def test_browser_and_os(self):
|
|
result = parse_ua(CHROME_MAC_UA)
|
|
assert _get_device_name(result) == "Chrome on Mac OS X"
|
|
|
|
def test_browser_only(self):
|
|
result = replace(parse_ua(CHROME_MAC_UA), os=None)
|
|
assert _get_device_name(result) == "Chrome"
|
|
|
|
def test_os_only(self):
|
|
result = replace(parse_ua(CHROME_MAC_UA), user_agent=None)
|
|
assert _get_device_name(result) == "Mac OS X"
|
|
|
|
def test_neither(self):
|
|
result = replace(parse_ua(CHROME_MAC_UA), user_agent=None, os=None)
|
|
assert _get_device_name(result) == "Web Browser"
|
|
|
|
|
|
class TestCreateOrFindWebDevice:
|
|
def test_creates_new_device(self, admin_user: User):
|
|
request = _make_request()
|
|
device = create_or_find_web_device(request, admin_user)
|
|
|
|
assert device.id is not None
|
|
assert len(device.id) == 36
|
|
assert device.user_id == admin_user.id
|
|
assert device.name == "Chrome on Mac OS X"
|
|
assert device.platform == "Web"
|
|
assert device.client == "web"
|
|
assert device.ip_address == "127.0.0.1"
|
|
assert device.last_seen is not None
|
|
|
|
def test_returns_existing_device_on_matching_fingerprint(self, admin_user: User):
|
|
request = _make_request()
|
|
first = create_or_find_web_device(request, admin_user)
|
|
second = create_or_find_web_device(request, admin_user)
|
|
|
|
assert second.id == first.id
|
|
|
|
def test_updates_last_seen_on_existing_device(self, admin_user: User):
|
|
request = _make_request()
|
|
first = create_or_find_web_device(request, admin_user)
|
|
|
|
second = create_or_find_web_device(request, admin_user)
|
|
refreshed = db_device_handler.get_device(
|
|
device_id=second.id, user_id=admin_user.id
|
|
)
|
|
|
|
assert refreshed is not None
|
|
assert refreshed.last_seen is not None
|
|
assert first.id == refreshed.id
|