156 lines
4.9 KiB
Python
156 lines
4.9 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import pytest
|
|
|
|
from banban.dao.binding import SESSION_STATUS_COMPLETED, SESSION_STATUS_EXPIRED
|
|
from banban.service.binding import BindingService
|
|
|
|
|
|
class FakeSession:
|
|
def __init__(self):
|
|
self.commits = 0
|
|
self.rollbacks = 0
|
|
self.closed = False
|
|
|
|
async def commit(self):
|
|
self.commits += 1
|
|
|
|
async def rollback(self):
|
|
self.rollbacks += 1
|
|
|
|
async def close(self):
|
|
self.closed = True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_finalize_nfc_bind_expires_late_card_without_binding(monkeypatch):
|
|
service = BindingService()
|
|
db_session = FakeSession()
|
|
expired_session = {
|
|
"id": 42,
|
|
"bind_token": "bind-token-expired",
|
|
"device_id": "TalkingQ_device001",
|
|
"target_child_id": 7,
|
|
"initiator_user_id": 99,
|
|
"expires_at": datetime.utcnow() - timedelta(seconds=1),
|
|
}
|
|
marked_statuses = []
|
|
activated_cards = []
|
|
completed_sessions = []
|
|
|
|
async def fake_get_session():
|
|
return db_session
|
|
|
|
async def fake_get_latest_pending_session_by_device(self, device_id):
|
|
assert device_id == "TalkingQ_device001"
|
|
return expired_session
|
|
|
|
async def fake_mark_session_status(self, session_id, status):
|
|
marked_statuses.append((session_id, status))
|
|
|
|
async def fake_activate_card(**kwargs):
|
|
activated_cards.append(kwargs)
|
|
|
|
async def fake_complete_nfc_bind(self, **kwargs):
|
|
completed_sessions.append(kwargs)
|
|
|
|
monkeypatch.setattr(service, "get_session", fake_get_session)
|
|
monkeypatch.setattr(
|
|
"banban.service.binding.BindingDAO.get_latest_pending_session_by_device",
|
|
fake_get_latest_pending_session_by_device,
|
|
)
|
|
monkeypatch.setattr("banban.service.binding.BindingDAO.mark_session_status", fake_mark_session_status)
|
|
monkeypatch.setattr("banban.service.binding.card_service.activate_card", fake_activate_card)
|
|
monkeypatch.setattr("banban.service.binding.BindingDAO.complete_nfc_bind", fake_complete_nfc_bind)
|
|
|
|
result = await service.finalize_nfc_bind(
|
|
device_id="TalkingQ_device001",
|
|
card_uuid="CARD_UUID_LATE",
|
|
)
|
|
|
|
assert result == {
|
|
"device_id": "TalkingQ_device001",
|
|
"bind_token": "bind-token-expired",
|
|
"status": SESSION_STATUS_EXPIRED,
|
|
}
|
|
assert marked_statuses == [(42, SESSION_STATUS_EXPIRED)]
|
|
assert activated_cards == []
|
|
assert completed_sessions == []
|
|
assert db_session.commits == 1
|
|
assert db_session.rollbacks == 0
|
|
assert db_session.closed is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_finalize_nfc_bind_completes_before_expiry(monkeypatch):
|
|
service = BindingService()
|
|
db_session = FakeSession()
|
|
pending_session = {
|
|
"id": 43,
|
|
"bind_token": "bind-token-active",
|
|
"device_id": "TalkingQ_device001",
|
|
"target_child_id": 7,
|
|
"initiator_user_id": 99,
|
|
"expires_at": datetime.utcnow() + timedelta(minutes=10),
|
|
}
|
|
marked_statuses = []
|
|
activated_cards = []
|
|
completed_sessions = []
|
|
|
|
async def fake_get_session():
|
|
return db_session
|
|
|
|
async def fake_get_latest_pending_session_by_device(self, device_id):
|
|
assert device_id == "TalkingQ_device001"
|
|
return pending_session
|
|
|
|
async def fake_mark_session_status(self, session_id, status):
|
|
marked_statuses.append((session_id, status))
|
|
|
|
async def fake_activate_card(**kwargs):
|
|
activated_cards.append(kwargs)
|
|
|
|
async def fake_complete_nfc_bind(self, **kwargs):
|
|
completed_sessions.append(kwargs)
|
|
|
|
monkeypatch.setattr(service, "get_session", fake_get_session)
|
|
monkeypatch.setattr(
|
|
"banban.service.binding.BindingDAO.get_latest_pending_session_by_device",
|
|
fake_get_latest_pending_session_by_device,
|
|
)
|
|
monkeypatch.setattr("banban.service.binding.BindingDAO.mark_session_status", fake_mark_session_status)
|
|
monkeypatch.setattr("banban.service.binding.card_service.activate_card", fake_activate_card)
|
|
monkeypatch.setattr("banban.service.binding.BindingDAO.complete_nfc_bind", fake_complete_nfc_bind)
|
|
|
|
result = await service.finalize_nfc_bind(
|
|
device_id="TalkingQ_device001",
|
|
card_uuid="CARD_UUID_ON_TIME",
|
|
)
|
|
|
|
assert result == {
|
|
"device_id": "TalkingQ_device001",
|
|
"bind_token": "bind-token-active",
|
|
"status": SESSION_STATUS_COMPLETED,
|
|
"child_id": 7,
|
|
"card_uuid": "CARD_UUID_ON_TIME",
|
|
}
|
|
assert marked_statuses == []
|
|
assert activated_cards == [
|
|
{
|
|
"card_uuid": "CARD_UUID_ON_TIME",
|
|
"device_id": "TalkingQ_device001",
|
|
"db_session": db_session,
|
|
}
|
|
]
|
|
assert completed_sessions == [
|
|
{
|
|
"session_id": 43,
|
|
"device_id": "TalkingQ_device001",
|
|
"child_id": 7,
|
|
"user_id": 99,
|
|
}
|
|
]
|
|
assert db_session.commits == 1
|
|
assert db_session.rollbacks == 0
|
|
assert db_session.closed is True
|