Vladless
50 lines
2.3 KiB
Python
50 lines
2.3 KiB
Python
import unittest
|
|
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from database.access.resolution import (
|
|
ActorSurface,
|
|
resolve_actor_from_identity,
|
|
resolve_actor_from_legacy_ref,
|
|
)
|
|
|
|
|
|
class ActorScenarioTests(unittest.IsolatedAsyncioTestCase):
|
|
async def test_tg_only_user_detected_as_telegram_surface(self):
|
|
session = object()
|
|
user = SimpleNamespace(id=1, tg_id=1001, identity_id="ident-tg")
|
|
with patch("database.access.resolution.resolve_user_optional", new=AsyncMock(return_value=user)):
|
|
actor = await resolve_actor_from_legacy_ref(session, 1001)
|
|
self.assertEqual(actor.surface, ActorSurface.TELEGRAM)
|
|
self.assertEqual(actor.billing_user_id, 1)
|
|
self.assertEqual(actor.telegram_chat_id, 1001)
|
|
|
|
async def test_web_only_user_detected_as_web_surface(self):
|
|
session = object()
|
|
user = SimpleNamespace(id=2, tg_id=None, identity_id="ident-web")
|
|
with patch("database.access.resolution.resolve_user_optional", new=AsyncMock(return_value=user)):
|
|
actor = await resolve_actor_from_legacy_ref(session, 2)
|
|
self.assertEqual(actor.surface, ActorSurface.WEB)
|
|
self.assertEqual(actor.billing_user_id, 2)
|
|
self.assertIsNone(actor.telegram_chat_id)
|
|
|
|
async def test_linked_user_is_web_by_identity_and_telegram_by_tg(self):
|
|
session = object()
|
|
identity = SimpleNamespace(id="ident-linked")
|
|
linked_user = SimpleNamespace(id=3, tg_id=1003, identity_id="ident-linked")
|
|
|
|
with patch("database.identities.ensure_billing_user_for_identity", new=AsyncMock(return_value=3)):
|
|
with patch("database.access.resolution.resolve_user_optional", new=AsyncMock(return_value=linked_user)):
|
|
web_actor = await resolve_actor_from_identity(session, identity)
|
|
|
|
with patch("database.access.resolution.resolve_user_optional", new=AsyncMock(return_value=linked_user)):
|
|
tg_actor = await resolve_actor_from_legacy_ref(session, 1003)
|
|
|
|
self.assertEqual(web_actor.surface, ActorSurface.WEB)
|
|
self.assertEqual(web_actor.billing_user_id, 3)
|
|
self.assertEqual(web_actor.telegram_chat_id, 1003)
|
|
self.assertEqual(tg_actor.surface, ActorSurface.TELEGRAM)
|
|
self.assertEqual(tg_actor.billing_user_id, 3)
|
|
self.assertEqual(tg_actor.telegram_chat_id, 1003)
|