← Back
Editing: 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
Save File
Cancel