wren
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
import asyncio
|
|
from unittest.mock import patch
|
|
|
|
from fastapi_modules.fastapi_leaudit.services.impl.contractTemplateServiceImpl import ContractTemplateServiceImpl
|
|
|
|
|
|
class _EmptyMappingResult:
|
|
def mappings(self):
|
|
return self
|
|
|
|
def all(self):
|
|
return []
|
|
|
|
|
|
class _FakeSession:
|
|
def __init__(self):
|
|
self.executed_sql = None
|
|
self.executed_params = None
|
|
|
|
async def execute(self, sql, params=None):
|
|
self.executed_sql = sql
|
|
self.executed_params = params
|
|
return _EmptyMappingResult()
|
|
|
|
|
|
class _FakeSessionContext:
|
|
def __init__(self, session):
|
|
self.session = session
|
|
|
|
async def __aenter__(self):
|
|
return self.session
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
return None
|
|
|
|
|
|
def test_contract_template_search_category_stats_binds_expanding_scope_params():
|
|
service = ContractTemplateServiceImpl()
|
|
fake_session = _FakeSession()
|
|
|
|
async def noop_ensure_schema(session):
|
|
return None
|
|
|
|
async def fake_user_context(current_user_id, session):
|
|
return {
|
|
"id": current_user_id,
|
|
"area": "梅州",
|
|
"tenant_code": "MZ",
|
|
"tenant_name": "梅州",
|
|
"tenant_scope_value": "梅州",
|
|
"is_global": False,
|
|
"can_manage": True,
|
|
"is_area_admin": True,
|
|
}
|
|
|
|
service._ensureContractTemplateSchema = noop_ensure_schema
|
|
service._getCurrentUserContext = fake_user_context
|
|
|
|
with patch(
|
|
"fastapi_modules.fastapi_leaudit.services.impl.contractTemplateServiceImpl.GetAsyncSession",
|
|
return_value=_FakeSessionContext(fake_session),
|
|
):
|
|
asyncio.run(service._load_search_category_stats("买卖", None, None, 5))
|
|
|
|
assert fake_session.executed_sql._bindparams["visible_tenant_codes"].expanding is True
|
|
assert fake_session.executed_sql._bindparams["visible_regions"].expanding is True
|
|
assert fake_session.executed_params["visible_tenant_codes"] == ["PROVINCIAL", "PUBLIC", "MZ"]
|
|
assert fake_session.executed_params["visible_regions"] == ["省级", "公共", "梅州"]
|