Current File : //lib/python3/dist-packages/twisted/test/__pycache__/test_htb.cpython-312.pyc
�
Ϫ�f� � �� � d dd Z ddlmZ ddlmZ ddlmZ G d� d � Z G d
� d ej � Z G d � d
ej � Z G d� de � Z G d� de � Z
G d� de � Zy)z$Revision: 1.3 $� ���� )�htb)�unittest� )�
DummyConsumerc �, � e Zd ZdZdeddfd�Zdefd�Zy)�
DummyClockr �when�returnNc � � || _ y �N��time)�selfr s �7/usr/lib/python3/dist-packages/twisted/test/test_htb.py�setzDummyClock.set
s � ��� � c � � | j S r r �r s r �__call__zDummyClock.__call__ s � ��y�y�r )�__name__�
__module__� __qualname__r �intr r � r r r
r
s& � � �D��� �� ��#� r r
c � � e Zd ZdZdZy)�
SomeBucket�d � N)r r r �maxburst�rater r r r r s � ��H� �Dr r c � � e Zd Zdd�Zdd�Zy)�TestBucketBaseNc �v � t j | _ t � | _ | j t _ y r )r r �
_realTimeFuncr
�clockr s r �setUpzTestBucketBase.setUp s" � � �X�X��� �\��
��:�:��r c �. � | j t _ y r )r& r r r s r �tearDownzTestBucketBase.tearDown s � ��%�%��r �r N)r r r r( r* r r r r$ r$ s � � �
&r r$ c �$ � e Zd Zdd�Zdd�Zdd�Zy)� BucketTestsNc �^ � t � }|j d� }| j d|� y)z1
Testing the size of the bucket.
�� r N)r �add� assertEqual�r �b�fits r �testBucketSizezBucketTests.testBucketSize$ s( � �
�L���e�e�D�k�� ����c�"r c � � t � }|j d� }| j j d� |j d� }| j d|� y)z2
Testing the bucket's drain rate.
r/ �
� N)r r0 r' r r1 r2 s r �testBucketDrainzBucketTests.testBucketDrain, sE � �
�L���e�e�D�k�� �
�
���r���e�e�D�k�� ����S�!r c �* � t � }|j d� | j j d� |j � }| j |� | j j d� |j � }| j
|� y)z\
L{htb.Bucket.drip} returns C{True} if the bucket is empty after that drip.
r8 � r7 N)r r0 r' r �drip� assertFalse�
assertTrue)r r3 �emptys r �test_bucketEmptyz BucketTests.test_bucketEmpty6 sf � �
�L�� ���b� � �
�
���q������� ���� � �
�
���r������� ���� r r+ )r r r r5 r9 r@ r r r r- r- # s � �#�"� r r- c �$ � e Zd Zdd�Zdd�Zdd�Zy)�BucketNestingTestsNc � � t j | � t � | _ t | j � | _ t | j � | _ y r )r$ r( r �parent�child1�child2r s r r( zBucketNestingTests.setUpE s9 � �� � �T�"� �l�� � ����-�� � ����-�� r c � � | j j d� | j j d� }| j d|� y )N�Z r7 )rE r0 rF r1 �r r4 s r �testBucketParentSizez'BucketNestingTests.testBucketParentSizeK s4 � � � � ������k�k�o�o�b�!�� ����S�!r c �� � d| j _ | j j d� | j j d� | j j d� }| j
d|� y )Nr r r7 )rD r" rE r0 r' r r1 rI s r �testBucketParentRatez'BucketNestingTests.testBucketParentRateQ sR � � �� � �� � � ���� � �
�
���r���k�k�o�o�c�"��
����S�!r r+ )r r r r( rJ rL r r r rB rB D s � �.� "� "r rB c �$ � e Zd Zdd�Zdd�Zdd�Zy)�ConsumerShaperTestsNc �� � t j | � t � | _ t � | _ t
j | j | j
� | _ y r ) r$ r( r �
underlyingr �bucketr �ShapedConsumer�shapedr s r r( zConsumerShaperTests.setUpc s> � �� � �T�"�'�/��� �l�� ��(�(�� � �$�+�+�F�� r c �r � d}| j j d� | j j d� | j j |� | j j
� | j t | j j � � || j j z � y )Nr7 r �dxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx) rQ r0 rS �writer' r �resumeProducingr1 �lenrP �getvaluer" )r �delta_ts r �testRatez ConsumerShaperTests.testRatei s| � � �� � � ���� � � � ���)�$� �
�
���w� � � � �#�#�%� ����T�_�_� 5� 5� 7�8�'�D�K�K�DT�DT�:T�Ur c �� � | j | j j d� | j j � | j | j j d� y )Nr r )r1 rQ � _refcountrS �
stopProducingr s r �testBucketRefsz"ConsumerShaperTests.testBucketRefss sF � � ������.�.��2� � � �!�!�#� ������.�.��2r r+ )r r r r( r[ r_ r r r rN rN b s � �G� V�3r rN N)� __version__�twisted.protocolsr �
twisted.trialr �test_pcpr r
�Bucketr �TestCaser$ r- rB rN r r r �<module>rf sm �� !��B�'� � !� "� #�� �
����
�
&�X�&�&� &� �.� �B"� � "�<3�.� 3r
Mini Shell