File: //usr/lib/python3/dist-packages/twisted/cred/test/__pycache__/test_cred.cpython-38.pyc
U
W[5 � @ s� d Z ddlmZmZ ddlmZmZ ddlmZm Z ddl
m Z ddl m
Z
mZ ddlmZ ddlmZ dd lmZmZmZmZ zdd
lmZ W n ek
r� d ZY nX G d d
� d
e�ZG dd� de�Z ee�G dd� dej ��Z e� e e e� G dd� dej �Z!ee!e�G dd� de��Z"eej#�G dd� de��Z$G dd� de j%�Z&G dd� de j%�Z'G d d � d e j%�Z(G d d � d e�Z)G d d!� d!e�Z*G d"d#� d#e*�Z+G d$d%� d%e*�Z,G d&d'� d'e*e)e j%�Z-G d(d)� d)e+e)e j%�Z.G d*d+� d+e,e)e j%�Z/d S ),z6
Tests for L{twisted.cred}, now with 30% more starch.
� )�absolute_import�division)� implementer� Interface)�hexlify� unhexlify)�unittest)� nativeString�
networkString)�
components)�defer)�checkers� credentials�portal�error)�cryptNc @ s e Zd ZdZdS )� ITestablez2
An interface for a theoretical protocol.
N��__name__�
__module__� __qualname__�__doc__� r r �=/usr/lib/python3/dist-packages/twisted/cred/test/test_cred.pyr s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
TestAvatarz
A test avatar.
c C s || _ d| _d| _d S )NF)�name�loggedIn� loggedOut)�selfr r r r �__init__' s zTestAvatar.__init__c C s | j r
t�d| _ d S �NT)r �AssertionError�r r r r �login- s
zTestAvatar.loginc C s
d| _ d S r )r r" r r r �logout2 s zTestAvatar.logoutN)r r r r r r# r$ r r r r r # s r c @ s e Zd ZdZdS )�Testablez-
A theoretical protocol for testing.
Nr r r r r r% 7 s r% c @ s e Zd ZdS )�IDerivedCredentialsN�r r r r r r r r& B s r& c @ s e Zd Zdd� Zdd� ZdS )�DerivedCredentialsc C s || _ || _d S �N)�username�password)r r* r+ r r r r J s zDerivedCredentials.__init__c C s
|| j kS r) )r+ �r r+ r r r �
checkPasswordO s z DerivedCredentials.checkPasswordN)r r r r r- r r r r r( G s r( c @ s e Zd ZdZdd� Zdd� ZdS )� TestRealmz
A basic test realm.
c C s
i | _ d S r) )�avatarsr" r r r r Y s zTestRealm.__init__c G sH || j kr| j | }nt|�}|| j |< |�� |d |d |�|jfS )Nr )r/ r r# r$ )r �avatarIdZmindZ
interfacesZavatarr r r �
requestAvatar] s
�zTestRealm.requestAvatarN)r r r r r r1 r r r r r. T s r. c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� CredTestsz^
Tests for the meat of L{twisted.cred} -- realms, portals, avatars, and
checkers.
c C s@ t � | _t�| j�| _t�� | _| j�dd� | j�| j� d S )N� bob� hello) r. Zrealmr �Portalr
Z'InMemoryUsernamePasswordDatabaseDontUseZcheckerZaddUser�registerCheckerr" r r r �setUpn s
zCredTests.setUpc C s. t jt jg}| j�� }| �t|�t|�� dS )z�
The checkers in a portal can check only certain types of credentials.
Since this portal has
L{checkers.InMemoryUsernamePasswordDatabaseDontUse} registered, it
N)r �IUsernamePasswordZIUsernameHashedPasswordr ZlistCredentialsInterfaces� assertEqual�sorted)r ZexpectedZgotr r r �test_listCheckersv s
�
zCredTests.test_listCheckersc C s� | � | j�t�dd�| t��}|\}}}| �|t� | �|�|�d||f � | �|j j
� | �|j j � |� | �|j j � dS )z�
Calling C{login} on a portal with correct credentials and an interface
that the portal's realm supports works.
r3 r4 �%s does not implement %sN) �successResultOfr r# r �UsernamePasswordr r9 �
assertTrue�
providedBy�originalr r �r r# Ziface�implr$ r r r �test_basicLogin� s
�
�zCredTests.test_basicLoginc C s� | � | j�tdd�| t��}|\}}}| �|t� | �|�|�d||f � | �|jj � | �|jj
� |� | �|jj
� dS )z~
Logging in with correct derived credentials and an interface
that the portal's realm supports works.
r3 r4 r< N) r= r r# r( r r9 r? r@ rA r r rB r r r �test_derivedInterface� s
�
�z CredTests.test_derivedInterfacec C s< | � | j�t�dd�| t��}| �|� | �tj |j
� dS )z�
Calling C{login} with incorrect credentials (in this case a wrong
password) causes L{error.UnauthorizedLogin} to be raised.
r3 s h3lloN� ZfailureResultOfr r# r r> r r? r9 r �UnauthorizedLogin�type�r r# r r r �test_failedLoginPassword� s
�
z"CredTests.test_failedLoginPasswordc C s< | � | j�t�dd�| t��}| �|� | �tj |j
� dS )z�
Calling C{login} with incorrect credentials (in this case no known
user) causes L{error.UnauthorizedLogin} to be raised.
s jayr4 NrF rI r r r �test_failedLoginName� s
�
z CredTests.test_failedLoginNameN)
r r r r r7 r; rD rE rJ rK r r r r r2 i s r2 c @ sF e Zd ZdddgZdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )�OnDiskDatabaseTests�� user1s pass1�� user2s pass2�� user3s pass3c C sL | � � | _t| jd��,}| jD ] \}}|�|d | d � q W 5 Q R X d S )N�wb� :�
)�mktemp�dbfile�open�users�write)r �f�u�pr r r r7 � s
zOnDiskDatabaseTests.setUpc C s$ t �d�| _| �tj| jjd� dS )zg
A missing db file will cause a permanent rejection of authorization
attempts.
z test_thisbetternoteverexist.db�userN)r
�FilePasswordDB�db� assertRaisesr rG �getUserr" r r r � test_getUserNonexistentDatabase� s z3OnDiskDatabaseTests.test_getUserNonexistentDatabasec C sP t �| j�| _| jD ]6\}}| �t| jj|�� � | � | j�|�||f� qd S r) )
r
r_ rW r` rY ra �KeyErrorrb �upperr9 �r r\ r] r r r �testUserLookup� s z"OnDiskDatabaseTests.testUserLookupc C sB t j| jdd�| _| jD ]$\}}| �| j�|�� �||f� qd S )NF)Z
caseSensitive)r
r_ rW r` rY r9 rb re rf r r r �testCaseInSensitivity� s z)OnDiskDatabaseTests.testCaseInSensitivityc sT t �� j�� _dd� � jD �}t�� fdd�|D ��}|�� jdd� � jD �� |S )Nc S s g | ]\}}t �||��qS r �r r> ��.0r\ r] r r r �
<listcomp>� s z;OnDiskDatabaseTests.testRequestAvatarId.<locals>.<listcomp>c s g | ]}t �� jj|��qS r �r Z
maybeDeferredr` �requestAvatarId�rk �cr" r r rl � s c S s g | ] \}}|�qS r r rj r r r rl � s � r
r_ rW r` rY r �
gatherResults� addCallbackr9 �r Zcreds�dr r" r �testRequestAvatarId� s �z'OnDiskDatabaseTests.testRequestAvatarIdc sT t �� j�� _dd� � jD �}t�� fdd�|D ��}|�� jdd� � jD �� |S )Nc S s g | ]\}}t �||��qS r )r �UsernameHashedPasswordrj r r r rl � s �zBOnDiskDatabaseTests.testRequestAvatarId_hashed.<locals>.<listcomp>c s g | ]}t �� jj|��qS r rm ro r" r r rl � s c S s g | ] \}}|�qS r r rj r r r rl � s rq rt r r" r �testRequestAvatarId_hashed� s ��z.OnDiskDatabaseTests.testRequestAvatarId_hashedN)
r r r rY r7 rc rg rh rv rx r r r r rL � s �
rL c @ sZ e Zd ZdddgZdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
e dkrVdZ dS )�!HashedPasswordOnDiskDatabaseTestsrM rO rQ c C s� | � � }tj|| jd�| _t|d��>}| jD ]0\}}|�|d | �|||d d� � d � q,W 5 Q R X t� }t �
|�| _ | j � | j� d S )N)�hashrS rT � rU )
rV r
r_ rz r` rX rY rZ r. r r5 �portr6 )r rW r[ r\ r] �rr r r r7 s 4 z'HashedPasswordOnDiskDatabaseTests.setUpc C s t tt|�t|���S r) )r
r r )r r\ r] �sr r r rz s z&HashedPasswordOnDiskDatabaseTests.hashc sF dd� � j D �}t�� fdd�|D ��}|�� jdd� � j D �� |S )Nc S s g | ]\}}t �||��qS r ri rj r r r rl s zIHashedPasswordOnDiskDatabaseTests.testGoodCredentials.<locals>.<listcomp>c s g | ]}� j �|��qS r )r` rn ro r" r r rl s �c S s g | ] \}}|�qS r r rj r r r rl s �rY r rr rs r9 �r Z goodCredsru r r" r �testGoodCredentials s �z5HashedPasswordOnDiskDatabaseTests.testGoodCredentialsc sT dd� � j D �}t�� fdd�|D ��}|�dd� � |�� jdd� � j D �� |S )Nc S s g | ]\}}t �||��qS r ri rj r r r rl s zOHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<listcomp>c s g | ]}� j �|d t��qS r) �r| r# r ro r" r r rl s �c S s dd� | D �S )Nc S s g | ]\}}}|j j�qS r )rA r )rk �i�a�lr r r rl s zaHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<lambda>.<locals>.<listcomp>r ��xr r r �<lambda> � zMHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<lambda>c S s g | ] \}}|�qS r r rj r r r rl s r r� r r" r �testGoodCredentials_login s �z;HashedPasswordOnDiskDatabaseTests.testGoodCredentials_loginc s@ dd� � j D �}tj� fdd�|D �dd�}|�� jtj� |S )Nc S s g | ]\}}t �|d ��qS )zwrong passwordri rj r r r rl % s �zHHashedPasswordOnDiskDatabaseTests.testBadCredentials.<locals>.<listcomp>c s g | ]}� j �|d t��qS r) r� ro r" r r rl ' s �T�Z
consumeErrors)rY r � DeferredListrs �_assertFailuresr rG )r ZbadCredsru r r" r �testBadCredentials$ s ���z4HashedPasswordOnDiskDatabaseTests.testBadCredentialsc sD � fdd�� j D �}tj� fdd�|D �dd�}|�� jtj� |S )Nc s. g | ]&\}}t �|� �d ||d d� ���qS )Nr{ )r rw rz rj r" r r rl . s
� �zKHashedPasswordOnDiskDatabaseTests.testHashedCredentials.<locals>.<listcomp>c s g | ]}� j �|d t��qS r) r� ro r" r r rl 0 s �Tr� )rY r r� rs r� r ZUnhandledCredentials)r Z hashedCredsru r r" r �testHashedCredentials- s
���z7HashedPasswordOnDiskDatabaseTests.testHashedCredentialsc G s* |D ] \}}| � |tj� |j|� qd S r) )r9 r �FAILUREZtrap)r ZfailuresZexpectedFailures�flagZfailurer r r r� 6 s z1HashedPasswordOnDiskDatabaseTests._assertFailuresNzcrypt module not available)
r r r rY r7 rz r� r� r� r� r� r �skipr r r r ry � s � ry c @ s, e Zd ZdZejdd� �Zejdd� �ZdS )�
CheckersMixina�
L{unittest.TestCase} mixin for testing that some checkers accept
and deny specified credentials.
Subclasses must provide
- C{getCheckers} which returns a sequence of
L{checkers.ICredentialChecker}
- C{getGoodCredentials} which returns a list of 2-tuples of
credential to check and avaterId to expect.
- C{getBadCredentials} which returns a list of credentials
which are expected to be unauthorized.
c c s< | � � D ].}| �� D ] \}}|�|�V }| �||� qqdS )zt
The given credentials are accepted by all the checkers, and give
the expected C{avatarID}s
N)� getCheckers�getGoodCredentialsrn r9 )r �chk�credr0 r} r r r �
test_positiveO s zCheckersMixin.test_positivec c s: | � � D ],}| �� D ] }|�|�}| �|tj�V qqdS )zI
The given credentials are rejected by all the checkers.
N)r� �getBadCredentialsrn Z
assertFailurer rG )r r� r� ru r r r �
test_negative[ s
zCheckersMixin.test_negativeN)r r r r r ZinlineCallbacksr� r� r r r r r� A s
r� c @ sD e Zd ZejZdZedd� �ZdddgZ dd� Z
d d
� Z d d � Z dS )
�HashlessFilePasswordDBMixinNc C s | S r) r r� r r r r� j r� z$HashlessFilePasswordDBMixin.<lambda>)rN � password1)rP s password2)rR � password3c c s, | j D ] \}}| �|| �|��|fV qd S r) )�_validCredentials� credClass� networkHashrf r r r r� r s z.HashlessFilePasswordDBMixin.getGoodCredentialsc c s& dD ] \}}| � || �|��V qd S )N))rN r� )rP r� )s bloofs blarf)r� r� rf r r r r� w s z-HashlessFilePasswordDBMixin.getBadCredentialsc
# sT � j p dd� }� j o � fdd�}dD �](}� �� }t|d��0}� jD ]"\}}|�|d ||� d � qDW 5 Q R X tj|||d�V � �� }t|d��0}� jD ]"\}}|�||�d | d � q�W 5 Q R X tj|d
d d ||d�V � �� }t|d��8}� jD ]*\}}|�d
|�� d ||� d � q�W 5 Q R X tj|dddd||d�V q$d S )Nc S s | S r) r r� r r r r� r� z9HashlessFilePasswordDBMixin.getCheckers.<locals>.<lambda>c s
� � |�S r) )�diskHash)r* r+ Zstoredr" r r r� � s )TFrS rT rU )�cacherz s dingle dongle � � r s zip,zap,s ,zup,� ,r{ � F)r� rV rX r� rZ r
r_ �title)r r� Z hashCheckr� �fnZfObjr\ r] r r" r r� ~ s8
& & � �� �z'HashlessFilePasswordDBMixin.getCheckers)
r r r r r> r� r� � staticmethodr� r� r� r� r� r r r r r� g s �r� c @ s e Zd Zedd� �ZdS )� LocallyHashedFilePasswordDBMixinc C s t | �S r) �r r� r r r r� � r� z)LocallyHashedFilePasswordDBMixin.<lambda>N)r r r r� r� r r r r r� � s r� c @ s* e Zd Zedd� �ZG dd� dej�ZdS )� NetworkHashedFilePasswordDBMixinc C s t | �S r) r� r� r r r r� � r� z)NetworkHashedFilePasswordDBMixin.<lambda>c @ s e Zd Zdd� ZdS )z*NetworkHashedFilePasswordDBMixin.credClassc C s t | j�|kS r) )r Zhashedr, r r r r- � s z8NetworkHashedFilePasswordDBMixin.credClass.checkPasswordN)r r r r- r r r r r� � s r� N)r r r r� r� r rw r� r r r r r� � s r� c @ s e Zd ZdS )�"HashlessFilePasswordDBCheckerTestsNr' r r r r r� � s r� c @ s e Zd ZdS )�'LocallyHashedFilePasswordDBCheckerTestsNr' r r r r r� � s r� c @ s e Zd ZdS )�'NetworkHashedFilePasswordDBCheckerTestsNr' r r r r r� � s r� )0r Z
__future__r r Zzope.interfacer r Zbinasciir r Z
twisted.trialr Ztwisted.python.compatr r
Ztwisted.pythonr Ztwisted.internetr Z twisted.credr
r r r r � ImportErrorr �objectr ZAdapterr% ZregisterAdapterr8 r& r( ZIRealmr. ZTestCaser2 rL ry r� r� r� r� r� r� r� r r r r �<module> sR
Z9E&4 � � �