pj( d dl mZ d dlZd dlmc mZ d dlZd dl m
Z
d dl Z d dl m
Z
mZmZmZmZmZmZmZmZ d dlmZmZ d dlmZ d dlm Z d Z d dZ d dZ d!dZ d"dZ!d"dZ"d"dZ#d"dZ$d"dZ%d"dZ&d"dZ'd!dZ(d!d Z)d"d Z*d"d Z+dS )# ) annotationsN)Path) CandidateInputconfirm_candidatedeterministic_parse_docx#import_confirmed_candidate_snapshotqwen_extract_candidates qwen_resolve_staged_candidatesresolve_candidatestage_docx_candidatesupsert_candidate)CatalogEntryInputupsert_catalog_entry)connect_memory)apply_migrationsc t } t | | d | d | S )NzINSERT INTO platforms(platform_id,name,platform_type,default_currency,created_at) VALUES('p1','PostFinance','broker','CHF','now')zINSERT INTO accounts(account_id,platform_id,account_name,account_type,currency,created_at) VALUES('a1','p1','Depot','brokerage','CHF','now'))r r execute)conns f/home/agent/.hermes/worktrees/FinanceManager-sprint20d/tests/unit/test_instrument_import_candidates.py
setup_connr s^ DTLL U V V VLL ` a a a K pathr lines list[str]returnc d d |D }d| d}t j | d 5 }| d| d d d n # 1 swxY w Y | S )N c 3 " K | ]
}d | dV dS )zzN ).0lines r z make_docx..! s. PP$=T===PPPPPPr z
zwzword/document.xml)joinzipfileZipFilewritestr)r r bodyxmlzfs r make_docxr+ s
77PP%PPP P PD {\` { { {C s # # .r
'---. . . . . . . . . . . . . . . Ks A A"%A"r c ~ t ddddddddd dddd d
}| | t d i |S )
N broker.docx postfinance1
Sample ETFSAMPr CHF1.2345etfneeds_manual_review) source_fileplatformsource_row_numberraw_name
raw_tickerraw_isin raw_currency raw_quantityraw_asset_class
proposed_nameproposed_tickerproposed_currencyproposed_asset_classmapping_statusr )dictupdater ) overridesdatas r candidaterH ) sh ! " ", D KK ! !D ! !!r tmp_pathNonec t | dz dg }t |d }t | }d}||k }|st j d|fd||f dt j v st j t rt j t ndd t j v st j | rt j | nd t j | t j | d
z }d d |iz }t t j
| d x}x}}|d
}|j }d}||k }|st j d|fd||f t j | t j | t j | dz }d d |iz }t t j
| d x}x}x}}|d
}|j }d}||k }|st j d|fd||f t j | t j | t j | dz }d d |iz }t t j
| d x}x}x}}d S )NzPostfinance.docx"Sample ETF CH0000000001 CHF 12.345r. r7 ==z0%(py3)s
{%(py3)s = %(py0)s(%(py1)s)
} == %(py6)slenrowspy0py1py3py6assert %(py8)spy8r CH0000000001)z0%(py3)s
{%(py3)s = %(py1)s.raw_isin
} == %(py6)srV rW rX exact_isin_match)z6%(py3)s
{%(py3)s = %(py1)s.mapping_status
} == %(py6)s)
r+ r rR
@pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _safereprAssertionError_format_explanationr; rC ) rI docxrS @py_assert2 @py_assert5 @py_assert4 @py_format7 @py_format9 @py_assert0s r ,test_deterministic_parser_creates_candidatesrm > s X 225Y4Z [ [D #D= A A AD t999>933tt9 7-7 -~- ~ ---------- ~-----7---- ----~----------------------------- 777 !7%77 !%7 7777777777 !%77777777777 !7777%77777777777777777777777777777777r c $ i G d d fd}| d| t }t dgdd \ }}d }||u }|st j d
|fd ||f d t j v st j | rt j | nd t j | d
z }dd|iz }t t j
| d x}}t | }d} || k }
|
st j d|
fd|| f dt j v st j t rt j t nddt j v st j | rt j | ndt j | t j | dz } dd| iz } t t j
| d x}x}
} d}
d }|
|v}|slt j d|fd|
|f t j |
t j | d z }d d |iz } t t j
| d x}
x}}|d }
|
j }d } || k }
|
st j d|
fd!|| f t j |
t j | t j | d"z } dd| iz } t t j
| d x}
x}x}
} t ||d | d# d$ }
d}|
|k }|slt j d|fd%|
|f t j |
t j | d z }d d |iz } t t j
| d x}
x}}| d& d$ }
d }|
|k }|slt j d|fd%|
|f t j |
t j | d z }d d |iz } t t j
| d x}
x}}d S )'Nc e Zd Zd Zd Zd ZdS )Ntest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponsec | S Nr selfs r __enter__zXtest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.__enter__I D[r c dS NFr rt argss r __exit__zWtest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.__exit__J %%r c dS )Ns {"choices":[{"message":{"content":"{\"positions\":[{\"name\":\"Q ETF\",\"ticker\":\"QETF\",\"currency\":\"CHF\",\"quantity\":\"1.23\",\"asset_class\":\"etf\",\"row_number\":2,\"confidence\":\"medium\"}]}"}}]}r rs s r readzStest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.readK s _ _r N__name__
__module__ __qualname__ru r{ r~ r r r FakeResponserp H s: (((/// _ _ _ _ _r r c R | j d d< S Nzutf-8r( rG decodereqkwargsr captureds r fake_urlopenzNtest_qwen_extraction_creates_only_candidates_no_bookings..fake_urlopenM ' 8??733|~~ r 933tt9 )&))6) ))))))))))6))))))6)))))))))))))))))))))))))))))))) 7%7 %2% 2 %%%%%%%%%% 2%%%%%7%%%% %%%%2%%%%%%%%%%%%%%%%%%%%%%%%%%%%%T47### <<P Q Q Z Z \ \]` afeff aef ffffffffff aeffffff affffeffffffffffffffffffffffffffff <<@ A A J J L LS QVUVV QUV VVVVVVVVVV QUVVVVVV QVVVVUVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVr c t } t | t ddd }t | | }|d }d}||k }|slt j d|fd||f t j | t j | dz }dd |iz }t t j | d x}x}}d S )
Nr r;
proposed_isinproposed_exchangerC r5 rO r r r rX ) r r
rH r r^ r_ rc rd re r cidresultrl r rg r rj s r 5test_candidate_without_isin_stays_needs_manual_reviewr \ s <PVjo DI J J J J
,,ehkgm
n
n
w
w
y
yC /K/ K ////////// K///// ////K/////////////////////////// <<^ _ _ h h j jkn otstt ost tttttttttt ostttttt ottttsttttttttttttttttttttttttttttttr c t } t | t dd }t | |ddddd t | |dd d
d }| d
|f }|d }d}||k }|slt j d|fd||f t j | t j | dz }dd|iz }t t j | d x}x}}|d }d}||k }|slt j d|fd||f t j | t j | dz }dd|iz }t t j | d x}x}}|d }d}||k }|slt j d|fd||f t j | t j | dz }dd|iz }t t j | d x}x}}| d d }d}||k }|slt j d|fd||f t j | t j | dz }dd|iz }t t j | d x}x}}| d d }d}||k }|slt j d|fd||f t j | t j | dz }dd|iz }t t j | d x}x}}d S ) Nr[ r r r1 r r2 )r r r@ r rA r r zapproved importTr zWSELECT transaction_type, source_type, quantity FROM transactions WHERE transaction_id=?transaction_typeinitial_position_snapshotrO r r r rX source_type broker_import_reviewed_snapshotquantityr3 zESELECT COUNT(*) AS c FROM alerts WHERE rule_id='cost_basis_uncertain'r rN zVSELECT COUNT(*) AS c FROM audit_log WHERE action='import_confirmed_candidate_snapshot') r r
rH r r r r r^ r_ rc rd re ) r r tx_idr rl r rg r rj s r Ntest_confirmed_candidate_can_create_initial_snapshot_with_uncertain_cost_basisr } s <nnt IN bg h h h h /cdZfm~ IM
N
N
NE
,,psxrz
{
{ E E G GC ! "A&AA "&A AAAAAAAAAA "&AAAAAA "AAAA&AAAAAAAAAAAAAAAAAAAAAAAAAAAA } B!BB !B BBBBBBBBBB !BBBBBB BBBB!BBBBBBBBBBBBBBBBBBBBBBBBBBBB z?&h&?h &&&&&&&&&&?h&&&&&?&&&&h&&&&&&&&&&&&&&&&&&&&&&&&&&& <<_ ` ` i i k klo putuu ptu uuuuuuuuuu ptuuuuuu puuuutuuuuuuuuuuuuuuuuuuuuuuuuuuuu <<p q q z z | | ~A B G FG G B FG G G G G G G G G G G B FG G G G G G B G G G G FG G G G G G G G G G G G G G G G G G G G G G G G G G G G G Gr c 8 t } t | t }t j t
5 t
| |dddd d d d n # 1 swxY w Y | d d }d}||k }|slt j
d |fd
||f t j | t j | d z }d d
|iz }t t j
| d x}x}}d S )Nr r blockedTr r r r rO r r r rX )r r
rH r r r r r r r^ r_ rc rd re )r r rl r rg r rj s r 5test_unconfirmed_candidate_cannot_productively_importr s <
>
>F "'a' "a '''''''''' "a''''''''''''6'''''6'''''' "''''a''''''''''''''''''''''''''' <<P Q Q Z Z \ \]` afeff aef ffffffffff aeffffff affffeffffffffffffffffffffffffffffffr c B | dz dz }|j t |dg d}t || }|sddt j v st
j t rt
j t nddt j v st
j | rt
j | ndt
j | t
j | d z }t t
j
| d x}}d S )
Nrepor- z Synthetic ETF CH0000000001 CHF 1r rM z>assert %(py5)s
{%(py5)s = %(py0)s(%(py1)s, platform=%(py3)s)
}r repo_docx)rU rV rW r ) parentmkdirr+ r r` ra r^ rb rc rd re )rI r rg ri r s r 0test_no_real_data_in_repo_git_safety_placeholderr s( 6!M1I
i<=>>>8>? #I ? ? ?? ???????????? #????? #?????????????I?????I?????????? ??????????????????????r c t }t |t dd i G fdd fd}| d| t |d }|j }d
}||u }|st
j d |fd ||f d
t j v st
j
| rt
j | nd
t
j | t
j | dz }dd|iz }t t
j
| d x}x}}|j }d
}||u }|st
j d |fd||f d
t j v st
j
| rt
j | nd
t
j | t
j | dz }dd|iz }t t
j
| d x}x}}|j }d}||k }|st
j d|fd||f d
t j v st
j
| rt
j | nd
t
j | t
j | dz }dd|iz }t t
j
| d x}x}}d} d }| |v}
|
slt
j d|
fd| |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}d} d }| |v}
|
slt
j d|
fd| |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}| d f }
|
d } d }| |k }
|
slt
j d|
fd | |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}d } |
d! }| |v }
|
slt
j d"|
fd#| |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}d S )$Nz999.123r )r= r c & e Zd Zd Zd Z fdZdS )Htest_qwen_resolver_sends_sanitized_candidates_only..FakeResponsec | S rr r rs s r ru zRtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__enter__ rv r c dS rx r ry s r r{ zQtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__exit__ r| r c r dz }dd l }| ddd|iigi S )Na {"updates":[{"candidate_id":"%s","proposed_name":"Sample ETF","proposed_ticker":"SAMP","proposed_isin":"","proposed_exchange":"","proposed_currency":"CHF","proposed_asset_class":"etf","mapping_status":"needs_manual_review","review_note":"Ticker-only candidate remains manual review."}]}r choicesmessagecontentjsondumpsencode)rt r( r r s r r~ zMtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.read sQ t wz zD KKK::yI 47H+I*J KLLSSUU Ur Nr r s r r r G (((/// V V V V V V Vr r c R | j d d< S r r r s r r zHtest_qwen_resolver_sends_sanitized_candidates_only..fake_urlopen r r r zqwen-3.6-agent)modelTr )z6%(py2)s
{%(py2)s = %(py0)s.qwen_available
} is %(py5)sr r r r )z1%(py2)s
{%(py2)s = %(py0)s.qwen_used
} is %(py5)sr rO )z7%(py2)s
{%(py2)s = %(py0)s.confirmed_after
} == %(py5)sr( r r r r rX r= zYSELECT mapping_status, review_note FROM instrument_import_candidates WHERE candidate_id=?rC r5 r z
manual review review_note)in)z%(py1)s in %(py4)s)r r
rH r r
qwen_availabler^ r_ r` ra rb rc rd re qwen_usedconfirmed_afterr r )r r r r r ri r r r rl rg r rj r r r r s @@@r 2test_qwen_resolver_sends_sanitized_candidates_onlyr s9 <!1 1111111111>!111111>1111!1111111111111111111111111111
,,ruxtz
{
{ E E G GC 9$99 $9 9999999999 $999999 9999$9999999999999999999999999999 0c- 00? 0 0000000000? 000000?0000 000000000000000000000000000000r c
t }t |t ddd G fdd
| d
fd t | }|j }d}||k }|st
j d|fd ||f d
t j v st
j
| rt
j | nd
t
j | t
j | d z }d d
|iz }t t
j
| d x}x}}| df }|d } d}| |k }
|
slt
j d|
fd| |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}|d } d}| |k }
|
slt
j d|
fd| |f t
j | t
j | dz } dd| iz } t t
j
| d x} x}
}d S )Nr r c & e Zd Zd Zd Z fdZdS )Ftest_qwen_resolver_does_not_accept_invented_isin..FakeResponsec | S rr r rs s r ru zPtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__enter__ rv r c dS rx r ry s r r{ zOtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__exit__ r| r c r dd l }dz }| ddd|iigi S )Nr a {"updates":[{"candidate_id":"%s","proposed_name":"Sample ETF","proposed_ticker":"SAMP","proposed_isin":"CH9999999999","proposed_exchange":"SWX","proposed_currency":"CHF","proposed_asset_class":"etf","mapping_status":"probable","review_note":"Qwen guessed an ISIN."}]}r r r r )rt r r( r s r r~ zKtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.read sQ KKK a dg gD::yI 47H+I*J KLLSSUU Ur Nr r s r r r r r r r c S rr r )akr s r zBtest_qwen_resolver_does_not_accept_invented_isin.. s hththvhv r rN rO )z0%(py2)s
{%(py2)s = %(py0)s.warnings
} == %(py5)sr r r r z[SELECT proposed_isin, mapping_status FROM instrument_import_candidates WHERE candidate_id=?r r r r rX rC probable)r r
rH r r
warningsr^ r_ r` ra rb rc rd re r r )r r r r ri r r r r rl rg r rj r r s @@r 0test_qwen_resolver_does_not_accept_invented_isinr
s <r sO " " " " " "
W V V V V V V V : : : : : : > > > > > > " " " "*8 8 8 8W W W W,= = = =: : : :} } } }u u u u
G
G
G
GW W W Wg g g g@ @ @ @1 1 1 1:/ / / / / /r