100 lines
4.9 KiB
Python
100 lines
4.9 KiB
Python
from decimal import Decimal
|
|
|
|
from app.extensions import db
|
|
from app.models.catalog import Customer, Product, InvoiceLine
|
|
from app.models.company import Company
|
|
from app.models.invoice import Invoice, InvoiceStatus, InvoiceType
|
|
from app.models.setting import AppSetting
|
|
from app.services.invoice_service import InvoiceService
|
|
|
|
|
|
def _enable_nfz(app):
|
|
with app.app_context():
|
|
company = Company.query.first()
|
|
AppSetting.set(f'company.{company.id}.modules.nfz_enabled', 'true')
|
|
customer = Customer.query.filter_by(company_id=company.id, tax_id='1070001057').first()
|
|
if not customer:
|
|
customer = Customer(company_id=company.id, name='NFZ Client', tax_id='1070001057', is_active=True)
|
|
db.session.add(customer)
|
|
product = Product.query.filter_by(company_id=company.id, name='Swiadczenie NFZ').first()
|
|
if not product:
|
|
product = Product(company_id=company.id, name='Swiadczenie NFZ', unit='usl', net_price=Decimal('100.00'), vat_rate=Decimal('8.00'), is_active=True)
|
|
db.session.add(product)
|
|
db.session.commit()
|
|
return company.id, customer.id, product.id
|
|
|
|
|
|
def test_duplicate_nfz_redirects_to_nfz_form(auth_client, app):
|
|
company_id, customer_id, product_id = _enable_nfz(app)
|
|
with app.app_context():
|
|
invoice = Invoice(
|
|
company_id=company_id,
|
|
customer_id=customer_id,
|
|
ksef_number='NFZ-PENDING/1',
|
|
invoice_number='FV/NFZ/1',
|
|
contractor_name='Narodowy Fundusz Zdrowia - Slaski OW NFZ',
|
|
contractor_nip='1070001057',
|
|
issue_date=InvoiceService.today_date(),
|
|
received_date=InvoiceService.today_date(),
|
|
net_amount=Decimal('100.00'),
|
|
vat_amount=Decimal('8.00'),
|
|
gross_amount=Decimal('108.00'),
|
|
invoice_type=InvoiceType.SALE,
|
|
status=InvoiceStatus.NEW,
|
|
source='nfz',
|
|
issued_status='draft',
|
|
external_metadata={'nfz': {'recipient_branch_id': '1070001057-00122', 'settlement_from': '2026-02-01', 'settlement_to': '2026-02-28', 'template_identifier': 'TMP', 'provider_identifier': 'SWD1', 'service_code': '01.02', 'contract_number': 'C/1', 'nfz_schema': 'FA(3)'}},
|
|
)
|
|
db.session.add(invoice)
|
|
db.session.flush()
|
|
db.session.add(InvoiceLine(invoice_id=invoice.id, product_id=product_id, description='Swiadczenie NFZ', quantity=1, unit='usl', unit_net=Decimal('100.00'), vat_rate=Decimal('8.00'), net_amount=Decimal('100.00'), vat_amount=Decimal('8.00'), gross_amount=Decimal('108.00')))
|
|
db.session.commit()
|
|
invoice_id = invoice.id
|
|
|
|
response = auth_client.get(f'/invoices/{invoice_id}/duplicate', follow_redirects=False)
|
|
assert response.status_code == 302
|
|
assert f'/nfz/?duplicate_id={invoice_id}' in response.headers['Location']
|
|
|
|
form_response = auth_client.get(response.headers['Location'])
|
|
body = form_response.get_data(as_text=True)
|
|
assert form_response.status_code == 200
|
|
assert 'FV/NFZ/1/COPY' in body
|
|
assert 'SWD1' in body
|
|
assert 'C/1' in body
|
|
|
|
|
|
def test_build_ksef_payload_contains_nfz_xml(app):
|
|
company_id, customer_id, product_id = _enable_nfz(app)
|
|
with app.app_context():
|
|
company = Company.query.first()
|
|
invoice = Invoice(
|
|
company_id=company_id,
|
|
company=company,
|
|
customer_id=customer_id,
|
|
ksef_number='NFZ-PENDING/2',
|
|
invoice_number='FV/NFZ/2',
|
|
contractor_name='Narodowy Fundusz Zdrowia - Slaski OW NFZ',
|
|
contractor_nip='1070001057',
|
|
issue_date=InvoiceService.today_date(),
|
|
received_date=InvoiceService.today_date(),
|
|
net_amount=Decimal('100.00'),
|
|
vat_amount=Decimal('8.00'),
|
|
gross_amount=Decimal('108.00'),
|
|
invoice_type=InvoiceType.SALE,
|
|
status=InvoiceStatus.NEW,
|
|
source='nfz',
|
|
issued_status='draft',
|
|
external_metadata={'nfz': {'recipient_branch_id': '1070001057-00122', 'settlement_from': '2026-02-01', 'settlement_to': '2026-02-28', 'template_identifier': 'TMP', 'provider_identifier': 'SWD1', 'service_code': '01.02', 'contract_number': 'C/1', 'nfz_schema': 'FA(3)'}},
|
|
)
|
|
db.session.add(invoice)
|
|
db.session.flush()
|
|
db.session.add(InvoiceLine(invoice_id=invoice.id, product_id=product_id, description='Swiadczenie NFZ', quantity=1, unit='usl', unit_net=Decimal('100.00'), vat_rate=Decimal('8.00'), net_amount=Decimal('100.00'), vat_amount=Decimal('8.00'), gross_amount=Decimal('108.00')))
|
|
db.session.commit()
|
|
|
|
payload = InvoiceService().build_ksef_payload(invoice)
|
|
assert payload['schemaVersion'] == 'FA(3)'
|
|
assert '<IDWew>1070001057-00122</IDWew>' in payload['xml_content']
|
|
assert '<P_6_Od>2026-02-01</P_6_Od>' in payload['xml_content']
|
|
assert '<P_6_Do>2026-02-28</P_6_Do>' in payload['xml_content']
|
|
assert 'NrUmowy' in payload['xml_content']
|