j1ddlmZddlZddlmcmZddlZddl m Z ddZ dd Z d Z d Zd ZdZdZdZdZdZdZdZdZddZdZdZdS)) annotationsN)hash_keykeystrreturndictcdd|iS)N AuthorizationzBearer )rs -C:\Projects\Muse2API\tests\test_monitoring.py_bearerr s _s__ -- list[dict]c |d||}|j}d}||k}|stjd|fd||fdt jvstj|rtj|ndtj|tj|dz}dd |iz}ttj |dx}x}}| d S) Nz/admin/requestsheadersparams==z3%(py2)s {%(py2)s = %(py0)s.status_code } == %(py5)srpy0py2py5assert %(py7)spy7data) get status_code @pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _safereprAssertionError_format_explanationjson) clientadminrr @py_assert1 @py_assert4 @py_assert3 @py_format6 @py_format8s r _requestsr2 s  $eF CCA =C=C =C11=C 6688F rc $|d|ddd}|j}d}||k}|stjd|fd||fd t jvstj|rtj|nd tj|tj|d z}d d |iz}ttj |dx}x}}| } | d | d} } | j }d}||} | sddt jvstj| rtj| ndtj|tj|tj| dz} ttj | dx}x}} | d}| dd}||k}|sltjd|fd||ftj|tj|dz}dd|iz} ttj | dx}x}}d}|| v}|stjd|fd|| ftj|dt jvstj| rtj| nddz}dd|iz}ttj |dx}}|j d }| |v}|stjd|fd!| |fdt jvstj| rtj| ndd"t jvstj|rtj|nd"d#z}d$d%|iz}ttj |d}tj|d&d}t| } || k}|stjd|fd'|| ftj|d(t jvstjtrtjtnd(dt jvstj| rtj| ndtj| d)z} d*d+| iz}ttj |dx}x}} |j}d,}t#| }|||-}|j}d}||k}|s|tjd|fd.||fd/t jvstj|rtj|nd/tj|tj|d0t jvstjt"rtjt"nd0dt jvstj| rtj| ndtj|tj|tj|tj|d1 z}d2d3|iz}ttj |dx}x}x}x}x}x}}|d|- d4}d5|D}dg}||k}|sltjd|fd||ftj|tj|dz}dd|iz} ttj | dx}x}}d}|d&}||v}|sltjd|fd6||ftj|tj|dz}dd|iz} ttj | dx}x}}|d&d7}d&}||k}|sltjd8|fd9||ftj|tj|dz}dd|iz} ttj | dx}x}}|d:| d;|d|fd?||ftj|tj|dz}dd|iz} ttj | dx}x}}|j}d,}t#| }|||-}|j}d@}||k}|s|tjd|fd.||fd/t jvstj|rtj|nd/tj|tj|d0t jvstjt"rtjt"nd0dt jvstj| rtj| ndtj|tj|tj|tj|d1 z}d2d3|iz}ttj |dx}x}x}x}x}x}}|d:| d;|dAdBdC}| ddD}dB}||k}|sltjd|fd||ftj|tj|dz}dd|iz} ttj | dx}x}}|j}d,}t#| }|||-}|j}d}||k}|s|tjd|fd.||fd/t jvstj|rtj|nd/tj|tj|d0t jvstjt"rtjt"nd0dt jvstj| rtj| ndtj|tj|tj|tj|d1 z}d2d3|iz}ttj |dx}x}x}x}x}x}}|j}d:| d;}|||-}|j}d}||k}|s!tjd|fdE||fd/t jvstj|rtj|nd/tj|tj|dFt jvstj|rtj|ndFtj|tj|tj|dGz} dHdI| iz}!ttj |!dx}x}x}x}x}}|j}d,}t#| }|||-}|j}d@}||k}|s|tjd|fd.||fd/t jvstj|rtj|nd/tj|tj|d0t jvstjt"rtjt"nd0dt jvstj| rtj| ndtj|tj|tj|tj|d1 z}d2d3|iz}ttj |dx}x}x}x}x}x}}|j}d:| d;}|||-}|j}dJ}||k}|s!tjd|fdE||fd/t jvstj|rtj|nd/tj|tj|dFt jvstj|rtj|ndFtj|tj|tj|dGz} dHdI| iz}!ttj |!dx}x}x}x}x}}dS)KN /admin/keysalicelaptop)namenoterr*rrrrrrrapi_keyrzm2a-zLassert %(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.startswith }(%(py4)s) } plaintext)rrpy4py6prefixz%(py1)s == %(py4)spy1r<assert %(py6)sr=hashnot inz%(py1)s not in %(py3)srBpy3assert %(py5)srzutf-8)encoding)z%(py0)s not in %(py2)sstored)rrzassert %(py4)sr<r)z0%(py1)s == %(py6)s {%(py6)s = %(py3)s(%(py4)s) }r)rBrIr<r=assert %(py8)spy8 /v1/modelsr)z%(py12)s {%(py12)s = %(py10)s {%(py10)s = %(py2)s {%(py2)s = %(py0)s.get }(%(py4)s, headers=%(py8)s {%(py8)s = %(py5)s(%(py6)s) }) }.status_code } == %(py15)sr+r ) rrr<rr=rNpy10py12py15zassert %(py17)spy17rcg|] }|d Sr7r .0ks r z&test_key_lifecycle.."s & & &!AfI & & &r)z%(py1)s not in %(py4)s last_used_at>z%(py1)s > %(py4)sz /admin/keys/idrevokedTisz%(py1)s is %(py4)sFalice2)r`r7r7)z%(py9)s {%(py9)s = %(py7)s {%(py7)s = %(py2)s {%(py2)s = %(py0)s.delete }(%(py4)s, headers=%(py5)s) }.status_code } == %(py12)sr,rrr<rrpy9rRassert %(py14)spy14i)postr!r"r#r$r%r&r'r(r)r* startswith keys_file read_textloadsrr r patchdelete)"r+r,settingsrr-r.r/r0r1bodyr;r @py_assert5 @py_format7 @py_assert0 @py_assert2 @py_format5 @py_format4rL @py_format3 @py_format9 @py_assert7 @py_assert9 @py_assert11 @py_assert14 @py_assert13 @py_format16 @py_format18listed @py_assert6 @py_assert8 @py_assert10 @py_format13 @py_format15s" r test_key_lifecyclers M5QY7Z7Z [[A =C=C =C11=C 6688D)_d5ksI  ''   ' '' ''''''''''''9'''''9'''''' '''''''' '''''''''''''''''''''' x=)IbqbM)=M ))))))))))=M)))))=))))M))))))))))))))))))))))))))) 6 66  ) )7 ) ; ;F F """"""""""9F""""""""""""9"""""9"""""""""""""F"""""F""""""""""""""""""""""""" :f  a  (?HY,?,?? (,? ?????????? (,?????? (???????????H?????H?????????????Y?????Y??????,???????????????????????????? :RlRGI,>,>R::l,> ? ? ?R ? KRsR Ks RRRRRRRRRR KsRRRRRRRRRRRR6RRRRR6RRRRRR:RRRRlRRRRRRRRRRRGRRRRRGRRRRRRRRRRRRRIRRRRRIRRRRRR,>RRRR ?RRRR KRRRRsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRR ZZ uZ 5 5 : : < ,>R::l,> ? ? ?R ? KRsR Ks RRRRRRRRRR KsRRRRRRRRRRRR6RRRRR6RRRRRR:RRRRlRRRRRRRRRRRGRRRRRGRRRRRRRRRRRRRIRRRRRIRRRRRR,>RRRR ?RRRR KRRRRsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRR /CI//&+X>>  @ @A 6688E?6 ".h. "h .......... "h..... "....h........................... :RlRGI,>,>R::l,> ? ? ?R ? KRsR Ks RRRRRRRRRR KsRRRRRRRRRRRR6RRRRR6RRRRRR:RRRRlRRRRRRRRRRRGRRRRRGRRRRRRRRRRRRRIRRRRRIRRRRRR,>RRRR ?RRRR KRRRRsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRR =V3D 33V==3U C C CV C OVSVV OSV VVVVVVVVVV OSVVVVVVVVVVVVV6VVVVV6VVVVVV=VVVV3VVVVVVVVVVVUVVVVVUVVVVVV CVVVV OVVVVSVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV :RlRGI,>,>R::l,> ? ? ?R ? KRsR Ks RRRRRRRRRR KsRRRRRRRRRRRR6RRRRR6RRRRRR:RRRRlRRRRRRRRRRRGRRRRRGRRRRRRRRRRRRRIRRRRRIRRRRRR,>RRRR ?RRRR KRRRRsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRR =V3D 33V==3U C C CV C OVSVV OSV VVVVVVVVVV OSVVVVVVVVVVVVV6VVVVV6VVVVVV=VVVV3VVVVVVVVVVVUVVVVVUVVVVVV CVVVV OVVVVSVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVrc |j}d}|||}|j}d}||k}|s!tjd|fd||fdt jvstj|rtj|ndtj|tj|dt jvstj|rtj|ndtj|tj|tj|dz}d d |iz} ttj | dx}x}x}x}x}}|j }d}d d i}|||| }|j}d} || k} | s4tjd| fd|| fdt jvstj|rtj|ndtj|tj|dt jvstj|rtj|ndtj|tj|tj|tj| dz} dd| iz} ttj | dx}x}x}x}x}x} } dS)Nr4rPrdrz|%(py9)s {%(py9)s = %(py7)s {%(py7)s = %(py2)s {%(py2)s = %(py0)s.get }(%(py4)s, headers=%(py5)s) }.status_code } == %(py12)sr+authrfrhrir7xr9z%(py11)s {%(py11)s = %(py9)s {%(py9)s = %(py2)s {%(py2)s = %(py0)s.post }(%(py4)s, headers=%(py5)s, json=%(py7)s) }.status_code } == %(py14)srrr<rrrgpy11riassert %(py16)spy16) r r!r"r#r$r%r&r'r(r)rj) r+rr-r/rrr}rrrr @py_assert12 @py_format17s r test_keys_require_adminr4s :EmE::mT 2 2 2E 2 >E#E ># EEEEEEEEEE >#EEEEEEEEEEEE6EEEEE6EEEEEE:EEEEmEEEEEEEEEEETEEEEETEEEEEE 2EEEE >EEEE#EEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEE ;Z}Z&#Z;;}d G G GZ G SZWZZ SWZ ZZZZZZZZZZ SWZZZZZZZZZZZZZ6ZZZZZ6ZZZZZZ;ZZZZ}ZZZZZZZZZZZdZZZZZdZZZZZZZZZZ GZZZZ SZZZZWZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZZrc  |j}d}|||}|j}d}||k}|s!tjd|fd||fdt jvstj|rtj|ndtj|tj|dt jvstj|rtj|ndtj|tj|tj|dz} d d | iz} ttj | dx}x}x}x}x}}|j}d}|||}|j}d}||k}|s!tjd|fd||fdt jvstj|rtj|ndtj|tj|d t jvstj|rtj|nd tj|tj|tj|dz} d d | iz} ttj | dx}x}x}x}x}}|j}d}d }t|}|||}|j} d } | | k} | sQtjd| fd| | fdt jvstj|rtj|ndtj|tj|dt jvstjtrtjtndtj|tj|tj|tj| tj| d z}dd|iz}ttj |dx}x}x}x}x}x} x} } dt||D}gd}||k}|stjd|fd||fdt jvstj|rtj|ndtj|dz}dd|iz}ttj |dx}}dS)NrOrPrrrr+rrfrhrir,zm2a-noperd)z%(py13)s {%(py13)s = %(py11)s {%(py11)s = %(py2)s {%(py2)s = %(py0)s.get }(%(py4)s, headers=%(py9)s {%(py9)s = %(py5)s(%(py7)s) }) }.status_code } == %(py16)sr ) rrr<rrrgrpy13rzassert %(py18)spy18cg|] }|d S)key_namer rXrs r rZz9test_legacy_and_admin_keys_still_work..=s = = =qQz] = = =r)Nr,legacy)z%(py0)s == %(py3)snames)rrIrJr) r r!r"r#r$r%r&r'r(r)r r2)r+rr,r-r/rrr}rrrr @py_assert15r~r @py_format19rrvrxr0s r %test_legacy_and_admin_keys_still_workr9s :DlD::lD 1 1 1D 1 =DD = DDDDDDDDDD =DDDDDDDDDDDD6DDDDD6DDDDDD:DDDDlDDDDDDDDDDDDDDDDDDDDDDDD 1DDDD =DDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDDD :ElE::lE 2 2 2E 2 >E#E ># EEEEEEEEEE >#EEEEEEEEEEEE6EEEEE6EEEEEE:EEEElEEEEEEEEEEEEEEEEEEEEEEEE 2EEEE >EEEE#EEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEE :SlSJSGJ,?,?S::l,? @ @ @S @ LSPSS LPS SSSSSSSSSS LPSSSSSSSSSSSSS6SSSSS6SSSSSS:SSSSlSSSSSSSSSSSGSSSSSGSSSSSSJSSSS,?SSSS @SSSS LSSSSPSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSSS = =Ife$<$< = = =E----5- ----------5-------------5-----5----------------------------------rc |d|ddi}it|dddi}|d|d d d d gd }|j}d}||k}|st jd|fd||fdt jvst j|rt j |ndt j |t j |dz}dd|iz} tt j | dx}x}}t||d} | d} d}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d} d}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d} d }| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d} |dd }| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d!} d}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d"} d#}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d$} d}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d%} d}| |k} | slt jd&| fd'| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d(} d}| |k} | slt jd| fd| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d)} d*}| |u} | slt jd+| fd,| |ft j | t j |dz} dd| iz}tt j |dx} x} }| d-} d}| |u} | slt jd+| fd,| |ft j | t j |dz} dd| iz}tt j |dx} x} }dS).Nr4r7bobr9r:zcf-connecting-ipz 203.0.113.7/v1/chat/completionszgpt-4ouserhirolecontentmodelmessagesrrrrrrrrmethodPOSTr@rArCr=pathrkey_idrr_r account_id anonymousr! latency_ms>=z%(py1)s >= %(py4)s client_ipstreamFrarcerror) rjr*r r!r"r#r$r%r&r'r(r)r2)r+r,rrrr-r.r/r0r1rowrurvrwrts r test_request_loggedrAs ++mU&%+ I I N N P PCLY((L*.e D D DAdG D D Dr4xxstatusr_cg|] }|d Srr rs r rZz0test_stream_and_error_logged..frr2xx)rr!r"r#r$r%r&r'r(r)list iter_linesrjr2append_format_boolop)r+rr,rr-r.r/r0r1badstreamedrvrsrurrrrtrzrrr @py_format20rw @py_format2s r test_stream_and_error_loggedrVs v5t$556$KHKH I ILM}##}##########}############q#####q######}############################### Q\\^^  *DV,L,L+M@O@O  P PA =C=C =C11=Cfe,,RaR0MCHH8H HH  %HHHHH(=*AHSH*AS*HHHHHHHHHHHH HHHHH HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH*ASHHHHH*AHHHHSHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH } $$  $$$$$$$$$$ $$$$$ $$$$$$$$$$$$$$$$$$$$$$$$$$$$$$$ w<'<'<< ''''''''''<<'''''<''''<''''''''''''''''''''''''''' w<<< D DYvuUCCC D D DST S D SSSSSSSSSS D SSSSS DSSSS SSSSSSSSSSSSSSSSSSSSSSSSSSS D DYvuUCCC D D DXRVHXX DHX XXXXXXXXXX DHXXXXXX DXXXXHXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXsC#DDDc  |d|ddi}|j}d|d}|||}|j}d}||k} | s!t jd | fd ||fd t jvst j|rt j |nd t j |t j |d t jvst j|rt j |nd t j |t j |t j |d z} dd| iz} tt j | dx}x}x}x}x} }|dt||} d| D} d|ddg}| |k}|slt jd |fd| |ft j | t j |dz}dd|iz}tt j |dx} x}}| \}}g}|d}d}||u}|} |r|d} |d}| |k}|} | st jd|fd||ft j |t j |dz}dd|iz}| ||r`t jd |fd| |ft j | t j |dz} d d!| iz}| |t j|d"iz}d#d$|iz}tt j |dx} x}x}x}x}x} x}}g}|d}d%}||u}|} |r|d} |d}| |k}|} | st jd|fd||ft j |t j |dz}dd|iz}| ||r`t jd |fd| |ft j | t j |dz} d d!| iz}| |t j|d"iz}d#d$|iz}tt j |dx} x}x}x}x}x} x}}|d&} d'}| |k}|slt jd |fd| |ft j | t j |dz}dd|iz}tt j |dx} x}}d(t||d)*D} dg}| |k}|slt jd |fd| |ft j | t j |dz}dd|iz}tt j |dx} x}}dS)+N /v1/videosprompta waver9 /v1/videos/r_rPrrrr+rrfrhriz/v1/media/does-not-exist.pngcg|] }|d Srr rs r rZz8test_polls_flagged_and_media_skipped..os $ $ $!AfI $ $ $rr@rArCr=pollTtask_idrarrrrNrrrrrrrFrrcg|] }|d Srr rs r rZz8test_polls_flagged_and_media_skipped..ts K K K!AfI K K Krtrue hide_polls)rjr*r r!r"r#r$r%r&r'r(r)r2rr)r+rr,taskr-r/rrr}rrrrowsrurvrwrtrsubmitrsr.rrrzrrrs r $test_polls_flagged_and_media_skippedriso ;;|T88L; M M R R T TD :R0DJ00R::0$ ? ? ?R ? KRsR Ks RRRRRRRRRR KsRRRRRRRRRRRR6RRRRR6RRRRRR:RRRR0RRRRRRRRRRR$RRRRR$RRRRRR ?RRRR KRRRRsRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRRR JJ-... VU # #D $ $t $ $ $R)CtDz)C)C\(RR $(R RRRRRRRRRR $(RRRRRR $RRRR(RRRRRRRRRRRRRRRRRRRRRRRRRRRRLD&AA4<A4A<4 AAAAADOAtDzAOz$AAAAAAAAAAAA<4AAAAA<AAAA4AAAAAAAAAAAAAAAAAAAAAAAAAAAOzAAAAAOAAAAzAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAFF6&>FUF>U "FFFFFvi'8FDJF'8J'FFFFFFFFFFFF>UFFFFF>FFFFUFFFFFFFFFFFFFFFFFFFFFFFFFFF'8JFFFFF'8FFFFJFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF '?*l*?l **********?l*****?****l*************************** K Ky6JJJ K K K]P\~] K~ ]]]]]]]]]] K~]]]]] K]]]]~]]]]]]]]]]]]]]]]]]]]]]]]]]]]]rc |d||d|d|ddi}|j}d}||k}|stjd|fd ||fd t jvstj|rtj|nd tj|tj|d z}d d |iz}ttj |dx}x}}| } | d} d}| |k} | sltjd| fd| |ftj| tj|dz} dd| iz} ttj | dx} x} }g}| d} d}| |k}|} |r| d}d}||k}|} | stjd|fd| |ftj| tj|dz} dd| iz}| ||r`tjd|fd||ftj|tj|dz}dd|iz}| |tj |diz}dd |iz}ttj |dx} x}x} x}x}x}x}}| d!} d"}| |k} | sltjd| fd| |ftj| tj|dz} dd| iz} ttj | dx} x} }| d#}t|}d$d%h}||k}|stjd|fd&||fd't jvstjtrtjtnd'tj|tj|tj|d(z}d)d*|iz}ttj |dx}x}x}}d+D]}| |}t|t}|sd,d-t jvstjtrtjtnd-tj|d.t jvstjtrtjtnd.tj|d/z}ttj |dx}}d0| d1D} dd2h}| |k} | sltjd| fd| |ftj| tj|dz} dd| iz} ttj | dx} x} }| d3}t!|}d4}||k}|stjd5|fd6||fd7t jvstjt rtjt nd7tj|tj|tj|d(z}d)d*|iz}ttj |dx}x}x}}d8| d3D}t#|}d}||k}|stjd|fd&||fd9t jvstjt"rtjt"nd9tj|tj|tj|d(z}d)d*|iz}ttj |dx}x}x}}|j}d}dd:i}||||}|j}d;}||k}|s4tjd|fd<||fd=t jvstj|rtj|nd=tj|tj|d>t jvstj|rtj|nd>tj|tj|tj|tj|d?z}d@d|iz}ttj |dx}x}x}x}x}x}}dS)ANrOrP /admin/statswindow1hrrrrrrrrr@rArCr=totalrerrorsz%(py3)s == %(py6)srrrNrrrrrrr error_rateg?rp50p95z0%(py4)s {%(py4)s = %(py0)s(%(py2)s) } == %(py7)ssetrrr<rassert %(py9)srg)by_key by_accountby_model by_statusz5assert %(py5)s {%(py5)s = %(py0)s(%(py2)s, %(py3)s) } isinstancer)rrrIrch|] }|d S)r!r )rXgs r z#test_stats_shape..s 5 5 5Am  5 5 5rrrdseries;r)z0%(py4)s {%(py4)s = %(py0)s(%(py2)s) } >= %(py7)slenc3&K|] }|dV dScountNr rXps r z#test_stats_shape..&//aqz//////rsum2yr)z%(py11)s {%(py11)s = %(py9)s {%(py9)s = %(py2)s {%(py2)s = %(py0)s.get }(%(py4)s, headers=%(py5)s, params=%(py7)s) }.status_code } == %(py14)sr+r,rr)r r!r"r#r$r%r&r'r(r)r*rrrrrrr )r+rr,rr-r.r/r0r1rrurvrwrtrsrrrrzrrrrr @py_format10grouprs r test_stats_shaperws  JJ|TJ*** JJ| >5(D9I JJA =C=C =C11=C A X;$;$ ;$;$//1W://:?/////q{/a/{a////////////://///:///////////////////////////////{a/////{////a/////////////////////////////////////////////////////////// \?!c!?c !!!!!!!!!!?c!!!!!?!!!!c!!!!!!!!!!!!!!!!!!!!!!!!!!!13  1E5>1 > 1111111111 >11111111111131111131111111111 1111>11111111111111111111111111111B**E()z(D)))))))))))))))z)))))z))))))()))))))))))D)))))D))))))))))))))))))))))))))) 5 5a n 5 5 5C#sC 5 CCCCCCCCCC 5CCCCC 5CCCCCCCCCCCCCCCCCCCCCCCCCCCCCCC{!3{  !r! r !!!!!!!!!! r!!!!!!!!!!!!3!!!!!3!!!!!!{!!!! !!!!r!!!!!!!!!!!!!!!!!!!!!!!!!!!!!//1X;///43/ / /414 /1 4444444444 /14444444444443444443444444/4444 /4444144444444444444444444444444444 :`n`Xtd?|iz}tt j |dx}x}}g}|d9d)}d1}||k}|}|r|d9d0} d3} | | k}|}|st jd |fd@||ft j |t j |dAz}dBd?|iz}|||r`t jd |fdC| | ft j | t j | dDz} dEd| iz}||t j|d3iz}dFdG|iz}tt j |dx}x}x}x}x}x} x}} |d:d)}d3}||k}|slt jd |fd+||ft j |t j |d,z}d-d.|iz}tt j |dx}x}}|dH|} | dI}d3}||k}|slt jd4|fd5||ft j |t j |d,z}d-d.|iz}tt j |dx}x}}dJ| d'D}!|!|dKd dL}d}||k}|slt jd |fd+||ft j |t j |d,z}d-d.|iz}tt j |dx}x}}|!|dKd d)}d*}||k}|slt jd |fd+||ft j |t j |d,z}d-d.|iz}tt j |dx}x}}d}||!v}|st jdM|fdN||!ft j |dOtjvst j |!rt j |!ndOdPz}"dQdR|"iz}tt j |dx}}|j}dH}|||}|j}dS} || k} | s!t jd | fd || fd tjvst j |rt j |nd t j |t j |dTtjvst j |rt j |ndTt j |t j |t j | dz} dd| iz} tt j | dx}x}x}x}x} } dS)UNr4r7r5r9idler:rOrPrrrr+arfrhrirrrrrrrrrrrrrrrr_rrrrrc i|] }|d| SrVr rWs r z"test_key_usage..s222qy!222rrusagerr@rArCr= requests_24h errors_24hrlast_request_atrr\r^)rrrrc,i|]}|d|dS)r_rr )rXbs r rz"test_key_usage..s"<<.s 6 6 63CM3 6 6 6rrrrErGrrHrJrrdr)rjr*r r r!r"r#r$r%r&r'r(r)rrr)#r+rr,r5rr-r/rrr}rrrrrrrrr.r0r1rrby_namerurvrwrtrrsrzrrrawrrxs# r test_key_usager!s KK uFG;LK M M R R T TE KK uFF3CKDDDi !!A :AlA::lA . . .A . :AcA :c AAAAAAAAAA :cAAAAAAAAAAAA6AAAAA6AAAAAA:AAAAlAAAAAAAAAAAAAAAAAAAAAAAA .AAAA :AAAAcAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA ;c-cV,L,L+M@O@Oc;;-q@O P P Pc PP[c_bc \_b cccccQbcccc \_bccccc\bcccccJb ccccYb cccccYb cccYb.cccc\bcccccJb89ccccYb89cccccYb@OcccYb PcccYb \cccYb_bccccccccccccccObcccccccccccccccccccc ;;|Qh5I; J J O O Q QD JJ)T$Z))1J555 JJ|TJ*** =% 00A =C=C =C11=C 6688D22T&\222G 7 G $W -22 - 2222222222 -22222 -2222222222222222222222222222222 7 G $^ 499 4 9999999999 499999 49999999999999999999999999999999 7 G $\ 27a7 2a 7777777777 2a77777 27777a777777777777777777777777777 7 G $%6 7;!; 7! ;;;;;;;;;; 7!;;;;; 7;;;;!;;;;;;;;;;;;;;;;;;;;;;;;;;; 6?7 #AAUV;?(A(AA #(A AAAAA/@AAAA #(AAAAA7@ $AAA7@(AAAAAAAAAAAAAAA-@AAAAAAAAAAAA<muse2api dashboard)in)z,%(py1)s in %(py5)s {%(py5)s = %(py3)s.text })rBrIr) r r!r"r#r$r%r&r'r(r)rrktext) r+rr-r.r/r0r1rurvrs r test_dashboard_htmlrAs <  A =C=C =C11=C 9^ $< $ /< < / / < << <<<<< $<<<< /<<<< <<<< <<<<<<<<<<<<<<<<<<<<<<<< .8!&8 .& 8888888888 .&88888 .88888888888!88888!888888&88888888888888888888888888888rrcddl}tdD]Q}|d||}|ddvr|cS|dRt d|d ) NrrrrPr) completedfailed succeeded cancelled{Gz?ztask z did not finish)r0ranger r*sleepr()r+rrr0_rrs r _wait_taskrKsKKK 3ZZzz1114z@@EEGG >N N NKKK 4 9999 : ::rc6'ddlm}ddlm'|d|ddi}t |||d'fd }||d ||d|ddi}t |||dddl}td D]`} d t||d D} td| Drn| da| |dd} | |dd} | | k} d}| |k}| r|stjd| |fd| | |ftj| tj| tj|dz}dd|iz}t#tj|dx} x} x}x} }| |dd} d}| |k} | sltjd| fd| |ftj| tj|dz}dd|iz}t#tj|dx} x} }| |dd} d}| |k} | sltjd| fd| |ftj| tj|dz}dd|iz}t#tj|dx} x} }| |dd } d!}| |k} | sltjd| fd| |ftj| tj|dz}dd|iz}t#tj|dx} x} }| |dd"} d}| |k} | sltjd#| fd$| |ftj| tj|dz}dd|iz}t#tj|dx} x} }d%t||d&D} |dg}| |k} | sltjd| fd| |ftj| tj|dz}dd|iz}t#tj|dx} x} }|d'|d(d)i*}g}|d+} d,}| |k} | } | r&|d-}d.}||k}|} |r|d/}d}||k}|} | sdtjd| fd0| |ftj| tj|d1z}d2d3|iz}||| rtjd|fd4||ftj|tj|d5z}d6d7|iz}|||r`tjd|fd8||ftj|tj|d9z} d:d;| iz}!||!tj|diz}"d|d?D}t-|}d,}||k}|stjd|fd@||fdAt/jvstjt,rtjt,ndAtj|tj|tj|dBz}dd|iz}t#tj|dx}x}x}}dC|d?D}t-|}d.}||k}|stjd|fd@||fdAt/jvstjt,rtjt,ndAtj|tj|tj|dBz}dd|iz}t#tj|dx}x}x}}|dD|EdFddG}$g}|$dH} d,}| |k} | } | r|$dI}d.}||k}|} | stjd| fd0| |ftj| tj|d1z}d2d3|iz}||| r`tjd|fd4||ftj|tj|d5z}d6d7|iz}||tj|diz}%dJdK|%iz}&t#tj|&dx} x}x} x} x}x}x}}dS)LNr) MockDriver)UpstreamTimeoutrrrr9r_cKd)Nvideo generation timed outr )selfaccountreqrNs r boomz.test_failed_task_counts_as_error..boomso:;;;rgenerate_videodc i|] }|d| Srr rs r rz4test_failed_task_counts_as_error..sUUUA) aUUUrrrc3&K|] }|dV dS) task_statusNr rs r r z3test_failed_task_counts_as_error..s'77Aq777777rrGr!r)rr)z%(py1)s == %(py5)sz%(py5)s == %(py7)s)rBrrrrgrZrErr@rArCr=rDrrPtask_msrrcg|] }|d SrXr rs r rZz4test_failed_task_counts_as_error..s L L LQAiL L L Lrrrrrrrrrr server_errorsrrrrNrrrr)z%(py19)s == %(py22)s)rpy22z%(py24)spy24zassert %(py27)spy27c3&K|] }|dV dSrr rs r r z3test_failed_task_counts_as_error..r rrrr rc3&K|] }|dV dS)rNr rs r r z3test_failed_task_counts_as_error..s&00qq{000000rr4rPrrrrrr)muse2api.drivers.mockrMmuse2api.errorsrNrjr*rKsetattrr0rHr2allvaluesrIr"r#r'r(r)r rrr r$r%r&)(r+rr, monkeypatchrMokrTrr0rJrrur.rvrr/r1r rwrtrr-rsrrr @py_assert18 @py_assert21 @py_assert20rzrr @py_format23 @py_format25 @py_format26 @py_format28rrrrNs( @r test_failed_task_counts_as_errorrqso 000000////// \4x6J K K P P R RBvtRX&&&<<<<< $4d;;; ++lD(7K+ L L Q Q S SCvtSY'''KKK3ZZUU65V)T)T)TUUU 77777 7 7  E 4 4>- (QDTOM,JQ (,J QQcQ,Jc QQQQQQQQQQQQQ (,JcQQQQQ (QQQQ,JQQQQcQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ 4>- (7K7 (K 7777777777 (K77777 (7777K777777777777777777777777777 D ?= )5X5 )X 5555555555 )X55555 )5555X555555555555555555555555555 D ?7 #C'CC #'C CCCCCCCCCC #'CCCCCC #CCCC'CCCCCCCCCCCCCCCCCCCCCCCCCCCC D ?9 %** % ********** %***** %******************************* L L)FE("K"K"K L L L[QTUYQZP[[ LP[ [[[[[[[[[[ LP[[[[[[ L[[[[P[[[[[[[[[[[[[[[[[[[[[[[[[[[[ >5(D9I JJOOQQAKK1W:KK:?KKKKKq{KaK{a/KKKKKAo4FK!K4F!4KKKKKKKKKKKK:KKKKK:KKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK{aKKKKK{KKKKaKKKKKKKKKKKKKKKKKKKKKKKKKKK4F!KKKKK4FKKKK!KKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKKK//1X;///43/ / /414 /1 4444444444 /14444444444443444443444444/4444 /444414444444444444444444444444444400AhK000530 0 05A5 0A 5555555555 0A555555555555355555355555505555 05555A55555555555555555555555555555 JJ}eJ 4 4 9 9 ; ;I Fq I' REBB5 BAB A %BBBBB% *=BB*=*BBBBBBBBBBBB ABBBBB BBBBABBBBBBBBBBBBBBBBBBBBBBBBBBB*=BBBBB*=BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBrcKddl}ddlm}ddlm}ddlm}||j}|}dddd|d }|d d |d zd dd{V| i|dd id{V| i|ddid{V|d|j d|dzddi}| |gd{Vd| d{VdD} | d d} d } | | k} | sltjd| fd| | ftj| tj| dz} dd| iz}t!tj|dx} x} } | d d} d } | | k} | sltjd| fd| | ftj| tj| dz} dd| iz}t!tj|dx} x} } | dd} d } | | k} | sltjd| fd| | ftj| tj| dz} dd| iz}t!tj|dx} x} } |dd{Vd} d} | | k} | sltjd| fd| | ftj| tj| dz} dd| iz}t!tj|dx} x} } |d{VdS) Nr)SimpleNamespacer#) TaskStatusrz/v1/images/generationsr)rrr!rr( task_earlyrDrzno usable accountrtask_oldTrmessage interrupted)r_rfinished updated_atrc i|] }|d| SrXr rs r rz=test_task_outcome_before_row_and_backfill..s A A AAiL! A A ArrrZrr@rArCr=rrr)r0typesrsr1r$muse2api.services.tasksrtr2 finish_taskr3FAILEDbackfill_tasksr<r"r#r'r(r)statsr5)rqr0rsr$rtr6r/r7r'rrur/rvrwrts r )test_task_outcome_before_row_and_backfillrsKKK%%%%%%888888222222 *X) * *C ))++C&>s3 ( (D //,#';NPT U UUUUUUUU ''3T39l33 4 44444444 ''1T19j11 2 22222222 /Z 0AD%(1WY 4N P P PC  cU # ######## A ACIIKK&7&7&7&7&7&7%@ A A AD  m ,88 , 8888888888 ,88888 ,8888888888888888888888888888888  g &=*== &*= ========== &*====== &====*============================  M *6h6 *h 6666666666 *h66666 *6666h666666666666666666666666666))D// ! ! ! ! ! !8 ,11 , 1111111111 ,11111 ,1111111111111111111111111111111 ))++r)rrrr)rr)rrrr) __future__rbuiltinsr$_pytest.assertion.rewrite assertionrewriter"r*muse2api.auth.keysrr r2rrrrrrrr!r8r=rArKrqrr rr rsm"""""" ''''''.... WWWD[[[ ...   *YYY& ^ ^ ^aaa$ L L LF*   999;;;;#C#C#CLr