관리-도구
편집 파일: outputcapture.cpython-311.pyc
� �܋f� � �V � d dl Z d dlZd dlmZ d dlmZ d dlmZ G d� de� � Z dS )� N)�StringIO)� TemporaryFile)�comparec � � e Zd ZdZdZdZddededefd�Zd � Zd � Z d � Z d � Z d � Z e defd�� � Zddededefd�ZdS )� OutputCapturea� A context manager for capturing output to the :any:`sys.stdout` and :any:`sys.stderr` streams. :param separate: If ``True``, ``stdout`` and ``stderr`` will be captured separately and their expected values must be passed to :meth:`~OutputCapture.compare`. :param fd: If ``True``, the underlying file descriptors will be captured, rather than just the attributes on :mod:`sys`. This allows you to capture things like subprocesses that write directly to the file descriptors, but is more invasive, so only use it when you need it. :param strip_whitespace: When ``True``, which is the default, leading and training whitespace is trimmed from both the expected and actual values when comparing. .. note:: If ``separate`` is passed as ``True``, :attr:`OutputCapture.captured` will be an empty string. NFT�separate�fd�strip_whitespacec �0 � || _ || _ || _ d S �N)r r r )�selfr r r s �k/builddir/build/BUILD/cloudlinux-venv-1.0.6/venv/lib/python3.11/site-packages/testfixtures/outputcapture.py�__init__zOutputCapture.__init__# s � � �� ���� 0�� � � � c �" � | j r:t � � | _ t � � | _ t � � | _ n9t � � | _ t � � | _ t � � | _ | � � � | S r )r r �output�stdout�stderrr �enable�r s r � __enter__zOutputCapture.__enter__( sb � � �7� %�'�/�/�D�K�'�/�/�D�K�'�/�/�D�K�K�"�*�*�D�K�"�*�*�D�K�"�*�*�D�K� � � � � � �� r c �. � | � � � d S r )�disable)r �argss r �__exit__zOutputCapture.__exit__4 s � � � � �����r c �* � | j ri| j t j f| j t j ffD ]@\ }}t j ||� � � � � t j |� � �AdS | j t _ | j t _ dS )z,Disable the output capture if it is enabled.N) r �original_stdout�sysr �original_stderrr �os�dup2�fileno�close)r �original�currents r r zOutputCapture.disable7 s� � � �7� .��%�s�z�2��%�s�z�2�&� #� #�!��'� ���'�.�.�"2�"2�3�3�3����"�"�"�"� #� #� �-�C�J� �-�C�J�J�Jr c �0 � | j ��| j rkt j t j � � � � � | _ t j t j � � � � � | _ n"t j | _ t j | _ | j r�| j r�t j | j � � � t j � � � � � t j | j � � � t j � � � � � dS | j t _ | j t _ dS | j r�t j | j � � � t j � � � � � t j | j � � � t j � � � � � dS | j xt _ t _ dS )z,Enable the output capture if it is disabled.N) r r r �dupr r r" r r r r! r r s r r zOutputCapture.enableE sx � � � � '��w� 2�')�v�c�j�.?�.?�.A�.A�'B�'B��$�')�v�c�j�.?�.?�.A�.A�'B�'B��$�$�'*�z��$�'*�z��$� �=� 6��w� )���� �*�*�,�,�c�j�.?�.?�.A�.A�B�B�B���� �*�*�,�,�c�j�.?�.?�.A�.A�B�B�B�B�B� !�[�� � !�[�� � � ��w� 6���� �*�*�,�,�c�j�.?�.?�.A�.A�B�B�B���� �*�*�,�,�c�j�.?�.?�.A�.A�B�B�B�B�B�*.�+�5�� �S�Z�Z�Zr c � � | j r;|� d� � |� � � � � � S |� � � S )Nr )r �seek�read�decode�getvalue)r �streams r �_readzOutputCapture._read\ sE � � �7� %� �K�K��N�N�N��;�;�=�=�'�'�)�)� )��?�?�$�$� $r �returnc �6 � | � | j � � S )z?A property containing any output that has been captured so far.)r. r r s r �capturedzOutputCapture.capturedc s � � �z�z�$�+�&�&�&r � �expectedr r c � � i }i }d|| j fd|| � | j � � fd|| � | j � � ffD ]E\ }}}| j r(|� � � }|� � � }||k r |||< |||< �Ft |� � dk r[t t |� � � � � d t |� � � � � d �� � t ||�� � dS )a� Compare the captured output to that expected. If the output is not the same, an :class:`AssertionError` will be raised. :param expected: A string containing the expected combined output of ``stdout`` and ``stderr``. :param stdout: A string containing the expected output to ``stdout``. :param stderr: A string containing the expected output to ``stderr``. r1 r r � r )r3 �actualN) r1 r. r r r �strip�lenr �tuple�values) r r3 r r �expected_mapping�actual_mapping�prefix� _expectedr1 s r r zOutputCapture.compareh s, � � �����X�t�}�5��6�4�:�:�d�k�#:�#:�;��6�4�:�:�d�k�#:�#:�;�, � 2� 2� '�F�I�x� �$� ,� %�O�O� -� -� �#�>�>�+�+���H�$�$�+4� ��(�)1��v�&�� � � � �A� %� %� �U�#3�#:�#:�#<�#<� =� =�a� @� � �!6�!6�!8�!8�9�9�!�<� >� >� >� >��)�.�A�A�A�A�A�Ar )FFT)r2 r2 r2 )�__name__� __module__� __qualname__�__doc__r r �boolr r r r r r. �property�strr1 r � r r r r s � � � � � �� �, �O��O�1� 1�� 1�4� 1�SW� 1� 1� 1� 1� � � �� � � .� .� .� 6� 6� 6�.%� %� %� �'�#� '� '� '� �X�'� B� B�� B�#� B�C� B� B� B� B� B� Br r ) r r �ior �tempfiler �testfixtures.comparisonr �objectr rF r r �<module>rK s� �� � � � � � � � � � � � � � � "� "� "� "� "� "� +� +� +� +� +� +�{B� {B� {B� {B� {B�F� {B� {B� {B� {B� {Br