Your IP : 216.73.216.213
B
;��]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 �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_htb.py�set s zDummyClock.setc C s | j S )N)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 )N)r r �
_realTimeFuncr �clock)r
r r r �setUp s zTestBucketBase.setUpc C s | j t_d S )N)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.
i� 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.
i� �
� N)r r 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 )N)r r r �parent�child1�child2)r
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 )r
r r r r �testBucketParentRateW s
z'BucketNestingTests.testBucketParentRateN)r r r r r, r- 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 )N) r r r �
underlyingr �bucketr ZShapedConsumer�shaped)r
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) r0 r r1 �writer r
ZresumeProducingr �lenr/ �getvaluer )r
Zdelta_tr r r �testRateo s
z ConsumerShaperTests.testRatec C s. | � | jjd� | j�� | � | jjd� d S )Nr r )r r0 Z _refcountr1 Z
stopProducing)r
r r r �testBucketRefs{ s
z"ConsumerShaperTests.testBucketRefsN)r r r r r5 r6 r r r r r. g s r. N)� __version__Z
twisted.trialr Ztwisted.protocolsr Ztest_pcpr r ZBucketr ZTestCaser r r' r. r r r r �<module> s
$