Edit File: test_stream.cpython-311.pyc
� r�de � � � d Z ddlmZmZmZmZmZ ddlmZm Z m Z m Z m Z m Z ddlmZ ddlZddlmZ ddlZddlZddlmZ G d� d e� � ZdS ) zTest for object db� )�TestBase� DummyStream� make_bytes� make_object� fixture_path)�DecompressMemMapReader�FDCompressedSha1Writer� LooseObjectDB� Sha1Writer�MemoryDB�IStream)� hex_to_binN)� str_blob_type)�BytesIOc �: � e Zd ZdZdZd� fd�Zd� Zd� Zd� Zd� Z d S ) � TestStreamzTest stream classes)� i' i � c � � d S )N� )�ss �g/builddir/build/BUILD/cloudlinux-venv-1.0.2/venv/lib/python3.11/site-packages/gitdb/test/test_stream.py�<lambda>zTestStream.<lambda>) s � �4� � c � � d}t |� � |dz k s J d|t |� � fz � � �t |� � |z }t |� � D ]2}|� |� � }|||z |dz |z � }||k sJ ��3|� � � } | r | |t | � � d� k sJ �t |t � � r9t |� � � � � |� � � k sJ � ||� � |� � � } | |k sJ �t |t � � r9t |� � � � � |� � � k sJ �dS dS )a Make stream tests - the orig_stream is seekable, allowing it to be rewound and reused :param cdata: the data we expect to read from stream, the contents :param rewind_stream: function called to rewind the stream to make it ready for reuse� � z#Data must be larger than %i, was %iN)�len�range�read� isinstancer �data�compressed_bytes_read) �self�stream�cdata� rewind_stream�ns�ss�ir! �chunk�rest�rdatas r �_assert_stream_readerz TestStream._assert_stream_reader) s� � � ���5�z�z�B��F�"�"�"�$I�R�QT�UZ�Q[�Q[�L\�$\�"�"�"� ��Z�Z�2� ���r��� !� !�A��;�;�r�?�?�D��!�b�&�!�a�%�2� �-�.�E��5�=�=�=�=�=��{�{�}�}�� � .��5�#�d�)�)����-�-�-�-�-� �f� 4� 5� 5� H��v�{�{�}�}�%�%��)E�)E�)G�)G�G�G�G�G� � �f� � � � � � � � ����~�~�~�~� �f� 4� 5� 5� H��v�{�{�}�}�%�%��)E�)E�)G�)G�G�G�G�G� H� H�G�Gr c �� � t d� � D �]T}t d� � D �]@}| j D �]4}t |d�� � }|r�t j t t |� � � � }t j ||� � \ }}}|t |� � k sJ �|t k sJ �t |d�� � } | j t |� � k sJ �nLt j |� � }t ||t |� � � � }|j t |� � k sJ �| � ||d� � � t � � } | |_ | j rJ �~| j |k sJ ���6��B��Vd S )N� F�� randomize)�close_on_deletionc �, � | � d� � S )Nr )�seek)�rs r r z3TestStream.test_decompress_reader.<locals>.<lambda>g s � ����q� � � r )r � data_sizesr �zlib�compressr r r �newr �_sr- r �_m�closed) r# r2 � with_size�dsr% �zdata�typ�size�reader� test_reader�dummys r �test_decompress_readerz!TestStream.test_decompress_readerL s� � �!&�q��� " =� " =� � "�1�X�X� ! =� ! =� � �/� =� =�B� &�r�U� ;� ;� ;�E� !� 7� $� �k�-��.O�.O� P� P��,B�,F�u�N_�,`�,`�)��T�6� #�s�5�z�z� 1� 1� 1� 1� "�m� 3� 3� 3� 3� '=�U�V[�&\�&\�&\� � *�~��U��� ;� ;� ;� ;� ;� !%� �e� 4� 4��!7��?P�RU�V[�R\�R\�!]�!]�� %�y�C��J�J� 6� 6� 6� 6� �.�.�v�u�>Q�>Q�R�R�R� (�M�M�E� %�F�I� $�|�+�+�+� � �<�+<�<�<�<�<�<�A =�! =�" =� " =r c � � t � � }d|� d� � k sJ �t |� d�� � � � dk sJ �t |� d�� � � � dk sJ �|� � � }|� d� � |� � � |k sJ �d S ) Nr/ s hir )�as_hex�( r � s hi again)r �writer �sha)r# �writer�prev_shas r �test_sha_writerzTestStream.test_sha_writert s� � �� � ���F�L�L��'�'�'�'�'�'��6�:�:�Q�:�'�'�(�(�B�.�.�.�.��6�:�:�Q�:�'�'�(�(�B�.�.�.�.� �:�:�<�<��� � �[�!�!�!��z�z�|�|�x�'�'�'�'�'�'r c � � | j D �]x}t j � � \ }}t |� � }t |d�� � }t |� � |� |� � k sJ �|� � � | � t t j |� � t j |t j t t dd� � z � � }t j |t j � |� � � � }t |� � t j � |� � k sJ �t j |� � |t# j |d� � k sJ �t j |� � ��zd S )NFr0 �O_BINARYr r )r6 �tempfile�mkstempr r r rJ �close� assertRaises�OSError�os�open�O_RDONLY�getattrr �path�getsizer7 r8 �remove)r# r> �fdrZ �ostreamr! � written_datas r �test_compressed_writerz!TestStream.test_compressed_writer sA � ��/� � �B� �'�)�)�H�B��,�R�0�0�G� �b�E�2�2�2�D� �t�9�9�� � �d� 3� 3�3�3�3�3� �M�M�O�O�O� � � �g�r�x�� 4� 4� 4� ���r�{�W�R��Q�-G�-G� G�H�H�B� �7�2�r�w���t�'<�'<�=�=�L��|�$�$������(=�(=�=�=�=�=� �H�R�L�L�L� �4�=��q�#9�#9�9�9�9�9� �I�d�O�O�O�O�' � r c � � t t d� � � � }t � � }dD ]�}|� t |� � � � }|� � � }t |� � |j k sJ �|� t |j |j t |� � � � � � }|j |k sJ ���d S )N�objects)s( 888401851f15db0eed60eb1bc29dec5ddcace911s( 7bb839852ed5e3a069966281bb08d50012fb309b) r r r r$ r r r rA �storer �typer �hexsha)r# �odb�mdbrK r^ r! �dumps r �#test_decompress_reader_special_casez.TestStream.test_decompress_reader_special_case� s� � ��L�� 3� 3�4�4���j�j��B� &� &�C��j�j��C� � �1�1�G� �<�<�>�>�D��t�9�9�� �,�,�,�,� �9�9�W�W�\�7�<��� � � O� O�P�P�D��;�#�%�%�%�%�%� &� &r N) �__name__� __module__� __qualname__�__doc__r6 r- rE rN r` ri r r r r r # sz � � � � � � � �/�J�AO� � H� H� H� H�F#=� #=� #=�P (� (� (� � � �. &� &� &� &� &r r )rm �gitdb.test.libr r r r r �gitdbr r r r r r � gitdb.utilr r7 � gitdb.typr rQ rV �ior r r r r �<module>rs sB �� � �� � � � � � � � � � � � � �� � � � � � � � � � � � � � � � "� !� !� !� !� !� � � � �� � � � � � ���� � � � � � � � � � �@&� @&� @&� @&� @&�� @&� @&� @&� @&� @&r