@OOf lddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z GddZ y) N)command) eventsequence)hooks)core)scriptcBeZdZdZddddZdZdZdZd Zd Z d Z y) contextz A context for testing addons, which sets up the mitmproxy.ctx module so handlers can run as they would within mitmproxy. The context also provides a number of helper methods for common testing scenarios. NT)optionsloadcorec tj}|xstj j }tjj|||_|jj |_|r7|jjjtj|D]'}|jjj|)y#t$rtj}YwxYw)N) event_loop) asyncioget_running_loop RuntimeErrornew_event_loop mitmproxyr OptionsmasterMasteraddonsaddrCore)selfr r rloopas W/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/test/taddons.py__init__zcontext.__init__s ,++-D8Y..668&&--g$-G {{**  KK   " "499; / &A KK   " "1 % & ,))+D ,sCC?>C?c|S)N)rs r __enter__zcontext.__enter__#s cy)NFr)rexc_type exc_value tracebacks r__exit__zcontext.__exit__&sr!cKtj|D]>}|jjj ||d{|j s>yy7w)zl Cycles the flow through the events for the flow. Stops if the flow is intercepted. N)riteraterr invoke_addon intercepted)raddonfevts rcyclez context.cycle)sO !((+ C++$$11%= = =}}  =sAAAAAc ||jjvr%|jjj||jj |j d5|r|jj di|nA|jjj|tjtdddy#1swYyxYw)z A helper for testing configure methods. Modifies the registered Options object with the given keyword arguments, then calls the configure method on the addon with the updated value. T)reraiseNr) rrregisterr rollbackkeysupdateinvoke_addon_syncr ConfigureHookset)rr+kwargss r configurezcontext.configure3s  ** * KK   ' ' . \\ " "6;;=$ " ? X# ##-f- ""44UErLs)#!#AAr!