Yaj( 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 Zd dZd dZd!dZd"dZd"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 /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 | dk sJ |d j dk sJ |d j dk sJ d S ) NzPostfinance.docx"Sample ETF CH0000000001 CHF 12.345r. r7 r CH0000000001exact_isin_match)r+ r lenr; rC )rI docxrowss r ,test_deterministic_parser_creates_candidatesrT > sw X 225Y4Z [ [D #D= A A AD t99>>>> 7 ~ - - - - 7 !%7 7 7 7 7 7 7r c i G d d fd}| d| t }t dgdd \ }}|d u sJ t | d
k sJ d d vsJ |d
j dk sJ t ||d
| d d d
k sJ | d d d
k sJ 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 r[ 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 rZ s r readzStest_qwen_extraction_creates_only_candidates_no_bookings..FakeResponse.readK s _ _r N__name__
__module__ __qualname__r\ rb re r r r FakeResponserW H s: (((/// _ _ _ _ _r rj c R | j d d< S Nzutf-8r( rG decodereqkwargsrj captureds r fake_urlopenzNtest_qwen_extraction_creates_only_candidates_no_bookings..fake_urlopenM ' 8??733|~~ r >>> &) ) ) ) ) 7 2 % % % %T47### <<P Q Q Z Z \ \]` aef f f f f <<@ A A J J L LS QUV V V V V V Vr c t } t | t ddd }t | | }|d dk sJ d S )Nr r;
proposed_isinproposed_exchangerC r5 )r r
rH r r cidresults r 5test_candidate_without_isin_stays_needs_manual_reviewr \ sW <PVjo DI J J J J
,,ehkgm
n
n
w
w
y
yC K / / / / <<^ _ _ h h j jkn ost t t t t t tr c ( t } t | t dd }t | |ddddd t | |dd d
d }| d
|f }|d dk sJ |d dk sJ |d dk sJ | d d dk sJ | d d dk sJ d S )NrO 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_snapshot source_type broker_import_reviewed_snapshotquantityr3 zESELECT COUNT(*) AS c FROM alerts WHERE rule_id='cost_basis_uncertain'rx rN zVSELECT COUNT(*) AS c FROM audit_log WHERE action='import_confirmed_candidate_snapshot')r r
rH r r r r{ )r r tx_idr 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 A A A A } !B B B B B z?h & & & & <<_ ` ` i i k klo ptu u u u u <<p q q z z | | ~A B FG G G G G G Gr c L t } t | t }t j t
5 t
| |dddd d d d n # 1 swxY w Y | d d dk sJ d S ) Nr r blockedTr ry rx r ) r r
rH r r r r r r{ r s r 5test_unconfirmed_candidate_cannot_productively_importr s <
>
>F "a ' ' ' ' <<P Q Q Z Z \ \]` aef f f f f f fr c | dz dz }|j t |dg t |d sJ d S )Nrepor- z Synthetic ETF CH0000000001 CHF 1r rM )parentmkdirr+ r )rI repo_docxs r 0test_no_real_data_in_repo_git_safety_placeholderr sY 6!M1I
i<=>>> #I ? ? ??? ???r c t }t |t dd i G fdd fd}| d| t |d }|j d
u sJ |j d
u sJ |j d k sJ dd vsJ d
d vsJ | df }|d dk sJ d|d v sJ d S )Nz999.123r )r= r c & e Zd Zd Zd Z fdZdS )Htest_qwen_resolver_sends_sanitized_candidates_only..FakeResponsec | S rY r rZ s r r\ zRtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__enter__ r] r c dS r_ r r` s r rb zQtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.__exit__ rc 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)r[ r( r r s r re zMtest_qwen_resolver_sends_sanitized_candidates_only..FakeResponse.read sQ t wz zD KKK::yI 47H+I*J KLLSSUU Ur Nrf r s r rj r G (((/// V V V V V V Vr rj c R | j d d< S rl rm ro s r rs zHtest_qwen_resolver_sends_sanitized_candidates_only..fake_urlopen rt r ru zqwen-3.6-agent)modelTr r( r= zYSELECT mapping_status, review_note FROM instrument_import_candidates WHERE candidate_id=?rC r5 z
manual review review_note)
r r
rH rz r
qwen_available qwen_usedconfirmed_afterr r{ )r| r rs r r rj rr r s @@@r 2test_qwen_resolver_sends_sanitized_candidates_onlyr s <.FakeResponsec | S rY r rZ s r r\ zPtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__enter__ r] r c dS r_ r r` s r rb zOtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.__exit__ rc 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 )r[ r r( r s r re zKtest_qwen_resolver_does_not_accept_invented_isin..FakeResponse.read sQ KKK a dg gD::yI 47H+I*J KLLSSUU Ur Nrf r s r rj r r r rj ru c S rY r )akrj s r zBtest_qwen_resolver_does_not_accept_invented_isin.. s hththvhv r rN z[SELECT proposed_isin, mapping_status FROM instrument_import_candidates WHERE candidate_id=?r rC probable)r r
rH rz r
warningsr r{ )r| r r r rj r s @@r 0test_qwen_resolver_does_not_accept_invented_isinr s <r s5 " " " " " "
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