File: //lib/python3/dist-packages/twisted/cred/test/__pycache__/test_cred.cpython-310.pyc
o
�b9 � @ s� d Z ddlmZmZ ddlmZmZ ddlmZm Z m
Z
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ zdd lmZ W n eyI d
ZY nw eZed d d
d�ZG dd� de�ZG dd� d�Zee�G dd� dej��Ze� eee� G dd� de j �Z ee e�G dd� d��Z ee j �G dd� d��Z!G dd� dej"�Z#G d d � d ej"�Z$G d d � d ej"�Z%G d d!� d!�Z&G d"d#� d#�Z'G d$d%� d%e'�Z(G d&d'� d'e'�Z)G d(d)� d)e'e&ej"�Z*G d*d+� d+e(e&ej"�Z+G d,d-� d-e)e&ej"�Z,G d.d/� d/ej"�Z-d
S )0z6
Tests for L{twisted.cred}, now with 30% more starch.
� )�hexlify� unhexlify)� Interface� implementer)�checkers� credentials�error�portal)�defer)�
components)�Version)�unittest)�cryptN�Twisted� � c @ � 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 rJ �d| _ d S �NT)r �r r r r �login1 s
zTestAvatar.loginc C s
d| _ d S r! )r r" r r r �logout5 �
zTestAvatar.logoutN)r r r r r r# r$ r r r r r ' s
r c @ r )�Testablez-
A theoretical protocol for testing.
Nr r r r r r& 9 s r& c @ � e Zd ZdS )�IDerivedCredentialsN�r r r r r r r r( E s r( c @ s e Zd Zdd� Zdd� ZdS )�DerivedCredentialsc C s || _ || _d S �N)�username�password)r r, r- r r r r K s
zDerivedCredentials.__init__c C s
|| j kS r+ )r- �r r- r r r �
checkPasswordO r% z DerivedCredentials.checkPasswordN)r r r r r/ r r r r r* I 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 r% zTestRealm.__init__c G sH || j v r | j | }n t|�}|| j |< |�� |d |d |�|jfS )Nr )r1 r r# r$ )r �avatarId�mind�
interfaces�avatarr r r �
requestAvatar\ s
zTestRealm.requestAvatarN)r r r r r r6 r r r r r0 S s r0 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) r0 �realmr �Portalr �'InMemoryUsernamePasswordDatabaseDontUse�checker�addUser�registerCheckerr" r r r �setUpl 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 �IUsernamePassword�IUsernameHashedPasswordr �listCredentialsInterfaces� assertEqual�sorted)r �expected�gotr r r �test_listCheckerss s
zCredTests.test_listCheckersc C s� | � | j�t�dd�| t��}|\}}}| �|t� | �|�|�|� d|� �� | �|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.
r8 r9 � does not implement N) �successResultOfr r# r �UsernamePasswordr rD �
assertTrue�
providedBy�originalr r �r r# �iface�implr$ r r r �test_basicLogin} s ��
zCredTests.test_basicLoginc C s� | � | j�tdd�| t��}|\}}}| �|t� | �|�|�|� d|� �� | �|jj � | �|jj
� |� | �|jj
� dS )z~
Logging in with correct derived credentials and an interface
that the portal's realm supports works.
r8 r9 rI N) rJ r r# r* r rD rL rM rN r r rO r r r �test_derivedInterface� s �
z CredTests.test_derivedInterfacec C �< | � | 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.
r8 s h3lloN� �failureResultOfr r# r rK r rL rD r �UnauthorizedLogin�type�r r# r r r �test_failedLoginPassword� � ��
z"CredTests.test_failedLoginPasswordc C rT )z�
Calling C{login} with incorrect credentials (in this case no known
user) causes L{error.UnauthorizedLogin} to be raised.
s jayr9 NrU rY r r r �test_failedLoginName� r[ z CredTests.test_failedLoginNameN)
r r r r r@ rH rR rS rZ r\ r r r r r7 f s
r7 c @ sD e Zd Zg d�Zdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�OnDiskDatabaseTests�)� user1s pass1)� user2s pass2)� user3s pass3c C sb | � � | _t| jd�� }| jD ]\}}|�|d | d � qW d � d S 1 s*w Y d S )N�wb� :�
)�mktemp�dbfile�open�users�write)r �f�u�pr r r r@ � 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 rW �getUserr" r r r � test_getUserNonexistentDatabase� s z3OnDiskDatabaseTests.test_getUserNonexistentDatabasec C sP t �| j�| _| jD ]\}}| �t| jj|�� � | � | j�|�||f� q
d S r+ )
r rn rf ro rh rp �KeyErrorrq �upperrD �r rk rl r r r �testUserLookup� s
�z"OnDiskDatabaseTests.testUserLookupc C sB t j| jdd�| _| jD ]\}}| �| j�|�� �||f� q d S )NF)�
caseSensitive)r rn rf ro rh rD rq rt ru r r r �testCaseInSensitivity� s �z)OnDiskDatabaseTests.testCaseInSensitivityc sT t �� j�� _dd� � jD �}t�� fdd�|D ��}|�� jdd� � jD �� |S )Nc S � g | ]
\}}t �||��qS r �r rK ��.0rk rl r r r �
<listcomp>� � z;OnDiskDatabaseTests.testRequestAvatarId.<locals>.<listcomp>c � g | ]
}t �� jj|��qS r �r
�
maybeDeferredro �requestAvatarId�r| �cr" r r r} � r~ c S � g | ]\}}|�qS r r r{ r r r r} � � ) r rn rf ro rh r
�
gatherResults� addCallbackrD �r �creds�dr r" r �testRequestAvatarId� s �z'OnDiskDatabaseTests.testRequestAvatarIdc sf t ��j��_��ddt�� � fdd��jD �}t��fdd�|D ��}|� �j
dd� �jD �� |S )N�twisted.cred.credentials�UsernameHashedPasswordc s g | ] \}}� ||��qS r r r{ )r� r r r} � � zBOnDiskDatabaseTests.testRequestAvatarId_hashed.<locals>.<listcomp>c r r r� r� r" r r r} � r~ c S r� r r r{ r r r r} � r� ) r rn rf ro � getDeprecatedModuleAttribute� _uhpVersionrh r
r� r� rD r� r �r� r r �testRequestAvatarId_hashed� s ��z.OnDiskDatabaseTests.testRequestAvatarId_hashedN)
r r r rh r@ rr rv rx r� r� r r r r r] � s r] c @ sn e Zd Zg d�Zdd� Zdedededefdd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e du r5dZ
dS dS )�!HashedPasswordOnDiskDatabaseTestsr^ c C s� | � � }tj|| jd�| _t|d��$}| jD ]\}}|�|d | �|||d d� � d � qW d � n1 s9w Y t� }t �
|�| _ | j � | j� d S )N��hashrb rc r rd )
re r rn r� ro rg rh ri r0 r r; �portr? )r rf rj rk rl �rr r r r@ s *� � z'HashedPasswordOnDiskDatabaseTests.setUprk rl �s�returnc C s. t |�d�|�d��}t|t�r|S |�d�S )N�ascii)r �decode�
isinstance�bytes�encode)r rk rl r� �hashed_passwordr r r r� s
z&HashedPasswordOnDiskDatabaseTests.hashc sF dd� � j D �}t�� fdd�|D ��}|�� jdd� � j D �� |S )Nc S ry r rz r{ r r r r} r~ zIHashedPasswordOnDiskDatabaseTests.testGoodCredentials.<locals>.<listcomp>c s g | ]}� j �|��qS r )ro r� r� r" r r r} s c S r� r r r{ r r r r} r� �rh r
r� r� rD �r � goodCredsr� r r" r �testGoodCredentials s z5HashedPasswordOnDiskDatabaseTests.testGoodCredentialsc sT dd� � j D �}t�� fdd�|D ��}|�dd� � |�� jdd� � j D �� |S )Nc S ry r rz r{ r r r r} r~ zOHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<listcomp>c � g | ]
}� j �|d t��qS r+ �r� r# r r� r" r r r} r~ c S s dd� | D �S )Nc S s g | ] \}}}|j j�qS r )rN r )r| �i�a�lr r r r} r� zaHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<lambda>.<locals>.<listcomp>r ��xr r r �<lambda> s zMHashedPasswordOnDiskDatabaseTests.testGoodCredentials_login.<locals>.<lambda>c S r� r r r{ r r r r} r� 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 )s wrong passwordrz r{ r r r r} $ s �zHHashedPasswordOnDiskDatabaseTests.testBadCredentials.<locals>.<listcomp>c r� r+ r� r� r" r r r} ( r~ T��
consumeErrors)rh r
� DeferredListr� �_assertFailuresr rW )r �badCredsr� r r" r �testBadCredentials# s ��z4HashedPasswordOnDiskDatabaseTests.testBadCredentialsc sT �� ddt�� � �fdd��jD �}tj�fdd�|D �dd�}|��jtj� |S )Nr� r� c s, g | ]\}}� |�� d ||d d� ���qS )Nr r� r{ r� r r r} 1 s �zKHashedPasswordOnDiskDatabaseTests.testHashedCredentials.<locals>.<listcomp>c r� r+ r� r� r" r r r} 5 r~ Tr� ) r� r� rh r
r� r� r� r �UnhandledCredentials)r � hashedCredsr� r r� r �testHashedCredentials- s � ��z7HashedPasswordOnDiskDatabaseTests.testHashedCredentialsc G s* |D ]\}}| � |tj� |j|� qd S r+ )rD r
�FAILURE�trap)r �failures�expectedFailures�flag�failurer r r r� ; s z1HashedPasswordOnDiskDatabaseTests._assertFailuresNzcrypt module not available)r r r rh r@ r� r� r� r� r� r� r� r �skipr r r r r� � s
�r� 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 }| �||� q qdS )zt
The given credentials are accepted by all the checkers, and give
the expected C{avatarID}s
N)� getCheckers�getGoodCredentialsr� rD )r �chk�credr2 r� r r r �
test_positiveS s � ��zCheckersMixin.test_positivec c s< � | � � D ]}| �� D ]}|�|�}| �|tj�V q qdS )zI
The given credentials are rejected by all the checkers.
N)r� �getBadCredentialsr� �
assertFailurer rW )r r� r� r� r r r �
test_negative^ s �
��zCheckersMixin.test_negativeN)r r r r r
�inlineCallbacksr� r� r r r r r� E s
r� c @ sL e Zd ZejZdZededefdd��Z g d�Z
dd� Z d d
� Z d d � Z
dS )
�HashlessFilePasswordDBMixinNr� r� c C � | S r+ r r� r r r � networkHashm s z'HashlessFilePasswordDBMixin.networkHash))r_ � password1)r` s password2)ra � password3c c s. � | j D ]\}}| �|| �|��|fV qd S r+ )�_validCredentials� credClassr� ru r r r r� w s ��z.HashlessFilePasswordDBMixin.getGoodCredentialsc c s( � dD ]\}}| � || �|��V qd S )N))r_ r� )r` r� )s bloofs blarf)r� r� ru r r r r� { s � �z-HashlessFilePasswordDBMixin.getBadCredentialsc
# s� � � j pdd� }� j o� fdd�}dD ]�}� �� }t|d�� }� jD ]\}}|�|d ||� d � q"W d � n1 s>w Y tj|||d�V � �� }t|d�� }� jD ]\}}|�||�d | d � qYW d � n1 suw Y tj|d
d d ||d�V � �� }t|d��!}� jD ]\}}|�d
|�� d ||� d � q�W d � n1 s�w Y tj|dddd||d�V qd S )Nc S r� r+ r r� r r r r� � s z9HashlessFilePasswordDBMixin.getCheckers.<locals>.<lambda>c s
� � |�S r+ )�diskHash)r, r- �storedr" r r r� � s
)TFrb rc rd )�cacher� s dingle dongle � � r s zip,zap,s ,zup,� ,r � F)r� re rg r� ri r rn �title)r r� � hashCheckr� �fn�fObjrk rl r r" r r� � s@ �
� � � � � �� �
��z'HashlessFilePasswordDBMixin.getCheckers)r r r r rK r� r� � staticmethodr� r� r� r� r� r� r r r r r� i s r� c @ s e Zd Zedd� �ZdS )� LocallyHashedFilePasswordDBMixinc C � t | �S r+ �r r� r r r r� � � z)LocallyHashedFilePasswordDBMixin.diskHashN)r r r r� r� r r r r r� � s r� c @ s4 e Zd Zededefdd��ZG dd� dej�ZdS )� NetworkHashedFilePasswordDBMixinr� r� c C r� r+ r� r� r r r r� � r� z,NetworkHashedFilePasswordDBMixin.networkHashc @ s e Zd Zdd� ZdS )z*NetworkHashedFilePasswordDBMixin.credClassc C s t | j�|kS r+ )r r- r. 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� r rK r� r r r r r� � s r� c @ r' )�"HashlessFilePasswordDBCheckerTestsNr) r r r r r� � � r� c @ r' )�'LocallyHashedFilePasswordDBCheckerTestsNr) r r r r r� � r� r� c @ r' )�'NetworkHashedFilePasswordDBCheckerTestsNr) r r r r r� � r� r� c @ s e Zd ZdZdd� ZdS )�UsernameHashedPasswordTestszo
UsernameHashedPassword is a deprecated class that is functionally
equivalent to UsernamePassword.
c C s | � ddtd� dS )zB
Tests that UsernameHashedPassword is deprecated.
r� r� z6Use twisted.cred.credentials.UsernamePassword instead.N)r� r� r" r r r �test_deprecation� s �z,UsernameHashedPasswordTests.test_deprecationN)r r r r r� r r r r r� � s r� ).r �binasciir r �zope.interfacer r � twisted.credr r r r �twisted.internetr
�twisted.pythonr �twisted.python.versionsr �
twisted.trialr
r �_crypt� ImportErrorr� r r �Adapterr& �registerAdapterrA r( r* �IRealmr0 �TestCaser7 r] r� r� r� r� r� r� r� r� r� r r r r �<module> sT � \7L$8
�
�
�