249 lines
13 KiB
Python
249 lines
13 KiB
Python
"""统计只读、主对象分页与授权明细聚合回归。"""
|
|
from datetime import date
|
|
from types import SimpleNamespace
|
|
import importlib
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
from app.models.crm_models import CrmContract, CrmContractProduct
|
|
from app.models.pl_models import PlPlan
|
|
from app.models import pl_quota_models as m
|
|
|
|
|
|
@pytest.fixture
|
|
def case():
|
|
engine = create_engine('sqlite://')
|
|
models = (CrmContract, CrmContractProduct, PlPlan, m.PlPurchaseCert, m.PlTransportCert,
|
|
m.PlContractCertLink, m.PlContractCertProduct, m.PlPlanCertUsage,
|
|
m.PlPlanCertUsageItem, m.PlCertQtyLog)
|
|
for model in models:
|
|
model.__table__.create(engine)
|
|
db = Session(engine)
|
|
user = SimpleNamespace(tenant_no='t', to_user_info=lambda: {'role': 'super_admin', 'user_id': 'u'})
|
|
for cid in ('a', 'b'):
|
|
db.add(CrmContract(id=cid, contract_no=cid, contract_name=cid, customer_id='customer',
|
|
contract_type='sale', status='approved', tenant_no='t'))
|
|
db.add(CrmContractProduct(id=cid, contract_id=cid, product_id='chemical', product_name=cid,
|
|
unit='吨', quantity=100, tenant_no='t'))
|
|
db.add(m.PlContractCertLink(id=cid, contract_id=cid, contract_no_snapshot=cid,
|
|
cert_type='purchase', cert_id='same', tenant_no='t'))
|
|
db.add(m.PlContractCertProduct(id=cid, contract_cert_link_id=cid,
|
|
contract_product_id=cid, tenant_no='t'))
|
|
db.add(m.PlPurchaseCert(id='same', cert_no='P', product_name_on_cert='硫酸', total_qty=300,
|
|
quantity_unit='吨', reserved_qty=180, used_qty=20, valid_use_text='多次有效',
|
|
buyer_name='买方', seller_name='卖方', tenant_no='t'))
|
|
db.commit()
|
|
yield SimpleNamespace(db=db, user=user)
|
|
db.close()
|
|
engine.dispose()
|
|
|
|
|
|
def service():
|
|
assert importlib.util.find_spec('app.services.pl_quota_statistics'), '缺少统计服务'
|
|
return importlib.import_module('app.services.pl_quota_statistics')
|
|
|
|
|
|
def add_usage(c, cid, ident, reserved, used=0):
|
|
c.db.add(PlPlan(id=ident, plan_no=ident, create_date=date.today(), belongs_company='shanxia', plan_type='direct', tenant_no='t'))
|
|
c.db.add(m.PlPlanCertUsage(id=ident, plan_id=ident, plan_no_snapshot=ident, cert_type='purchase',
|
|
cert_id='same', cert_no_snapshot='P', cert_revision=1, quantity_unit='吨', tenant_no='t'))
|
|
c.db.add(m.PlPlanCertUsageItem(id=ident, usage_id=ident, contract_cert_link_id=cid,
|
|
contract_cert_product_id=cid, contract_scope_revision=1, contract_id=cid,
|
|
contract_product_id=cid, plan_product_id=ident, source_stop_type='delivery',
|
|
source_stop_id=ident, source_stop_product_id=ident, allocation_seq=1, source_unit='吨',
|
|
planned_source_qty=999, reserved_qty=reserved, used_qty=used, tenant_no='t'))
|
|
c.db.commit()
|
|
|
|
|
|
def test_shared_balance_and_contract_allocation_are_distinct_and_read_only(case):
|
|
s = service()
|
|
add_usage(case, 'a', 'one', 60, 20)
|
|
add_usage(case, 'a', 'two', 40)
|
|
add_usage(case, 'b', 'three', 80)
|
|
row = s.contract_statistics(case.db, case.user, 'a')['certificates'][0]
|
|
assert (row['totalQty'], row['reservedQty'], row['usedQty'], row['availableQty']) == (300, 180, 20, 100)
|
|
assert row['contractReservedQty'] == 100 and row['contractUsedQty'] == 20
|
|
assert row['products'][0]['reservedQty'] == 100
|
|
detail = s.certificate_statistics(case.db, case.user, 'purchase', 'same')
|
|
assert len(detail['usageItems']) == 3
|
|
assert detail['openingUsedQty'] == 0
|
|
assert case.db.query(m.PlContractCertLink).count() == 2
|
|
assert not case.db.new and not case.db.dirty
|
|
|
|
|
|
def test_lists_paginate_main_objects_and_filter_keywords(case):
|
|
s = service()
|
|
result = s.list_contracts(case.db, case.user, page=1, page_size=1)
|
|
assert result['total'] == 2 and len(result['list']) == 1 and result['pageSize'] == 1
|
|
assert s.list_contracts(case.db, case.user, keyword='a')['total'] == 1
|
|
assert s.list_certificates(case.db, case.user, keyword='硫酸')['total'] == 1
|
|
other = SimpleNamespace(**{**case.user.__dict__, 'tenant_no': 'other'})
|
|
assert s.list_contracts(case.db, other)['total'] == 0
|
|
assert s.list_certificates(case.db, other)['total'] == 0
|
|
|
|
|
|
def test_hidden_contracts_plans_and_deleted_products_never_enter_details(case, monkeypatch):
|
|
s = service()
|
|
add_usage(case, 'a', 'visible', 30)
|
|
add_usage(case, 'a', 'hidden-plan', 70)
|
|
add_usage(case, 'b', 'hidden-contract', 80)
|
|
monkeypatch.setattr(s, 'apply_crm_owner_scope', lambda q, model, *a: q.filter(model.id == 'a'))
|
|
monkeypatch.setattr(s, 'apply_pl_scope', lambda q, model, *a: q.filter(model.id != 'hidden-plan'))
|
|
detail = s.certificate_statistics(case.db, case.user, 'purchase', 'same')
|
|
assert [x['id'] for x in detail['contracts']] == ['a']
|
|
assert detail['contracts'][0]['contractReservedQty'] == 30
|
|
assert [x['planId'] for x in detail['usageItems']] == ['visible']
|
|
assert detail['openingUsedQty'] == 0 and detail['unattributedUsedQty'] == 0
|
|
case.db.query(CrmContractProduct).filter_by(id='a').one().del_flag = 1
|
|
case.db.commit()
|
|
assert s.certificate_statistics(case.db, case.user, 'purchase', 'same')['usageItems'] == []
|
|
|
|
|
|
def test_same_id_certificate_types_units_and_real_opening_are_separate(case):
|
|
s = service()
|
|
# 按当前模型构造运输证，验证同 ID 两个额度池不会在分页或汇总中合并。
|
|
purchase = case.db.query(m.PlPurchaseCert).one()
|
|
values = {col.name: getattr(purchase, col.name) for col in m.PlTransportCert.__table__.columns
|
|
if hasattr(purchase, col.name)}
|
|
values.update(cert_no='T', quantity_unit='kg', total_qty=500, reserved_qty=0, used_qty=0)
|
|
for col in m.PlTransportCert.__table__.columns:
|
|
if not col.nullable and col.default is None and col.name not in values:
|
|
values[col.name] = '运输主体'
|
|
case.db.add(m.PlTransportCert(**values))
|
|
for ident, event, amount, plan, tenant, deleted in (
|
|
('opening', 'opening', 7, None, 't', 0), ('adjust', 'adjust', 2, None, 't', 0),
|
|
('hidden', 'settle', 99, 'hidden-plan', 't', 0),
|
|
('other', 'opening', 99, None, 'other', 0), ('deleted', 'opening', 99, None, 't', 1)):
|
|
case.db.add(m.PlCertQtyLog(id=ident, operation_id=ident, request_hash='hash', cert_type='purchase',
|
|
cert_id='same', plan_id=plan, event_type=event, quantity_unit='吨', used_delta=amount,
|
|
before_total_qty=300, after_total_qty=300, before_reserved_qty=0, after_reserved_qty=0,
|
|
before_used_qty=0, after_used_qty=amount, reason='凭据', tenant_no=tenant, del_flag=deleted))
|
|
case.db.commit()
|
|
result = s.list_certificates(case.db, case.user, page_size=1)
|
|
assert result['total'] == 2 and len(result['list']) == 1
|
|
assert s.list_certificates(case.db, case.user, cert_type='transport')['list'][0]['quantityUnit'] == 'kg'
|
|
purchase = s.certificate_statistics(case.db, case.user, 'purchase', 'same')
|
|
transport = s.certificate_statistics(case.db, case.user, 'transport', 'same')
|
|
assert purchase['openingUsedQty'] == 7 and purchase['unattributedUsedQty'] == 2
|
|
assert transport['openingUsedQty'] == 0 and transport['contracts'] == []
|
|
|
|
|
|
@pytest.mark.parametrize('model,field,value', [
|
|
(m.PlPlanCertUsage, 'tenant_no', 'other'), (m.PlPlanCertUsage, 'del_flag', 1),
|
|
(m.PlPlanCertUsageItem, 'tenant_no', 'other'), (m.PlPlanCertUsageItem, 'del_flag', 1),
|
|
(PlPlan, 'tenant_no', 'other'), (PlPlan, 'del_flag', 1),
|
|
(CrmContract, 'del_flag', 1), (CrmContractProduct, 'tenant_no', 'other')])
|
|
def test_invalid_ancestors_exclude_usage(case, model, field, value):
|
|
s = service()
|
|
add_usage(case, 'a', 'one', 100)
|
|
row = case.db.query(model).filter_by(id='a' if model in (CrmContract, CrmContractProduct) else 'one').one()
|
|
setattr(row, field, value)
|
|
case.db.commit()
|
|
detail = s.certificate_statistics(case.db, case.user, 'purchase', 'same')
|
|
assert detail['usageItems'] == []
|
|
assert all(c['contractReservedQty'] == 0 for c in detail['contracts'])
|
|
|
|
|
|
def test_hidden_certificates_excluded_and_page_queries_do_not_grow_with_rows(case, monkeypatch):
|
|
from sqlalchemy import event
|
|
s = service()
|
|
counts = []
|
|
for size in (1, 20):
|
|
statements = []
|
|
def capture(*args):
|
|
statements.append(args[2])
|
|
event.listen(case.db.bind, 'before_cursor_execute', capture)
|
|
try:
|
|
s.list_contracts(case.db, case.user, page_size=size)
|
|
finally:
|
|
event.remove(case.db.bind, 'before_cursor_execute', capture)
|
|
counts.append(len(statements))
|
|
assert all(sql.lstrip().upper().startswith('SELECT') for sql in statements)
|
|
assert counts[0] == counts[1]
|
|
monkeypatch.setattr(s, 'apply_pl_scope', lambda q, model, *a: q.filter(model.id != 'same'))
|
|
assert s.list_contracts(case.db, case.user)['total'] == 0
|
|
assert s.contract_statistics(case.db, case.user, 'a')['certificates'] == []
|
|
|
|
|
|
def test_historical_usage_survives_removed_binding_without_recreation(case):
|
|
s = service()
|
|
add_usage(case, 'a', 'one', 0, 20)
|
|
case.db.query(m.PlContractCertLink).filter_by(id='a').one().del_flag = 1
|
|
case.db.query(m.PlContractCertProduct).filter_by(id='a').one().del_flag = 1
|
|
case.db.commit()
|
|
row = s.contract_statistics(case.db, case.user, 'a')['certificates'][0]
|
|
assert row['contractUsedQty'] == 20 and row['products'][0]['bindingId'] is None
|
|
assert case.db.query(m.PlContractCertLink).filter_by(id='a').one().del_flag == 1
|
|
|
|
|
|
def test_role_union_uses_existing_crm_owner_and_pl_creator_helpers(case, monkeypatch):
|
|
from app.core import data_scope
|
|
s = service()
|
|
add_usage(case, 'a', 'self-plan', 30)
|
|
add_usage(case, 'a', 'custom-plan', 40)
|
|
add_usage(case, 'a', 'hidden-plan', 50)
|
|
case.db.query(m.PlPurchaseCert).one().create_by = 'u'
|
|
case.db.query(CrmContract).filter_by(id='a').one().owner_user_id = 'u'
|
|
case.db.query(CrmContract).filter_by(id='b').one().owner_user_id = 'hidden'
|
|
for ident, creator in [('self-plan', 'u'), ('custom-plan', 'colleague'), ('hidden-plan', 'hidden')]:
|
|
case.db.query(PlPlan).filter_by(id=ident).one().create_by = creator
|
|
case.db.commit()
|
|
monkeypatch.setattr(data_scope, '_resolve_user_role_scopes', lambda *a: [
|
|
{'data_scope': '4', 'details': []},
|
|
{'data_scope': '5', 'details': [SimpleNamespace(scope_type='user', scope_value='colleague')]}])
|
|
user = SimpleNamespace(tenant_no='t', to_user_info=lambda: {'role': 'staff', 'user_id': 'u', 'dept_id': 'd'})
|
|
result = s.certificate_statistics(case.db, user, 'purchase', 'same')
|
|
assert [row['id'] for row in result['contracts']] == ['a']
|
|
assert result['contracts'][0]['contractReservedQty'] == 70
|
|
assert {row['planId'] for row in result['usageItems']} == {'self-plan', 'custom-plan'}
|
|
|
|
|
|
def test_all_statistics_reads_emit_no_mutations_or_commit(case, monkeypatch):
|
|
from sqlalchemy import event
|
|
s = service()
|
|
statements = []
|
|
def no_commit():
|
|
pytest.fail('统计不得提交事务')
|
|
monkeypatch.setattr(case.db, 'commit', no_commit)
|
|
def capture(*args):
|
|
statements.append(args[2])
|
|
event.listen(case.db.bind, 'before_cursor_execute', capture)
|
|
try:
|
|
s.list_contracts(case.db, case.user)
|
|
s.list_certificates(case.db, case.user)
|
|
s.contract_statistics(case.db, case.user, 'a')
|
|
s.certificate_statistics(case.db, case.user, 'purchase', 'same')
|
|
finally:
|
|
event.remove(case.db.bind, 'before_cursor_execute', capture)
|
|
assert all(sql.lstrip().upper().startswith('SELECT') for sql in statements)
|
|
|
|
@pytest.mark.parametrize('source_type,source_stop,expected', [
|
|
('delivery', {'id': 'source', 'customer_name': '本段客户', 'address': '本段送货地址'}, ('本段客户', '本段送货地址')),
|
|
('load', {'id': 'source', 'supplier_name': '本段供应商', 'address': '本段装货地址'}, ('本段供应商', '本段装货地址')),
|
|
('delivery', {'id': 'source'}, ('—', '—')),
|
|
('delivery', {'id': 'missing', 'customer_name': '其他客户', 'address': '其他地址'}, ('—', '—')),
|
|
])
|
|
def test_usage_party_is_taken_only_from_exact_source_stop_snapshot(case, source_type, source_stop, expected):
|
|
add_usage(case, 'a', 'source', 20)
|
|
item = case.db.query(m.PlPlanCertUsageItem).filter_by(id='source').one()
|
|
item.source_stop_type = source_type
|
|
item.source_snapshot = {'stops': [
|
|
{'id': 'related', 'customer_name': '关联段客户', 'supplier_name': '关联段供应商', 'address': '关联段地址'},
|
|
source_stop,
|
|
]}
|
|
case.db.commit()
|
|
row = service().certificate_statistics(case.db, case.user, 'purchase', 'same')['usageItems'][0]
|
|
assert (row['partyName'], row['address']) == expected
|
|
assert item.source_snapshot['stops'][0]['address'] == '关联段地址'
|
|
|
|
def test_number_filters_are_independent_from_keyword_and_applied_before_pagination(case):
|
|
s = service()
|
|
case.db.query(CrmContract).filter_by(id='a').one().contract_name = '目标合同'
|
|
case.db.commit()
|
|
assert s.list_contracts(case.db, case.user, keyword='目标', contract_no='a', page_size=1)['total'] == 1
|
|
assert s.list_contracts(case.db, case.user, keyword='目标', contract_no='b')['total'] == 0
|
|
assert s.list_contracts(case.db, case.user, contract_no='目标')['total'] == 0
|
|
assert s.list_certificates(case.db, case.user, keyword='硫酸', cert_no='P', page_size=1)['total'] == 1
|
|
assert s.list_certificates(case.db, case.user, keyword='硫酸', cert_no='missing')['total'] == 0
|
|
assert s.list_certificates(case.db, case.user, cert_no='硫酸')['total'] == 0
|