mj"dZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZdZdZdZdZd Zd Zd Zd Zd ZdZdZdZedk(reyy)a Staged multi-merchant load test: CloudLayer Direct API -> MastercardSimulator. Each stage sends ONE payment per merchant for the first N merchants in the keys CSV, all fired at the same moment (a barrier releases every thread together), then checks the simulator's /admin/stats to confirm every merchant's payment actually arrived there. python3 scripts/bulk_load_test.py --stages 10,50,100,250,500,1000 Stdlib only. API keys are read from the CSV and never printed or written to the results file. N)ThreadPoolExecutor)datetimez:https://mercurypay.aritic.app/pn/pgpayments/api/direct/payzhttp://localhost:4444zmerchant.BULKTEST0001bulk_password_123cttjttj}|j ddd|j dt d|j dd d |j d t d d |j dt dd|j ddd|j dd|j dd|j dtd|j ddd|j d dd!|j d"dd#|jS)$N) descriptionformatter_classz--keysz/home/priyanka/bulk_keys.csvz1bulk_keys.csv (user_id,email,merchant_id,api_key))defaulthelpz--urlz/CloudLayer Direct /pay endpoint (4056 instance)z--stagesz10,50,100,250,500,1000zcomma-separated merchant countsz--pause zseconds to wait between stages)typer r z --timeout<zper-request timeout in secondsz--card5123450000000008ztest card (default approves)z--amountz10.00)r z --currencyAEDz--simz+simulator base URL, for /admin/stats checksz--stop-on-fail store_truez6stop before the next stage if this one had any failure)actionr z--yeszdon't ask for confirmationz--outz3results CSV (default: load_results_.csv)) argparseArgumentParser__doc__RawDescriptionHelpFormatter add_argument DEFAULT_URLfloat DEFAULT_SIM parse_args)ps scripts/bulk_load_test.pyrr#s GXEiEijANN8%CJ}N~NN7K6gNhNN:'?FgNhNN95";[N\NN;UB=]N^NN8%7>\N]NN:wN/NN<N/NN7K6cNdNN#L?wNxNN7<6RNSNN7D/dNe <<>ct|d5}tj|Dcgc](}|jds|jds'|*}}dddj d|Scc}w#1swY#xYw)Nnewlineapi_key merchant_idc |dS)Nr#)rs rz load_merchants..7s Am,rkey)opencsv DictReadergetsort)pathfr&rowss rload_merchantsr24ss dB [1>>!,Zai0@QUU=EYZZ[II,I- K[[[s'A7A2A2A2A72A77Bctjj||z}tjt dt jj}|jdd|ztjj|d5}tj|cdddS#1swYyxYw)N: AuthorizationzBasic timeout) urllibrequestRequestbase64 b64encodeSIM_USERSIM_PASSencodedecode add_headerurlopenjsonload)simr/reqtokenr&s rsim_getrI;s .. t ,C   z8*5<<> ? F F HENN?Hu$45   R  0Ayy|s B66B?c &d|d|d|d}tj|j|j|d|jddddd d j }t jj|j|d }|jd d|jdd|dz||d|dddddd}|jtj} t jj||j5} | j |d<tj"| j%xsd} dddt5tj| z dzd|d<t7 j9dd|d<t7| j9dd|d<t7| j9d| j9d ddd!|d<|S#1swYxYw#t j&j($rT} | j*|d< tj"| j%xsd} n#t,$ri} YnwxYwYd} ~ d} ~ wt.$r'} dt1| j2d| i} Yd} ~ ?H ::++MM$ii "'   vx  .. V DCNN>#56NN?I0C$CDXm-DRZ2bRWZ\F LLN    E 7 ^^ # #C # > 1!XXF6N::affh/%0D 1!$"3"3"5"=!EqIF<488Hb12F8 ;.{s#dAvug*Nr) threadingBarrierrrkrlrlistmap) rv merchantsrxr^t0poolresultswallrys ` `` @r run_stagerwsI/G   B I 7\4txx NPYZ[\     #D D=\\s "BBc |Dcgc] }|ddk(s |}}|Dcgc] }|ddk7s |}}|Dcgc]}|d } }td|dt|dtdt|dt|td t| d d d t| d d dt| d dt j | d td|ddt||z dd|ri} |D]B}d|dd|dxsdd|dxsd} | j | gj|dDtdt| jdD]5\} } tdt| dd | d!d"j| dd#7|r(|d$|d$z } |Dcgc]i}t|d%j|dijt|d%j|dijz d&kr|dk}}|d'jDcic]6\}}||d'j|d(z r|||d'j|d(z 8}}}td)| d*||r7td+t|d,d"j|dd-t|Std.t|Scc}wcc}wcc}wcc}wcc}}w)/Nr`successrcu ── Stage rdu merchants ──z CloudLayer success z | not success z latency ms p50 2.0fz p95 _z max z mean z wall time z.1fzs -> z payments/szhttp=r_z status=rLz msg=rbr#z failures:c t|d S)Nre)r)kvs rr'zsummarise..sBqE {rr(z z>4u × z e.g. z, total by_merchantre by_statusrz simulator +z transactions u ⚠ z3 merchants have no new simulator transaction, e.g. u8 ✓ every merchant in this stage reached the simulator)printrrmax statisticsmean setdefaultappendritemsjoinsumr-r)r^rrbeforeaftersim_okr&okfailedlatgroupsr)midsdeltamissingkv status_deltas r summarisersW 9( y 8! 9B 9 =AAhK9$ ? @ m 4JK OIC DT2d3%x $r(8K7LM N Og0-4X% .221]3CRHOOQR .221]3CRHOOQRSUVW]#XX%*+$6$<$<$>e Aq!f[FYF]F]^_abFcBc1vk266q!<<<e e  ~l^DE  F3w<.([\`\e\efmnpopfq\r[st u v; L M v;? : = ,&Xes& K K  KK K(A.K.;K c  t}|jjdDcgc]}|jst | }}t |j }t|t|kDr.tjdt|dt|dt|}td|jtdt|d|j td|d |d |jd d td |jddd|jddd|j d|j"td|j$s?t'djj)dk7rtjdt+j,j/d}|j0xsd|d}g}t3|D],\}} t5|j6d} d} t;||d| || \} }|j=| d}| r t5|j6d}t?| | || || }tA|d d!"5}tCjD|tG|d#j $}|jI|jK|ddd|r |jLrtd%| d&|d'n5|t|d(z kstOjP|j/td)|ycc}w#t8$r} td| dd\} } Yd} ~ Gd} ~ wwxYw#t8$r} td| dd} Yd} ~ /d} ~ wwxYw#1swYxYw)*N,zLargest stage is z but the CSV only has z merchantsz Endpoint : z Merchants: z in z Stages : z (z payments in total, rzs pause between stages)z Card : u…z amount  zYEvery payment is a real CloudLayer transaction (pg_transactions) for a BULKTEST merchant.zProceed? [y/N] yzAborted.z %m%d%H%M%S load_results_z.csvz /admin/statsTz (simulator stats unavailable: ))NFFwrr r) fieldnamesz Stopping: stage z had z failures (--stop-on-fail).rez Per-request results: ))rstagessplitstriprr2keysrrsysexitrrripauserYrUrVyesinputlowerrnowstrftimeout enumeraterIrFrrrextendrr*r+ DictWriterr writeheader writerows stop_on_failrksleep)rvsrrrrxr all_resultsir^rrr}rrrfailuresr0rs rmainrs- >@CG  \\^ $ $\ 2F (( 2 fXT2CKf%#5 )TXX~6FF "$ &5(965I 7#  .9 UGT65&I #sB ' %1qT+a.2E2E2G-HIA MMO KK $ % )) &ugU8*   % %sGL, L,5L17M+AN1 M:MM N%M<<NN __main__)rrr<r+rDrrrrk urllib.errorr9urllib.requestconcurrent.futuresrrrrr>r?rr2rIr~rrrrrsr%rrrs    1J % " "*ZQ F6+r zFr