File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_htb.cpython-38.pyc
U
W[v � @ s� d dd� Z ddlmZ ddlmZ ddlmZ G dd � d �ZG d
d � d ej�Z G d d
� d
ej
�Z G dd� de �Z G dd� de �Z
G dd� de �ZdS )z$Revision: 1.3 $� ���� )�unittest)�htb� )�
DummyConsumerc @ s e Zd ZdZdd� Zdd� ZdS )�
DummyClockr c C s
|| _ d S �N��time)�selfZwhen� r
�7/usr/lib/python3/dist-packages/twisted/test/test_htb.py�set s zDummyClock.setc C s | j S r r
�r r
r
r �__call__ s zDummyClock.__call__N)�__name__�
__module__� __qualname__r r r r
r
r
r r s r c @ s e Zd ZdZdZdS )�
SomeBucket�d � N)r r r Zmaxburst�rater
r
r
r r s r c @ s e Zd Zdd� Zdd� ZdS )�TestBucketBasec C s t j| _t� | _| jt _d S r )r r �
_realTimeFuncr �clockr r
r
r �setUp s zTestBucketBase.setUpc C s | j t_d S r )r r r r r
r
r �tearDown s zTestBucketBase.tearDownN)r r r r r r
r
r
r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� BucketTestsc C s t � }|�d�}| �d|� dS )z1
Testing the size of the bucket.
�� r N)r �add� assertEqual�r �b�fitr
r
r �testBucketSize% s
zBucketTests.testBucketSizec C s6 t � }|�d�}| j�d� |�d�}| �d|� dS )z2
Testing the bucket's drain rate.
r �
� N)r r r r r! r" r
r
r �testBucketDrain. s
zBucketTests.testBucketDrainc C sP t � }|�d� | j�d� |�� }| �|� | j�d� |�� }| �|� dS )z\
L{htb.Bucket.drip} returns C{True} if the bucket is empty after that drip.
r' � r& N)r r r r ZdripZ assertFalseZ
assertTrue)r r# �emptyr
r
r �test_bucketEmpty9 s
z BucketTests.test_bucketEmptyN)r r r r% r( r+ r
r
r
r r $ s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�BucketNestingTestsc C s. t �| � t� | _t| j�| _t| j�| _d S r )r r r �parent�child1�child2r r
r
r r I s
zBucketNestingTests.setUpc C s( | j �d� | j�d�}| �d|� d S )N�Z r& )r. r r/ r! �r r$ r
r
r �testBucketParentSizeP s z'BucketNestingTests.testBucketParentSizec C s< d| j _| j�d� | j�d� | j�d�}| �d|� d S )Nr r r& )r- r r. r r r r! r1 r
r
r �testBucketParentRateW s
z'BucketNestingTests.testBucketParentRateN)r r r r r2 r3 r
r
r
r r, H s r, c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�ConsumerShaperTestsc C s0 t �| � t� | _t� | _t�| j| j�| _d S r ) r r r �
underlyingr �bucketr ZShapedConsumer�shapedr r
r
r r h s
zConsumerShaperTests.setUpc C sT d}| j �d� | j�d� | j�|� | j�� | �t| j �
� �|| j j � d S )Nr& r Zdxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx) r6 r r7 �writer r ZresumeProducingr! �lenr5 �getvaluer )r Zdelta_tr
r
r �testRateo s
�z ConsumerShaperTests.testRatec C s. | � | jjd� | j�� | � | jjd� d S )Nr r )r! r6 Z _refcountr7 Z
stopProducingr r
r
r �testBucketRefs{ s
z"ConsumerShaperTests.testBucketRefsN)r r r r r; r< r
r
r
r r4 g s r4 N)� __version__Z
twisted.trialr Ztwisted.protocolsr Ztest_pcpr r ZBucketr ZTestCaser r r, r4 r
r
r
r �<module> s
$