File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_randbytes.cpython-38.pyc
U
W[
� @ s| d Z ddlmZmZ ddlmZ ddlmZ G dd� de�Z G dd� de ej
�Z G d d
� d
e ej �Z
G d d � d e ej �Zd
S )z-
Test cases for L{twisted.python.randbytes}.
� )�division�absolute_import)�unittest)� randbytesc @ s e Zd ZdZdd� ZdS )�SecureRandomTestCaseBasez1
Base class for secureRandom test cases.
c C sP t dd�D ]@}||�}| �t|�|� ||�}| �t|�|� | �||� q
dS )a|
The given random bytes source should return the number of bytes
requested each time it is called and should probably not return the
same bytes on two consecutive calls (although this is a perfectly
legitimate occurrence and rejecting it may generate a spurious failure
-- maybe we'll get lucky and the heat death with come first).
� � N)�range� assertEqual�lenZassertNotEqual)�self�source�nbytes�s�s2� r �=/usr/lib/python3/dist-packages/twisted/test/test_randbytes.py�_check s z SecureRandomTestCaseBase._checkN)�__name__�
__module__� __qualname__�__doc__r r r r r r s r c @ s e Zd ZdZdd� ZdS )�SecureRandomTestsz4
Test secureRandom under normal conditions.
c C s | � tj� dS )z�
L{randbytes.secureRandom} should return a string of the requested
length and make some effort to make its result otherwise unpredictable.
N)r r � secureRandom�r r r r � test_normal+ s z SecureRandomTests.test_normalN)r r r r r r r r r r & s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � ConditionalSecureRandomTestsz<
Test random sources one by one, then remove it to.
c C s t �� | _dS )zJ
Create a L{randbytes.RandomFactory} to use in the tests.
N)r �
RandomFactory�factoryr r r r �setUp: s z"ConditionalSecureRandomTests.setUpc C s t �� �dS )zL
A factory raising an error when a source is not available.
N)r ZSourceNotAvailable)r r r r r � errorFactoryA s z)ConditionalSecureRandomTests.errorFactoryc C s | � | jj� dS )zy
L{RandomFactory._osUrandom} should work as a random source whenever
L{os.urandom} is available.
N)r r �
_osUrandomr r r r �test_osUrandomH s z+ConditionalSecureRandomTests.test_osUrandomc sN � j � j_� �tj� jjd� � fdd�}� �tdt |�}� �
t |�d� dS )zt
Remove all secure sources and assert it raises a failure. Then try the
fallback parameter.
� c s � j jddd�S )Nr# T)Zfallback)r r r r r r �wrapperX s zBConditionalSecureRandomTests.test_withoutAnything.<locals>.wrapperzPurandom unavailable - proceeding with non-cryptographically secure random sourceN) r r r! Z assertRaisesr ZSecureRandomNotAvailabler Z assertWarns�RuntimeWarning�__file__r
r )r r$ r r r r �test_withoutAnythingP s
� �z1ConditionalSecureRandomTests.test_withoutAnythingN)r r r r r r r" r' r r r r r 4 s
r c @ s e Zd ZdZdd� Zdd� ZdS )�RandomBaseTestsz%
'Normal' random test cases.
c C s | � tj� dS )z"
Test basic case.
N)r r �insecureRandomr r r r r i s zRandomBaseTests.test_normalc C s t �� }d|_| �|j� dS )zG
Test C{insecureRandom} without C{random.getrandbits}.
N)r r Z getrandbitsr r) )r r r r r �test_withoutGetrandbitsp s z'RandomBaseTests.test_withoutGetrandbitsN)r r r r r r* r r r r r( d s r( N)r Z
__future__r r Z
twisted.trialr Ztwisted.pythonr �objectr ZTestCaser ZSynchronousTestCaser r( r r r r �<module> s �0