"""额度账本内存事务回归。"""
from datetime import date, datetime
from decimal import Decimal as D
from types import SimpleNamespace
import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import Session

@pytest.fixture(params=[True, False], ids=["autoflush", "explicit_flush"])
def case(request):
    from app.models import pl_quota_models as m
    from app.models.pl_models import PlPlan
    from app.models.crm_models import CrmContract, CrmContractProduct
    engine = create_engine('sqlite://')
    for model in [PlPlan, CrmContract, CrmContractProduct] + [getattr(m,n) for n in dir(m) if n.startswith('Pl')]:
        model.__table__.create(engine)
    db = Session(engine, autoflush=request.param)
    user = SimpleNamespace(id='u',dept_id='d',tenant_no='t',to_user_info=lambda: {'role':'super_admin','user_id':'u'})
    for pid in ['plan','other']:
        db.add(PlPlan(id=pid,plan_no=pid,create_date=date.today(),belongs_company='shanxia',plan_type='direct',tenant_no='t'))
    db.add(CrmContract(id='c',contract_no='c',contract_name='c',customer_id='customer',contract_type='sale',status='approved',tenant_no='t'))
    db.add(CrmContractProduct(id='cp',contract_id='c',product_id='product',product_name='酸',quantity=100,tenant_no='t'))
    for kind,model,pid in [('purchase',m.PlPurchaseCert,'p'),('transport',m.PlTransportCert,'t')]:
        extra = dict(buyer_name='买',seller_name='卖') if kind=='purchase' else dict(shipper_name='货主',vehicle_numbers_json=['CAR'])
        db.add(model(id=pid,cert_no=pid,product_name_on_cert='酸',total_qty=100,quantity_unit='吨',valid_use_text='多次有效',valid_from=date(2020,1,1),valid_to=date(2099,12,31),status='active',verified_by='u',verified_time=datetime.now(),tenant_no='t',**extra))
        db.add(m.PlCertQtyLog(id='opening'+pid,operation_id='opening'+pid,request_hash='opening',cert_type=kind,cert_id=pid,event_type='opening',quantity_unit='吨',before_total_qty=100,after_total_qty=100,before_reserved_qty=0,after_reserved_qty=0,before_used_qty=0,after_used_qty=0,reason='期初',tenant_no='t'))
        db.add(m.PlContractCertLink(id='l'+pid,contract_id='c',contract_no_snapshot='c',cert_type=kind,cert_id=pid,tenant_no='t'))
        db.add(m.PlContractCertProduct(id='s'+pid,contract_cert_link_id='l'+pid,contract_product_id='cp',tenant_no='t'))
    db.add(m.PlTransportCertPurchaseRef(id='ref',transport_cert_id='t',reference_type='purchase_permit',reference_no='p',purchase_cert_id='p',tenant_no='t'))
    db.commit()
    yield SimpleNamespace(db=db,user=user,m=m)
    db.close(); engine.dispose()

def request(**kw):
    return dict(operation_id='op',action='reserve',plan_id='plan',purchase_cert_ids=['p'],transport_cert_ids=['t'],vehicle_no='CAR',sources=[dict(source_stop_type='load',source_stop_id='stop',source_stop_product_id='sp',plan_product_id='pp',contract_id='c',contract_product_id='cp',product_id='product',quantity='29',unit='吨',fingerprint='fp')],**kw)

def run(c, **kw):
    from app.services.pl_quota_usage_service import execute_usage
    body=request(); body.update(kw)
    return execute_usage(c.db,c.user,body)

def test_retry_and_conflict(case):
    c=case
    first=run(c)
    assert run(c)==first
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert c.db.query(c.m.PlCertQtyLog).filter_by(event_type='reserve').count()==2
    for kw in [dict(plan_id='other'),dict(action='resize'),dict(purchase_cert_ids=[])]:
        with pytest.raises(ValueError): run(c,**kw)

def test_resize_atomic_and_identity(case):
    c=case; run(c)
    ids=[x.id for x in c.db.query(c.m.PlPlanCertUsageItem).all()]
    sources=request()['sources']; sources[0]['quantity']='101'
    with pytest.raises(ValueError): run(c,operation_id='bad',action='resize',sources=sources)
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert c.db.query(c.m.PlCertOperation).count()==1
    sources[0]['quantity']='30'
    run(c,operation_id='resize',action='resize',sources=sources)
    assert [x.id for x in c.db.query(c.m.PlPlanCertUsageItem).all()]==ids
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==30

def test_settle_and_adjust_delta(case):
    c=case; run(c)
    run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':'18'})
    assert c.db.get(c.m.PlPurchaseCert,'p').used_qty==18
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==0
    run(c,operation_id='adjust',action='actual_adjust',actual_by_source={'load:sp':'20'},reason='更正',source_note='磅单核对')
    assert c.db.get(c.m.PlPurchaseCert,'p').used_qty==20
    assert c.db.query(c.m.PlCertQtyLog).filter_by(operation_id='adjust',cert_type='purchase').one().used_delta==2

def test_once_zero_and_opening(case):
    c=case; c.db.get(c.m.PlPurchaseCert,'p').valid_use_text='一次有效'; c.db.flush()
    run(c)
    run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':'0'})
    with pytest.raises(ValueError, match='已执行'):
        run(c,operation_id='release',action='release')
    assert all(u.usage_status == 'settled' for u in c.db.query(c.m.PlPlanCertUsage).all())
    with pytest.raises(ValueError): run(c,operation_id='other',plan_id='other')

def test_shared_pool_and_scope(case):
    c=case; run(c)
    sources=request()['sources']; sources[0]['quantity']='80'
    with pytest.raises(ValueError): run(c,operation_id='other',plan_id='other',sources=sources)
    c.db.get(c.m.PlContractCertProduct,'st').scope_status='inactive'; c.db.flush()
    with pytest.raises(ValueError): run(c,operation_id='invalid',action='resize')
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29

def test_opening_required_and_consumed(case):
    c=case
    c.db.get(c.m.PlCertQtyLog,'openingp').del_flag=1; c.db.flush()
    with pytest.raises(ValueError,match='期初'): run(c)

def test_zero_reservation_consumes_once_slot(case):
    c=case; c.db.get(c.m.PlPurchaseCert,'p').valid_use_text='一次有效'; c.db.flush()
    sources=request()['sources']; sources[0]['quantity']='0'
    run(c,sources=sources)
    assert c.db.query(c.m.PlPlanCertUsage).count()==2
    with pytest.raises(ValueError): run(c,operation_id='other',plan_id='other',sources=sources)

def test_double_cert_failure_keeps_all_balances(case):
    c=case; c.db.get(c.m.PlTransportCert,'t').total_qty=10; c.db.flush()
    with pytest.raises(ValueError): run(c)
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==0
    assert c.db.query(c.m.PlPlanCertUsage).count()==0
    assert c.db.query(c.m.PlCertOperation).count()==0

def test_mult_unit_requires_new_manual_actuals(case):
    c=case; c.db.get(c.m.PlTransportCert,'t').quantity_unit='kg'; c.db.flush()
    manual=[dict(permit_kind='transport',source_id='load:sp',permit_id='t',source_quantity='29',permit_quantity='58',note='人工核实')]
    run(c,manual_allocations=manual)
    with pytest.raises(ValueError): run(c,operation_id='fail',action='settle',actual_by_source={'load:sp':'18'})
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    actual=[dict(permit_kind='transport',source_id='load:sp',permit_id='t',source_quantity='18',permit_quantity='35',note='归档重新核实')]
    run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':'18'},manual_actuals=actual)
    assert c.db.get(c.m.PlTransportCert,'t').used_qty==35
    assert c.db.get(c.m.PlPurchaseCert,'p').used_qty==18

def test_actual_missing_and_overage_are_atomic(case):
    c=case; run(c)
    for actual in [{},{'load:sp':'30'}]:
        with pytest.raises(ValueError): run(c,operation_id='fail',action='settle',actual_by_source=actual)
        assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
        assert c.db.get(c.m.PlPurchaseCert,'p').used_qty==0

def test_total_adjust_floor_and_audit(case):
    c=case; run(c)
    from app.services.pl_quota_usage_service import execute_usage
    data=dict(action='total_adjust',operation_id='total',cert_type='purchase',cert_id='p',total_qty='28',reason='修订',source_note='发证机关核实')
    with pytest.raises(ValueError): execute_usage(c.db,c.user,data)
    data['total_qty']='120'
    result=execute_usage(c.db,c.user,data)
    assert execute_usage(c.db,c.user,data)==result
    log=c.db.query(c.m.PlCertQtyLog).filter_by(event_type='total_adjust').one()
    assert log.total_delta==20 and log.reserved_delta==0

def test_tenant_and_contract_product_identity(case):
    c=case
    from fastapi import HTTPException
    from app.services.pl_quota_usage_service import execute_usage
    other=SimpleNamespace(**{**c.user.__dict__,'tenant_no':'foreign'})
    with pytest.raises(HTTPException): execute_usage(c.db,other,request())
    sources=request()['sources']; sources[0]['product_id']='fake'
    with pytest.raises(ValueError): run(c,sources=sources)
    assert c.db.query(c.m.PlCertOperation).count()==0

def test_opening_consumed_and_expiry_vehicle_block(case):
    c=case; p=c.db.get(c.m.PlPurchaseCert,'p')
    p.valid_use_text='一次有效'; p.opening_single_use_consumed=1; c.db.flush()
    with pytest.raises(ValueError): run(c)
    p.opening_single_use_consumed=0; p.valid_to=date(2000,1,1); c.db.flush()
    with pytest.raises(ValueError): run(c)
    p.valid_to=date(2099,12,31); c.db.flush()
    with pytest.raises(ValueError): run(c,vehicle_no='unknown')

def test_no_cert_operation_is_still_idempotent(case):
    c=case
    first=run(c,sources=[],purchase_cert_ids=[],transport_cert_ids=[])
    assert run(c,sources=[],purchase_cert_ids=[],transport_cert_ids=[])==first
    assert c.db.query(c.m.PlCertOperation).count()==1

def test_removal_and_restore_preserve_item_id(case):
    c=case; run(c)
    ids={i.id for i in c.db.query(c.m.PlPlanCertUsageItem).all()}
    run(c,action='resize',operation_id='clear',sources=[],purchase_cert_ids=[],transport_cert_ids=[])
    assert all(i.del_flag==1 for i in c.db.query(c.m.PlPlanCertUsageItem).all())
    run(c,action='resize',operation_id='restore')
    assert {i.id for i in c.db.query(c.m.PlPlanCertUsageItem).all()}==ids
    assert all(i.del_flag==0 for i in c.db.query(c.m.PlPlanCertUsageItem).all())

def _second_pair(c, cross=False):
    for kind,model,old_id,new_id,total in [('purchase',c.m.PlPurchaseCert,'p','p2',100),('transport',c.m.PlTransportCert,'t','t2',100)]:
        original=c.db.get(model,old_id)
        values={col.name:getattr(original,col.name) for col in model.__table__.columns}
        values.update(id=new_id,cert_no=new_id,total_qty=total)
        c.db.add(model(**values))
        c.db.add(c.m.PlContractCertLink(id='l'+new_id,contract_id='c',contract_no_snapshot='c',cert_type=kind,cert_id=new_id,tenant_no='t'))
        c.db.add(c.m.PlContractCertProduct(id='s'+new_id,contract_cert_link_id='l'+new_id,contract_product_id='cp',tenant_no='t'))
        log=c.db.get(c.m.PlCertQtyLog,'opening'+old_id)
        lv={col.name:getattr(log,col.name) for col in c.m.PlCertQtyLog.__table__.columns}
        lv.update(id='opening'+new_id,operation_id='opening'+new_id,cert_id=new_id)
        c.db.add(c.m.PlCertQtyLog(**lv))
    c.db.add(c.m.PlTransportCertPurchaseRef(id='ref2',transport_cert_id='t2',reference_type='purchase_permit',reference_no='p2' if not cross else 'p',purchase_cert_id='p2' if not cross else 'p',tenant_no='t'))
    if cross:
        c.db.get(c.m.PlTransportCertPurchaseRef,'ref').purchase_cert_id='p2'
    c.db.get(c.m.PlPurchaseCert,'p').total_qty=20
    c.db.get(c.m.PlTransportCert,'t').total_qty=9 if cross else 20
    c.db.flush()

@pytest.mark.parametrize('actual,expected', [('29',[D('20'),D('9')]),('18',[D('18'),D('0')])])
def test_split_settlement_order(case,actual,expected):
    c=case; _second_pair(c)
    run(c,purchase_cert_ids=['p','p2'],transport_cert_ids=['t','t2'])
    run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':actual})
    assert [c.db.get(c.m.PlPurchaseCert,p).used_qty for p in ['p','p2']]==expected
    assert all(u.reserved_qty==0 for u in c.db.query(c.m.PlPlanCertUsage).all())

def test_split_pairing_failure_does_not_partially_settle(case):
    c=case; _second_pair(c,cross=True)
    run(c,purchase_cert_ids=['p','p2'],transport_cert_ids=['t','t2'])
    with pytest.raises(ValueError): run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':'18'})
    assert sum(u.reserved_qty for u in c.db.query(c.m.PlPlanCertUsage).all())==58
    assert all(u.used_qty==0 for u in c.db.query(c.m.PlPlanCertUsage).all())

def test_zero_foreign_unit_still_requires_manual_pair(case):
    c=case; c.db.get(c.m.PlTransportCert,'t').quantity_unit='kg'; c.db.flush()
    sources=request()['sources']; sources[0]['quantity']='0'
    manual=[dict(permit_kind='transport',source_id='load:sp',permit_id='t',source_quantity='0',permit_quantity='0',note='零量人工核实')]
    run(c,sources=sources,manual_allocations=manual)
    assert c.db.query(c.m.PlPlanCertUsage).count()==2


@pytest.mark.parametrize('marker', ['request', 'in_transit', 'settled', 'snapshot'])
def test_executed_release_cannot_free_multi_use_quota(case, marker):
    c=case; run(c)
    for u in c.db.query(c.m.PlPlanCertUsage).all():
        if marker in ('in_transit','settled'): u.usage_status=marker
        if marker == 'snapshot': u.confirmation_snapshot={'executed':True}
    c.db.flush()
    with pytest.raises(ValueError, match='已执行'):
        run(c, operation_id='release', action='release', executed=marker=='request')
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert all(u.reserved_qty==29 for u in c.db.query(c.m.PlPlanCertUsage).all())
    assert all(i.reserved_qty==29 for i in c.db.query(c.m.PlPlanCertUsageItem).all())
    assert c.db.query(c.m.PlCertOperation).count()==1

@pytest.mark.parametrize('field,value', [('fingerprint','changed'),('source_stop_id','new-stop'),
    ('source_stop_product_id','new-product'),('plan_product_id','other-pp'),('contract_id','other-contract'),
    ('contract_product_id','other-cp'),('product_id','other-product'),('quantity','30'),('unit','kg'),
    ('related_stop_products',['new-delivery'])])
@pytest.mark.parametrize('action', ['settle','actual_adjust'])
def test_settle_refuses_changed_authoritative_source(case, field, value, action):
    c=case; run(c)
    if action=='actual_adjust': run(c,operation_id='settle',action='settle',actual_by_source={'load:sp':'18'})
    sources=request()['sources']; sources[0][field]=value
    with pytest.raises(ValueError, match='来源'):
        run(c,operation_id='changed',action=action,sources=sources,actual_by_source={'load:sp':'19'},reason='纠错',source_note='磅单')
    cert=c.db.get(c.m.PlPurchaseCert,'p')
    assert (cert.reserved_qty,cert.used_qty)==((0,18) if action=='actual_adjust' else (29,0))

@pytest.mark.parametrize('source_mode', ['missing', 'duplicate', 'extra'])
def test_settle_requires_exact_source_set(case, source_mode):
    c=case; run(c); sources=request()['sources']
    if source_mode=='missing': sources=[]
    elif source_mode=='duplicate': sources=sources+sources
    else: sources=sources+[{**sources[0], 'source_stop_product_id':'extra'}]
    with pytest.raises(ValueError, match='来源'):
        run(c,operation_id='settle',action='settle',sources=sources,actual_by_source={'load:sp':'18'})
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29

@pytest.mark.parametrize('field', ['valid_from','valid_to'])
@pytest.mark.parametrize('action', ['reserve','resize'])
def test_incomplete_dates_block_reservation(case,field,action):
    c=case
    if action=='resize': run(c)
    setattr(c.db.get(c.m.PlPurchaseCert,'p'),field,None); c.db.flush()
    with pytest.raises(ValueError,match='有效期'):
        run(c,operation_id='date',action=action)
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==(29 if action=='resize' else 0)

@pytest.mark.parametrize('on_date', [date(2020,1,1), date(2099,12,31)])
def test_date_boundaries_are_inclusive(case,on_date):
    run(case,on_date=on_date)
    assert case.db.get(case.m.PlPurchaseCert,'p').reserved_qty==29

def test_preview_uses_own_reservation_without_writes(case):
    from app.services.pl_quota_usage_service import preview_usage
    c=case; run(c)
    before=(c.db.query(c.m.PlCertOperation).count(),c.db.query(c.m.PlCertQtyLog).count())
    data=request(); data['action']='resize'; data['sources'][0]['quantity']='100'
    result=preview_usage(c.db,c.user,data)
    assert result['purchaseAllocations'][0]['permitQuantity']=='100'
    assert result['purchaseAllocations'][0]['ownReservedQty']=='29'
    assert result['purchaseAllocations'][0]['contractCertProductId']=='sp'
    assert result['transportAllocations'][0]['certRevision']==1
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert (c.db.query(c.m.PlCertOperation).count(),c.db.query(c.m.PlCertQtyLog).count())==before
    assert not c.db.new and not c.db.dirty

def test_preview_uses_same_eligibility_and_never_records_operation(case):
    from app.services.pl_quota_usage_service import preview_usage
    c=case; c.db.get(c.m.PlPurchaseCert,'p').valid_to=None; c.db.flush()
    with pytest.raises(ValueError,match='有效期'): preview_usage(c.db,c.user,request())
    assert c.db.query(c.m.PlCertOperation).count()==0
    assert c.db.query(c.m.PlPlanCertUsage).count()==0


def test_preview_insufficiency_includes_current_shared_capacity(case):
    from app.services.pl_quota_usage_service import preview_usage
    from app.services.pl_quota_allocation import InsufficientQuota
    c = case
    c.db.get(c.m.PlPurchaseCert, 'p').total_qty = D('20')
    c.db.flush()
    with pytest.raises(InsufficientQuota) as raised:
        preview_usage(c.db, c.user, request())
    row = raised.value.diagnostics[0]
    assert row['certType'] == 'purchase' and D(row['demand']) == 29
    assert D(row['availableUpperBound']) == 20 and D(row['gapLowerBound']) == 9
    assert c.db.query(c.m.PlCertOperation).count() == 0
    assert c.db.query(c.m.PlPlanCertUsage).count()==0

@pytest.mark.parametrize('action', ['reserve','resize'])
@pytest.mark.parametrize('field,value', [('status','void'),('status','draft'),('contract_type','other')])
def test_current_contract_must_remain_executable(case,action,field,value):
    from app.models.crm_models import CrmContract
    c=case
    if action=='resize': run(c)
    setattr(c.db.get(CrmContract,'c'),field,value); c.db.flush()
    with pytest.raises(ValueError,match='合同'):
        run(c,operation_id='contract-changed',action=action)
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==(29 if action=='resize' else 0)


def test_initial_operation_lookup_does_not_lock_missing_key(case,monkeypatch):
    from sqlalchemy.orm import Query
    from sqlalchemy.dialects import mysql
    c=case; reads=[]; original=Query.first
    def observed(query):
        if query.column_descriptions[0].get('entity') is c.m.PlCertOperation:
            reads.append(str(query.statement.compile(dialect=mysql.dialect())))
        return original(query)
    monkeypatch.setattr(Query,'first',observed)
    run(c)
    assert reads and 'FOR UPDATE' not in reads[0]

@pytest.mark.parametrize('conflict', [False,True])
def test_operation_unique_insert_arbitrates_stale_snapshot(case,monkeypatch,conflict):
    """真实SQLite唯一冲突；只隐藏初次快照读以模拟另一事务已提交。"""
    from sqlalchemy.orm import Query
    from sqlalchemy.dialects import mysql
    c=case; result=run(c); reads=[]; first=Query.first; one=Query.one
    def stale(query):
        if query.column_descriptions[0].get('entity') is c.m.PlCertOperation and not reads:
            reads.append((str(query.statement.compile(dialect=mysql.dialect())),False))
            return None
        return first(query)
    def current(query):
        if query.column_descriptions[0].get('entity') is c.m.PlCertOperation:
            reads.append((str(query.statement.compile(dialect=mysql.dialect())),query.load_options._populate_existing))
        return one(query)
    monkeypatch.setattr(Query,'first',stale); monkeypatch.setattr(Query,'one',current)
    if conflict:
        with pytest.raises(ValueError,match='操作号'): run(c,plan_id='other')
    else:
        assert run(c)==result
    assert len(reads)==2
    assert 'FOR UPDATE' not in reads[0][0]
    assert 'FOR UPDATE' in reads[1][0] and reads[1][1]
    assert c.db.query(c.m.PlCertOperation).count()==1
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert c.db.query(c.m.PlCertQtyLog).filter_by(event_type='reserve').count()==2


def test_new_current_usage_rejects_without_out_of_order_cert_lock(case,monkeypatch):
    """仅隐藏usage快照读；当前读必须发现未锁证件，原子409而非临时补锁。"""
    from sqlalchemy.orm import Query
    from fastapi import HTTPException
    from app.services import pl_quota_usage_service as svc
    c=case; run(c); original=Query.all; cert_get=svc._get_row; hidden=[]; locked=[]
    def stale(query):
        if (query.column_descriptions[0].get('entity') is c.m.PlPlanCertUsage
                and query._for_update_arg is None and not hidden):
            hidden.append(True)
            return []
        return original(query)
    def tracked(db,user,kind,cert_id,lock=False):
        if lock: locked.append((kind,cert_id))
        return cert_get(db,user,kind,cert_id,lock=lock)
    monkeypatch.setattr(Query,'all',stale); monkeypatch.setattr(svc,'_get_row',tracked)
    with pytest.raises(HTTPException) as exc:
        run(c,operation_id='release-race',action='release',purchase_cert_ids=[],transport_cert_ids=[],sources=[])
    assert exc.value.status_code==409
    assert locked==[]
    assert c.db.get(c.m.PlPurchaseCert,'p').reserved_qty==29
    assert all(u.reserved_qty==29 for u in c.db.query(c.m.PlPlanCertUsage).all())
    assert c.db.query(c.m.PlCertOperation).count()==1