codekingpro/portable-devtools
114k
1�
2 ��j�3 � �l � S SK r S SKrS SKrS SKJr S SKJr S SKJr S SKJr S SKJ r " S S5 r4g) � N)�command)�
eventsequence)�hooks)�core)�scriptc �L � \ rS rSrSrSSS.S jrS rS rS rS5 r S r6S r S
r g)�context� z�7A context for testing addons, which sets up the mitmproxy.ctx module so8handlers can run as they would within mitmproxy. The context also9provides a number of helper methods for common testing scenarios.10NT)�options�loadcorec �8 � SU l [ R " 5 nU=( d [11 R R 5 n[12 R R XS9U l U R R U l U( a8 U R R R [ R " 5 5 U H( nU R R R U5 M* g ! [ a SU l [ R " 5 n N�f = f)NFT)�13event_loop)� owns_loop�asyncio�get_running_loop� RuntimeError�new_event_loop� mitmproxyr �Options�master�Master�addons�addr �Core)�selfr r r �loop�as �\D:\code\apps\devtools\python\user_packages\Python313\site-packages\mitmproxy/test/taddons.py�__init__�context.__init__ s� � � ��� ,��+�+�-�D�14 �8�Y� .� .� 6� 6� 8�� �&�&�-�-�g�-�G�� ��{�{�*�*�� � � �K�K� � � "� "�4�9�9�;� /��A� �K�K� � � "� "�1� %� �� � ,� !�D�N��)�)�+�D� ,�s �C0 �0&D�Dc � � U $ )N� )r s r � __enter__�context.__enter__% s � �� � c �� � U R ( aM U R R R 5 ( d$ U R R R 5 g)NF)r r r � is_closed�close)r �exc_type� exc_value� tracebacks r �__exit__�context.__exit__( s: � � �>�>�$�+�+�"8�"8�"B�"B�"D�"D� �K�K� "� "� (� (� *�r% c � �� # � [ R " U5 HC nU R R R X5 I Sh v�N UR15 ( d MC g g N 7f)zT16Cycles the flow through the events for the flow. Stops if the flow17is intercepted.18N)r �iterater r � invoke_addon� intercepted)r �addon�f�evts r �cycle�
context.cycle. sI � � �19 !�(�(��+�C��+�+�$�$�1�1�%�=� =� =��}�}�}�� ,� =�s �AA#�A!�A#� A#c �� � XR R ; a% U R R R U5 U R R UR 5 SS9 U( a U R R " S0 UD6 OBU R R R U[ R " [ 5 5 5 SSS5 g! , ( d f g= f)z�20A helper for testing configure methods. Modifies the registered21Options object with the given keyword arguments, then calls the22configure method on the addon with the updated value.23T)�reraiseNr" ) r r �registerr �rollback�keys�update�invoke_addon_syncr �
ConfigureHook�set)r r2 �kwargss r � configure�context.configure8 s� � � � � �*�*� *� �K�K� � � '� '�� .�
�\�\�
"�
"�6�;�;�=�$�
"�
?��� � �#�#�-�f�-�� � �"�"�4�4�U�E�<O�<O�PS�PU�<V�W� @�
?�
?�s
�'A'C�24C%c �t � [ R " US5 nUR ( a UR S $ S$ )z;25Loads a script from path, and returns the enclosed addon.26Fr N)r �Scriptr )r �path�scs r r �context.scriptF s- � � �]�]�4��
'�� !�y�y�r�y�y��|�2�d�2r% c �| � [ R " U R R SU5 nUR U5 $ )z27Invoke a command function with a list of string arguments within a command context, mimicking the actual command environment.28z test.command)r �Commandr �commands�call)r �func�args�cmds r r �context.commandM s. � � �o�o�d�k�k� 2� 2�N�D�I���x�x��~� r% )r r r )
�__name__�29__module__� __qualname__�__firstlineno__�__doc__r r# r, r5 rA r r �__static_attributes__r" r% r r r s2 � �� )-�t� &�$�� � X� 3� r% r ) r �mitmproxy.masterr �mitmproxy.optionsr r r �mitmproxy.addonsr r r r" r% r �<module>rY s) �� � � � � #� � !� #�F � F r%