OwlCyberSecurity - MANAGER
Edit File: fake_process.cpython-311.pyc
� ��bg� � � � d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G d� d� � Z dS )z FakeProcess class declaration� )� defaultdict)�deque)�Any)�Callable)�ClassVar)� DefaultDict)�Deque)�Dict)�List)�Optional)�Type)�Union� )� exceptions)�ProcessDispatcher)�ProcessRecorder)�COMMAND)�OPTIONAL_TEXT_OR_ITERABLE)�Command)�Programc �| � e Zd ZU dZeZeee ed< e Z eee ed< d d�Z d d e d e d e d ed ee dee deeeef dee dedee defd�ZeZ d!d e deddfd�Zd"d�Zdededdfd�Zde ddfd�Z d e defd�Z d e ddfd �Z e d"d �� � Z!dS )#� FakeProcessz-Main class responsible for process operations�any�program�returnNc � � t t � � | _ t � � | _ d| _ d| _ t | _ d S )NF)r r � definitions�calls�_allow_unregistered�_keep_last_processr ��selfs �o/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib/python3.11/site-packages/pytest_subprocess/fake_process.py�__init__zFakeProcess.__init__ sA � �KV� �L � L ��� &+�W�W�� �).�� �(-�� �$����� r r �command�stdout�stderr� returncode�wait�callback�callback_kwargs�signal_callback� occurrences�stdin_callablec � � |�|�t j d� � �t |t � � st |� � }t � � } | j | � ||||||||| | d� g| z � � | S )a� Main method for registering the subprocess instances. Args: command: register the command that will be faked stdout: value of the standard output stderr: value of the error output returncode: return code of the faked process wait: artificially wait for the process to finish callback: function that will be executed instead of the process callback_kwargs: keyword arguments that will be passed into callback occurrences: allow multiple usages of the same command stdin_callable: function that will interact with stdin NzbThe 'callback' and 'wait' arguments cannot be used together. Add sleep() to your callback instead.) r& r' r( r) r* r+ r, r- r/ �recorder)r �IncorrectProcessDefinition� isinstancer r r �extend) r" r& r' r( r) r* r+ r, r- r. r/ r1 s r# �registerzFakeProcess.register) s� � �6 � �� 4� �7�B�� � � �'�7�+�+� '� �g�&�&�G�"�$�$�� ���!�(�(� '� $� $�",� � (�'6�'6�&4� (� � � � � � � � �$ �r% c � � t |t � � st |� � }| j | � dg|z � � dS )z� Allow to use a real subprocess together with faked ones. Args: command: allow to execute the supplied command occurrences: allow multiple usages of the same command TN)r3 r r r4 )r" r& r. s r# � pass_commandzFakeProcess.pass_commandc sL � � �'�7�+�+� '� �g�&�&�G� ���!�(�(�$��+�)=�>�>�>�>�>r% c �. � t j | � � | S �N)r r5 r! s r# � __enter__zFakeProcess.__enter__t s � ��"�4�(�(�(�� r% �args�kwargsc �. � t j | � � d S r9 )r � deregister)r" r; r< s r# �__exit__zFakeProcess.__exit__x s � ��$�T�*�*�*�*�*r% �allowc � � || _ dS )z� Allow / block unregistered processes execution. When allowed, the real subprocesses will be called. Blocking will raise the exception. Args: allow: decide whether the unregistered process shall be allowed N)r )r" r@ s r# �allow_unregisteredz FakeProcess.allow_unregistered{ s � � $)�� � � r% c � �� t |t � � st |� � �t t t �fd�| j � � � � � � S )z� Count how many times a certain command was called. Can be used together with `fake_process.any()`. Args: command: lookup command Returns: number of times a command was called c � �� | �k S r9 � )�elem�command_instances �r# �<lambda>z(FakeProcess.call_count.<locals>.<lambda>� s �� �T�5E�-E� r% )r3 r �len�tuple�filterr )r" r& rG s @r# � call_countzFakeProcess.call_count� sQ �� � �'�7�+�+� 0� &�w� /� /� ��5�� E� E� E� E�t�z�R�R�S�S�T�T�Tr% �keepc � � || _ dS )z� Keep last process definition from being removed. That can allow / block multiple execution of the same command. Args: keep: decide whether last process shall be kept N)r )r" rM s r# �keep_last_processz FakeProcess.keep_last_process� s � � #'�� � � r% c � � | � � S )zAReturn a new FakeProcess instance to use it as a context manager.rE )�clss r# �contextzFakeProcess.context� s � � �s�u�u� r% )r N) NNr NNNNr N)r )r r )"�__name__� __module__� __qualname__�__doc__r r r r �__annotations__r r r$ r r �intr �floatr r �str�AnyTyper r5 �register_subprocessr7 r: r r? �boolrB rL rO � classmethodrR rE r% r# r r s1 � � � � � � �7�7� "�C��$�s�)� �"�"�"�'.�G�X�d�7�m� $�.�.�.�%� %� %� %� -1�,0�� $�'+�8<�.2� �-1�6� 6��6� *�6� *� 6� � 6� �u�o� 6� �8�$�6� "�$�s�G�|�"4�5�6� "�(�+�6� �6� !��*�6� �6� 6� 6� 6�p #�� �?� ?��?� �?� � ?� ?� ?� ?�"� � � �+�d� +�d� +�t� +� +� +� +�)�� )�� )� )� )� )� U�'� U�c� U� U� U� U� '�d� '�t� '� '� '� '� �� � � �[�� � r% r N) rV � collectionsr r �typingr r[ r r r r r r r r r � r �process_dispatcherr �process_recorderr �typesr r �utilsr r r rE r% r# �<module>rf s� �� #� #� #� #� #� #� #� #� � � � � � � !� !� !� !� !� !� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � 1� 1� 1� 1� 1� 1� -� -� -� -� -� -� � � � � � � ,� ,� ,� ,� ,� ,� � � � � � � � � � � � � � � � � � �H� H� H� H� H� H� H� H� H� Hr%