a
    ł!fī  ć                   @   s   d dl Z G dd de jZdS )é    Nc                   @   s\   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )ŚTest_BufferedStreamc                 C   s   ddl m} |S )Nr   )ŚBufferedStream)Z gcloud.streaming.buffered_streamr   )Śselfr   © r   śf/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/streaming/test_buffered_stream.pyŚ_getTargetClass   s    z#Test_BufferedStream._getTargetClassc                 O   s   |   ” |i |¤S )N)r   )r   ŚargsŚkwr   r   r   Ś_makeOne
   s    zTest_BufferedStream._makeOnec                 C   s   ddl m} d}d}d}||}|  |||”}|  |j|u ” |  |j|” |  |jd” |  |j|d | ” |  t	||” |  
|j” |  |j|” d S ©Nr   ©ŚBytesIOó   CONTENT GOES HEREé   )Śior   r
   Ś
assertTrueŚ_streamŚassertEqualŚ
_start_posŚ_buffer_posŚ_buffered_dataŚlenZassertFalseŚstream_exhaustedŚstream_end_position©r   r   ZCONTENTŚSTARTŚBUFSIZEŚstreamZ	bufstreamr   r   r   Ś'test_ctor_start_zero_longer_than_buffer   s    z;Test_BufferedStream.test_ctor_start_zero_longer_than_bufferc                 C   s“   ddl m} d}d}d}||}| |” |  |||”}|  |j|u ” |  |j|” |  |jd” |  |j	||d  ” |  t
|t
|| ” |  |j” |  |jt
|” d S ©Nr   r   r   é   é
   )r   r   Śreadr
   r   r   r   r   r   r   r   r   r   r   r   r   r   Ś+test_ctor_start_nonzero_shorter_than_buffer   s    
z?Test_BufferedStream.test_ctor_start_nonzero_shorter_than_bufferc                 C   s@   ddl m} d}d}d}||}|  |||”}|  |j|” d S r   )r   r   r
   r   Ś_bytes_remainingr   r   r   r   Ś3test__bytes_remaining_start_zero_longer_than_buffer,   s    zGTest_BufferedStream.test__bytes_remaining_start_zero_longer_than_bufferc                 C   sR   ddl m} d}d}d}||}| |” |  |||”}|  |jt|| ” d S r   )r   r   r"   r
   r   r$   r   r   r   r   r   Ś4test__bytes_remaining_start_zero_shorter_than_buffer5   s    
zHTest_BufferedStream.test__bytes_remaining_start_zero_shorter_than_bufferc                 C   sf   ddl m} d}d}d}||}|  |||”}|  t” | d ” W d    n1 sX0    Y  d S r   ©r   r   r
   ŚassertRaisesŚ
ValueErrorr"   r   r   r   r   Śtest_read_w_none?   s    z$Test_BufferedStream.test_read_w_nonec                 C   sf   ddl m} d}d}d}||}|  |||”}|  t” | d” W d    n1 sX0    Y  d S )Nr   r   r   r   éž’’’r'   r   r   r   r   Śtest_read_w_negative_sizeI   s    z-Test_BufferedStream.test_read_w_negative_sizec                 C   sL   ddl m} d}d}d}||}|  |||”}|  | d”|d d ” d S r   )r   r   r
   r   r"   r   r   r   r   Śtest_read_from_startS   s    z(Test_BufferedStream.test_read_from_startc                 C   s~   ddl m} d}t|}d}||}| |” |  |||”}|  |j” |  |jt|” |  |j	d” |  | d”d” d S )Nr   r   r   r!   ó    )
r   r   r   r"   r
   r   r   r   r   r$   r   r   r   r   Śtest_read_exhausted\   s    
z'Test_BufferedStream.test_read_exhaustedN)Ś__name__Ś
__module__Ś__qualname__r   r
   r   r#   r%   r&   r*   r,   r-   r/   r   r   r   r   r      s   	


	r   )Z	unittest2ZTestCaser   r   r   r   r   Ś<module>   s   