+j( d dl mZ d dlZd dlmZ d dlZd dlmZmZm Z m
Z
m Z m Z m
Z
mZmZ d dlmZmZ d dlmZ d dlmZ d Zdd Zdd
Zdd Zdd Zdd
Z ddZ ddZ ddZ ddZ ddZ!ddZ"ddZ#ddZ$ddZ%y) ) 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 } t | | j d | j 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 /tests/unit/test_instrument_import_candidates.py
setup_connr s= DTLL U VLL ` a K c dj d |D }d| d}t j | d 5 }|j d| d d d | S # 1 sw Y | S xY w)N c 3 ( K | ]
}d | d yw)zzN ).0lines r z make_docx..! s P$_TF*<=Ps z
zwzword/document.xml)joinzipfileZipFilewritestr)pathlinesbodyxmlzfs r make_docxr) sf
77P%P PD\\`[aaw {C s # .r
'-. K. Ks AA c l t ddddddddd dddd d
}|j | 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 candidaterF ) sR ! " ", D KK !D !!r c t | dz dg }t |d }t | dk( sJ |d j dk( sJ |d j dk( sJ y ) NzPostfinance.docx"Sample ETF CH0000000001 CHF 12.345r, r5 r CH0000000001exact_isin_match)r) r lenr9 rA )tmp_pathdocxrowss r ,test_deterministic_parser_creates_candidatesrQ > se X 225Y4Z [D #D= AD t9>> 7 ~ -- - 7 ! !%7 77 7r c i G d d fd}| j d| t }t dgdd \ }}|d u sJ t | d
k( sJ d d vsJ |d
j dk( sJ t ||d
|j
d j d d
k( sJ |j
d j d d
k( sJ y )Nc e Zd Zd Zd Zd Zy)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 yNFr rX argss r __exit__zWtest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.__exit__J %r c y)Ns {"choices":[{"message":{"content":"{\"positions\":[{\"name\":\"Q ETF\",\"ticker\":\"QETF\",\"currency\":\"CHF\",\"quantity\":\"1.23\",\"asset_class\":\"etf\",\"row_number\":2,\"confidence\":\"medium\"}]}"}}]}r rW s r readzStest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.readK s _r N__name__
__module__ __qualname__rY r_ rb r r r FakeResponserT H s (/ _r rg c N | j j d d< S Nzutf-8r& rE decodereqkwargsrg captureds r fake_urlopenzNtest_qwen_extraction_creates_only_candidates_no_bookings..fake_urlopenM " 88??73~ r > &) )) ) 7 2 %% %T47# <<P Q Z Z \]` aef ff f <<@ A J J LS QUV VV Vr c t t } t | t ddd }t | | }|d dk( sJ y )Nr r9
proposed_isinproposed_exchangerA r3 )r r
rF r r cidresults r 5test_candidate_without_isin_stays_needs_manual_reviewr \ s@ manual review confirms)r r> a1
2025-12-31noT
account_id
snapshot_dater confirm)r r
rF pytestraises
ValueErrorr r r r s r ,test_ticker_alone_does_not_confirm_or_importr k s r r? zLSELECT mapping_status FROM instrument_import_candidates WHERE candidate_id=?rA confirmedzDSELECT COUNT(*) AS c FROM audit_log WHERE action='confirm_candidate'ru rJ )r r
rF r r rx )r r rows r 'test_manual_confirmation_sets_confirmedr t s PVjo DI J
,,ehkgm
n
w
w
yC K // / <<^ _ h h jkn ost tt tr c t } t | t dd }t | |ddddd t | |dd d
d }| j d
|f j
}|d dk( sJ |d dk( sJ |d dk( sJ | j d j
d dk( sJ | j d j
d dk( sJ y )NrK r r r/ r r0 )r r~ r> r r? r r zapproved importTr zWSELECT transaction_type, source_type, quantity FROM transactions WHERE transaction_id=?transaction_typeinitial_position_snapshot source_type broker_import_reviewed_snapshotquantityr1 zESELECT COUNT(*) AS c FROM alerts WHERE rule_id='cost_basis_uncertain'ru rJ zVSELECT COUNT(*) AS c FROM audit_log WHERE action='import_confirmed_candidate_snapshot')r r
rF r r r rx )r r tx_idr s r Ntest_confirmed_candidate_can_create_initial_snapshot_with_uncertain_cost_basisr } s$ nnt IN bg h /cdZfm~ IM
NE
,,psxrz
{ E E GC ! "&A AA A } !B BB B z?h && & <<_ ` i i klo ptu uu u <<p q z z | ~A B FG G G Gr c t } t | t }t j t
5 t
| |dddd d d d | j d j d dk( sJ y # 1 sw Y 1xY w) Nr r blockedTr rv ru r ) r r
rF r r r r r rx r s r 5test_unconfirmed_candidate_cannot_productively_importr s F " "a '' ' <<P Q Z Z \]` aef ff fr c | dz dz }|j j t |dg t |d sJ y )Nrepor+ z Synthetic ETF CH0000000001 CHF 1r rI )parentmkdirr) r )rN repo_docxs r 0test_no_real_data_in_repo_git_safety_placeholderr sA 6!M1I
i<=> #I ?? ?r c t }t |t dd i G fdd fd}| j d| t |d }|j
d
u sJ |j d
u sJ |j d k( sJ dd vsJ d
d vsJ |j df j }|d dk( sJ d|d v sJ y )Nz999.123r )r; r c $ e Zd Zd Zd Z fdZy)Htest_qwen_resolver_sends_sanitized_candidates_only..FakeResponsec | S rV r rW s r rY zRtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__enter__ rZ r c yr\ r r] s r r_ zQtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__exit__ r` r c b dz }dd l }|j ddd|iigi j 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)rX r& r r s r rb zMtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.read sE t wz zD ::yI 47H+I*J KLSSU Ur Nrc r s r rg r (/ Vr rg c N | j j d d< S ri rj rl s r rp zHtest_qwen_resolver_sends_sanitized_candidates_only..fake_urlopen rq r rr zqwen-3.6-agent)modelTr r& r; zYSELECT mapping_status, review_note FROM instrument_import_candidates WHERE candidate_id=?rA r3 z
manual review review_note)
r r
rF rw r
qwen_available qwen_usedconfirmed_afterr rx )ry r rp r r rg ro r s @@@r 2test_qwen_resolver_sends_sanitized_candidates_onlyr s .FakeResponsec | S rV r rW s r rY zPtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__enter__ rZ r c yr\ r r] s r r_ zOtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__exit__ r` r c b dd l }dz }|j ddd|iigi j 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 )rX r r& r s r rb zKtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.read sE a dg gD::yI 47H+I*J KLSSU Ur Nrc r s r rg r r r rg rr c S rV r )akrg s r zBtest_qwen_resolver_does_not_accept_invented_isin.. s hthv r rJ z[SELECT proposed_isin, mapping_status FROM instrument_import_candidates WHERE candidate_id=?r~ rA probable)r r
rF rw r
warningsr rx )ry r r r rg r s @@r 0test_qwen_resolver_does_not_accept_invented_isinr s r s~ "
W : >"*8W,=:}u
GWg@1:/r