qzl
116 lines
3.5 KiB
Python
116 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
from typing import Any, cast
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
|
|
from models.automation_jobs import ScheduleType
|
|
from v1.auth.registration_bootstrap import (
|
|
compute_next_local_time_utc,
|
|
)
|
|
|
|
|
|
def test_compute_next_local_time_utc_from_asia_shanghai() -> None:
|
|
now_utc = datetime(2026, 3, 23, 0, 30, tzinfo=timezone.utc)
|
|
|
|
run_at, next_run_at = compute_next_local_time_utc(
|
|
now_utc=now_utc,
|
|
timezone_name="Asia/Shanghai",
|
|
local_hour=8,
|
|
local_minute=0,
|
|
schedule_type=ScheduleType.DAILY,
|
|
)
|
|
|
|
assert run_at == datetime(2026, 3, 24, 0, 0, tzinfo=timezone.utc)
|
|
assert next_run_at == datetime(2026, 3, 25, 0, 0, tzinfo=timezone.utc)
|
|
|
|
|
|
def test_compute_next_local_time_utc_rolls_to_next_day_when_passed() -> None:
|
|
now_utc = datetime(2026, 3, 23, 2, 30, tzinfo=timezone.utc)
|
|
|
|
run_at, next_run_at = compute_next_local_time_utc(
|
|
now_utc=now_utc,
|
|
timezone_name="Asia/Shanghai",
|
|
local_hour=8,
|
|
local_minute=0,
|
|
schedule_type=ScheduleType.DAILY,
|
|
)
|
|
|
|
assert run_at == datetime(2026, 3, 24, 0, 0, tzinfo=timezone.utc)
|
|
assert next_run_at == datetime(2026, 3, 25, 0, 0, tzinfo=timezone.utc)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_registration_service_is_idempotent_when_job_exists() -> None:
|
|
from v1.auth.registration_bootstrap import RegistrationAutomationBootstrapService
|
|
|
|
expected_owner_id = uuid4()
|
|
|
|
class _Repo:
|
|
inserted = 0
|
|
upsert_calls = 0
|
|
|
|
async def get_profile_timezone(self, *, user_id):
|
|
assert user_id == expected_owner_id
|
|
return "Asia/Shanghai"
|
|
|
|
async def insert_bootstrap_automation_job_if_absent(self, **kwargs):
|
|
assert kwargs["owner_id"] == expected_owner_id
|
|
assert kwargs["bootstrap_key"] == "memory_extraction"
|
|
self.inserted += 1
|
|
return False
|
|
|
|
async def upsert_initial_memory(self, **kwargs):
|
|
self.upsert_calls += 1
|
|
return False
|
|
|
|
class _Session:
|
|
async def commit(self):
|
|
raise AssertionError("must not commit when already exists")
|
|
|
|
async def rollback(self):
|
|
raise AssertionError("must not rollback when no error")
|
|
|
|
service = RegistrationAutomationBootstrapService(
|
|
repository=cast(Any, _Repo()), session=cast(Any, _Session())
|
|
)
|
|
await service.ensure_user_automation_jobs(user_id=str(expected_owner_id))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_registration_service_creates_initial_memories_when_missing() -> None:
|
|
from v1.auth.registration_bootstrap import RegistrationAutomationBootstrapService
|
|
|
|
expected_owner_id = uuid4()
|
|
|
|
class _Repo:
|
|
async def get_profile_timezone(self, *, user_id):
|
|
assert user_id == expected_owner_id
|
|
return "Asia/Shanghai"
|
|
|
|
async def upsert_initial_memory(self, **kwargs):
|
|
return True
|
|
|
|
async def insert_bootstrap_automation_job_if_absent(self, **kwargs):
|
|
_ = kwargs
|
|
return True
|
|
|
|
class _Session:
|
|
committed = 0
|
|
|
|
async def commit(self):
|
|
self.committed += 1
|
|
|
|
async def rollback(self):
|
|
raise AssertionError("must not rollback when no error")
|
|
|
|
session = _Session()
|
|
service = RegistrationAutomationBootstrapService(
|
|
repository=cast(Any, _Repo()), session=cast(Any, session)
|
|
)
|
|
await service.ensure_user_automation_jobs(user_id=str(expected_owner_id))
|
|
|
|
assert session.committed == 1
|