File: //lib/python3/dist-packages/twisted/python/test/__pycache__/test_fakepwd.cpython-38.pyc
U
W[7 � @ s� d Z z ddlZW n ek
r( dZY nX z ddlZW n ek
rN dZY nX ddlZddlmZ ddlmZ ddl m
Z
m Z dZ dd� Z
e
� ZG d d
� d
e�ZG d d � d ee�ZG d
d� dee�ZG dd� de�ZG dd� dee�ZG dd� dee�ZdS )z&
Tests for L{twisted.python.fakepwd}.
� N)�getitem)�TestCase)� UserDatabase�ShadowDatabasei� c C sB t } tdk r>zt�| � W n tk
r2 Y q>Y q X | d8 } q | S )a�
By convention, UIDs less than 1000 are reserved for the system. A system
which allocated every single one of those UIDs would likely have practical
problems with allocating new ones, so let's assume that we'll be able to
find one. (If we don't, this will wrap around to negative values and
I{eventually} find something.)
@return: a user ID which does not exist on the local system. Or, on
systems without a L{pwd} module, return C{SYSTEM_UID_MAX}.
N� )�SYSTEM_UID_MAX�pwd�getpwuid�KeyError)Zguess� r �B/usr/lib/python3/dist-packages/twisted/python/test/test_fakepwd.py�findInvalidUID s
r
c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�UserDatabaseTestsMixina^
L{UserDatabaseTestsMixin} defines tests which apply to any user database
implementation. Subclasses should mix it in, implement C{setUp} to create
C{self.database} bound to a user database instance, and implement
C{getExistingUserInfo} to return information about a user (such information
should be unique per test method).
c
C s� t d�D ]�}| �� \}}}}}}}| j�|�} | �| j|� | �| j|� | �| j|� | �| j|� | �| j |� | �| j
|� | �| j |� qdS )zc
I{getpwuid} accepts a uid and returns the user record associated with
it.
� N) �range�getExistingUserInfo�databaser � assertEqual�pw_name� pw_passwd�pw_uid�pw_gid�pw_gecos�pw_dir�pw_shell�
�self�i�username�password�uid�gid�gecos�dir�shell�entryr r r �
test_getpwuid> s z$UserDatabaseTestsMixin.test_getpwuidc C s | � t| jjt� dS )zu
I{getpwuid} raises L{KeyError} when passed a uid which does not exist
in the user database.
N)� assertRaisesr
r r � INVALID_UID�r r r r �test_noSuchUIDR s z%UserDatabaseTestsMixin.test_noSuchUIDc
C s� t d�D ]�}| �� \}}}}}}}| j�|�} | �| j|� | �| j|� | �| j|� | �| j|� | �| j |� | �| j
|� | �| j |� qdS )zh
I{getpwnam} accepts a username and returns the user record associated
with it.
r N) r r r �getpwnamr r r r r r r r r r r r �
test_getpwnamZ s z$UserDatabaseTestsMixin.test_getpwnamc C s | � t| jjd� dS )zz
I{getpwnam} raises L{KeyError} when passed a username which does not
exist in the user database.
z.nosuchuserexiststhenameistoolongandhasinittooN)r' r
r r+ r) r r r �test_noSuchNamen s
�z&UserDatabaseTestsMixin.test_noSuchNamec
C sd | j }| �� \}}}}}}}|�|�|�|�|�� d fD ]$} | �t| �t� | �t| �d� q:dS )zm
The user record returned by I{getpwuid}, I{getpwnam}, and I{getpwall}
has a length.
r � N) r r r r+ �getpwall�assertIsInstance�len�intr �
r �dbr r r r! r" r# r$ r% r r r �test_recordLengthy s
"z(UserDatabaseTestsMixin.test_recordLengthc
C s� | j }| �� \}}}}}}}|�|�|�|�|�� d fD ]�} | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �t| �tt| ��� | �t t
| d� q:d S )
a:
The user record returned by I{getpwuid}, I{getpwnam}, and I{getpwall}
is indexable, with successive indexes starting from 0 corresponding to
the values of the C{pw_name}, C{pw_passwd}, C{pw_uid}, C{pw_gid},
C{pw_gecos}, C{pw_dir}, and C{pw_shell} attributes, respectively.
r r r � � � � r. N) r r r r+ r/ r r1 �listr' �
IndexErrorr r3 r r r �test_recordIndexable� s "z+UserDatabaseTestsMixin.test_recordIndexableN)
�__name__�
__module__� __qualname__�__doc__r& r* r, r- r5 r< r r r r r 6 s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �UserDatabaseTestsz$
Tests for L{UserDatabase}.
c C s t � | _td | _dS )zC
Create a L{UserDatabase} with no user data in it.
r N)r r r �_counterr) r r r �setUp� s zUserDatabaseTests.setUpc C s~ | j d7 _ dt| j � }d| }d| }| j }| j d }d| }d| }d| }| j�|||||||� |||||||fS ) �P
Add a new user to C{self.database} and return its information.
r �_r r i� r" r# r$ �rB �strr �addUser) r �suffixr r r r! r" r# r$ r r r r � s
z%UserDatabaseTests.getExistingUserInfoc
C s� d}d}d}d}d}d}d}| j }|�|||||||� |�|�g|�|�g|�� fD ]h\} | �| j|� | �| j|� | �| j|� | �| j |� | �| j
|� | �| j |� | �| j |� qVdS ) �
L{UserDatabase.addUser} accepts seven arguments, one for each field of
a L{pwd.struct_passwd}, and makes the new record available via
L{UserDatabase.getpwuid}, L{UserDatabase.getpwnam}, and
L{UserDatabase.getpwall}.
�alice�secr3t�{ i� zAlice,,,z /users/alicez/usr/bin/fooshN)
r rH r r+ r/ r r r r r r r r )
r r r r r! r" �homer$ r4 r% r r r � test_addUser� s& �
z UserDatabaseTests.test_addUserN�r= r> r? r@ rC r rO r r r r rA � s rA c @ s2 e Zd ZdZedkrdZneZdd� Zdd� ZdS )�PwdModuleTestsz�
L{PwdModuleTests} runs the tests defined by L{UserDatabaseTestsMixin}
against the built-in C{pwd} module. This serves to verify that
L{UserDatabase} is really a fake of that API.
Nz2Cannot verify UserDatabase against pwd without pwdc C s t | j�� �| _t� | _d S �N)�iterr r/ �_users�set�_uidsr) r r r rC � s zPwdModuleTests.setUpc C s0 t | j�}|j}|| jkr | j�|� |S q dS )z�
Read and return the next record from C{self._users}, filtering out
any records with previously seen uid values (as these cannot be
found with C{getpwuid} and only cause trouble).
N)�nextrT r rV �add)r r% r r r r r � s
z"PwdModuleTests.getExistingUserInfo) r= r> r? r@ r �skipr rC r r r r r rQ � s rQ c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ShadowDatabaseTestsMixinan
L{ShadowDatabaseTestsMixin} defines tests which apply to any shadow user
database implementation. Subclasses should mix it in, implement C{setUp} to
create C{self.database} bound to a shadow user database instance, and
implement C{getExistingUserInfo} to return information about a user (such
information should be unique per test method).
c
C s� t d�D ]�}| �� \ }}}}}}}} }
| j�|�} | �| j|� | �| j|� | �| j|� | �| j|� | �| j |� | �| j
|� | �| j |� | �| j | � | �| j
|
� qdS )zh
L{getspnam} accepts a username and returns the user record associated
with it.
r N)r r r �getspnamr �sp_nam�sp_pwd� sp_lstchg�sp_min�sp_max�sp_warn�sp_inact� sp_expire�sp_flag) r r r r �
lastChange�min�max�warn�inact�expire�flagr% r r r �
test_getspnam� s � z&ShadowDatabaseTestsMixin.test_getspnamc C s | � t| jjd� dS )zz
I{getspnam} raises L{KeyError} when passed a username which does not
exist in the user database.
rK N)r' r
r r[ r) r r r r- s z(ShadowDatabaseTestsMixin.test_noSuchNamec C sR | j }| �� d }|�|�|�� d fD ]$}| �t|�t� | �t|�d� q(dS )zf
The shadow user record returned by I{getspnam} and I{getspall} has a
length.
r � N)r r r[ �getspallr0 r1 r2 r )r r4 r r% r r r r5 s
z*ShadowDatabaseTestsMixin.test_recordLengthc C s� | j }| �� \ }}}}}}}} }
|�|�|�� d fD ]�} | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �| d |� | �| d | � | �| d |
� | �t| �tt| ��� | �tt | d
� q6d S ) aS
The shadow user record returned by I{getpwnam} and I{getspall} is
indexable, with successive indexes starting from 0 corresponding to the
values of the C{sp_nam}, C{sp_pwd}, C{sp_lstchg}, C{sp_min}, C{sp_max},
C{sp_warn}, C{sp_inact}, C{sp_expire}, and C{sp_flag} attributes,
respectively.
r r r r6 r7 r8 r9 r. � rm N)
r r r[ rn r r1 r: r' r; r ) r r4 r r re rf rg rh ri rj rk r% r r r r< ) s �z-ShadowDatabaseTestsMixin.test_recordIndexableN)r= r> r? r@ rl r- r5 r< r r r r rZ � s
rZ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ShadowDatabaseTestsz&
Tests for L{ShadowDatabase}.
c C s t � | _d| _dS )zE
Create a L{ShadowDatabase} with no user data in it.
r N)r r rB r) r r r rC H s zShadowDatabaseTests.setUpc C s� | j d7 _ dt| j � }d| }d| }| j d }| j d }| j d }| j d }| j d }| j d } | j d
}
| j�|||||||| |
� |||||||| |
f S ) rD r rE r r r r6 r7 r8 r9 r. rF ) r rI r r re rf rg rh ri rj rk r r r r P s(
� �z'ShadowDatabaseTests.getExistingUserInfoc C s� d}d}d}d}d}d}d}d}d} | j }
|
�||||||||| � |
�|�g|
�� fD ]�\} | �| j|� | �| j|� | �| j|� | �| j|� | �| j |� | �| j
|� | �| j |� | �| j |� | �| j
| � qXd S )
rJ rK rL � �* �i � r6 i� N)r rH r[ rn r r\ r] r^ r_ r` ra rb rc rd ) r r r re rf rg rh ri rj rk r4 r% r r r rO f s0 �z ShadowDatabaseTests.test_addUserNrP r r r r rp D s rp c @ sD e Zd ZdZedkrdZne�� dkr,dZneZdd� Z dd � Z
dS )
�SPwdModuleTestsz�
L{SPwdModuleTests} runs the tests defined by L{ShadowDatabaseTestsMixin}
against the built-in C{spwd} module. This serves to verify that
L{ShadowDatabase} is really a fake of that API.
Nz6Cannot verify ShadowDatabase against spwd without spwdr z1Cannot access shadow user database except as rootc C s t | j�� �| _d S rR )rS r rn rT r) r r r rC � s zSPwdModuleTests.setUpc C s
t | j�S )zF
Read and return the next record from C{self._users}.
)rW rT r) r r r r � s z#SPwdModuleTests.getExistingUserInfo) r= r> r? r@ �spwdrY �os�getuidr rC r r r r r ru � s ru )r@ r � ImportErrorrv rw �operatorr Ztwisted.trial.unittestr Ztwisted.python.fakepwdr r r r
r( �objectr rA rQ rZ rp ru r r r r �<module> s*
f< MD