68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Tests for SWIFT FIN Parser
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from parsers.swift_fin_parser import SWIFTFINParser
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def parser():
|
||
|
|
return SWIFTFINParser()
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def sample_mt103():
|
||
|
|
return """{1:F01BANKUS33AXXX1234567890}
|
||
|
|
{2:O1031200240101BANKUS33AXXX1234567890123456789012345678901234567890}
|
||
|
|
{3:{108:MT103}}
|
||
|
|
{4:
|
||
|
|
:20:REF123
|
||
|
|
:23B:CRED
|
||
|
|
:32A:240101USD1000,00
|
||
|
|
:50A:/US1234567890
|
||
|
|
BANK US
|
||
|
|
:59:/US9876543210
|
||
|
|
CREDITOR NAME
|
||
|
|
:70:Payment for services
|
||
|
|
:71A:SHA
|
||
|
|
-}
|
||
|
|
{5:{MAC:12345678}{CHK:ABCDEF123456}{TNG:}{PDE:}}
|
||
|
|
{S:{COP:P}}"""
|
||
|
|
|
||
|
|
def test_parse_mt103(parser, sample_mt103):
|
||
|
|
"""Test parsing MT103 message"""
|
||
|
|
result = parser.parse(sample_mt103)
|
||
|
|
|
||
|
|
assert result['type'] == 'SWIFT_FIN'
|
||
|
|
assert result['messageType'] == 'MT103'
|
||
|
|
assert 'parsed' in result
|
||
|
|
assert 'senderReference' in result['parsed']
|
||
|
|
assert 'amount' in result['parsed']
|
||
|
|
|
||
|
|
def test_extract_message_type(parser, sample_mt103):
|
||
|
|
"""Test message type extraction"""
|
||
|
|
lines = sample_mt103.strip().split('\n')
|
||
|
|
app_header = parser._parse_block(lines, 2)
|
||
|
|
message_type = parser._extract_message_type(app_header.get('raw', ''))
|
||
|
|
assert message_type == 'MT103'
|
||
|
|
|
||
|
|
def test_parse_swift_fields(parser):
|
||
|
|
"""Test SWIFT field parsing"""
|
||
|
|
text = ":20:REF123\n:32A:240101USD1000,00\n:50A:/US1234567890\nBANK US"
|
||
|
|
fields = parser._parse_swift_fields(text)
|
||
|
|
assert '20' in fields
|
||
|
|
assert '32A' in fields
|
||
|
|
assert '50A' in fields
|
||
|
|
|
||
|
|
def test_parse_balance(parser):
|
||
|
|
"""Test balance parsing"""
|
||
|
|
balance_str = "D230101USD1000,00"
|
||
|
|
balance = parser._parse_balance(balance_str)
|
||
|
|
assert balance['creditDebitIndicator'] == 'Debit'
|
||
|
|
assert balance['currency'] == 'USD'
|
||
|
|
assert balance['amount'] == '1000.00'
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
pytest.main([__file__])
|
||
|
|
|