o ¯b[;ã@s dZzddlZWn eydZYnweZzddlZWn ey'dZYnweZddlZddlmZddl m Z ddl m Z m Z ddlmZdZdd „ZeƒZGd d „d ƒZGd d „d eeƒZGdd„deeƒZGdd„dƒZGdd„deeƒZGdd„deeƒZdS)z& Tests for L{twisted.python.fakepwd}. éN)Úgetitem)Ú_PYPY)ÚShadowDatabaseÚ UserDatabase)ÚTestCaseiçcCs@t}tdur zt |¡Wn tyY|Sw|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}. NTé)ÚSYSTEM_UID_MAXÚpwdÚgetpwuidÚKeyError)Úguess©r úB/usr/lib/python3/dist-packages/twisted/python/test/test_fakepwd.pyÚfindInvalidUID s  üúrc@sHeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„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 Cs–tdƒ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_getpwuidCó  ôz$UserDatabaseTestsMixin.test_getpwuidcCs| t|jjt¡dS)zu I{getpwuid} raises L{KeyError} when passed a uid which does not exist in the user database. N)Ú assertRaisesr rr Ú INVALID_UID©rr r rÚtest_noSuchUIDVóz%UserDatabaseTestsMixin.test_noSuchUIDc Cs–tdƒ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. rN) rrrÚgetpwnamrrrrrrrrrr r rÚ test_getpwnam]r)z$UserDatabaseTestsMixin.test_getpwnamcCs"t}trt}| ||jjd¡dS)zN L{getpwnam} rejects a non-L{str} username with an exception. ó i-am-bytesN)Ú TypeErrorrÚ Exceptionr*rr/)rÚexc_typer r rÚtest_getpwnamRejectsBytespsz0UserDatabaseTestsMixin.test_getpwnamRejectsBytescCó| 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 rr/r,r r rÚtest_noSuchName{s ýz&UserDatabaseTestsMixin.test_noSuchNamec Csd|j}| ¡\}}}}}}}| |¡| |¡| ¡dfD]} | t| ƒt¡| t| ƒd¡qdS)zm The user record returned by I{getpwuid}, I{getpwnam}, and I{getpwall} has a length. réN) rrr r/ÚgetpwallÚassertIsInstanceÚlenÚintr© rÚdbr r!r"r#r$r%r&r'r r rÚtest_recordLength”s "þz(UserDatabaseTestsMixin.test_recordLengthc CsÜ|j}| ¡\}}}}}}}| |¡| |¡| ¡dfD]N} | | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| t| ƒtt| ƒƒ¡| t t | d¡qd 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. rrréééér8N) rrr r/r9rr;Úlistr*Ú IndexErrorrr=r r rÚtest_recordIndexableŸs"öz+UserDatabaseTestsMixin.test_recordIndexableN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r(r-r0r5r7r?rFr r r rr:s  rc@ó(eZdZdZdd„Zdd„Zdd„ZdS) ÚUserDatabaseTestsz$ Tests for L{UserDatabase}. cCstƒ|_td|_dS)zC Create a L{UserDatabase} with no user data in it. rN)rrrÚ_counterr,r r rÚsetUpºszUserDatabaseTests.setUpc Cs~|jd7_dt|jƒ}d|}d|}|j}|jd}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&©rMÚstrrÚaddUser) rÚsuffixr r!r"r#r$r%r&r r rrÁs z%UserDatabaseTests.getExistingUserInfoc CsÄd}d}d}d}d}d}d}|j}| |||||||¡| |¡g| |¡g| ¡fD]4\} | | j|¡| | j|¡| | j|¡| | j |¡| | j |¡| | j |¡| | j |¡q+dS) á 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) rrSr r/r9rrrrrrrr) rr r!r"r#r$Úhomer&r>r'r r rÚ test_addUserÒs$$ùzUserDatabaseTests.test_addUserN©rGrHrIrJrNrrZr r r rrLµs  rLc@s2eZdZdZedur dZneZdd„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 pwdcCst|j ¡ƒ|_tƒ|_dS©N)Úiterrr9Ú_usersÚsetÚ_uidsr,r r rrNús zPwdModuleTests.setUpcCs. t|jƒ}|j}||jvr|j |¡|Sq)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). )Únextr_rraÚadd)rr'r"r r rrþs   ûz"PwdModuleTests.getExistingUserInfo) rGrHrIrJr ÚskiprrNrr r r rr\îs r\c@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „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 Cs¶tdƒD]T}| ¡\ }}}}}}}} } |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. rN)rrrÚgetspnamrÚsp_namÚsp_pwdÚ sp_lstchgÚsp_minÚsp_maxÚsp_warnÚsp_inactÚ sp_expireÚsp_flag) rrr r!Ú lastChangeÚminÚmaxÚwarnÚinactÚexpireÚflagr'r r rÚ test_getspnams.  ö éz&ShadowDatabaseTestsMixin.test_getspnamcCr6)zz I{getspnam} raises L{KeyError} when passed a username which does not exist in the user database. rVN)r*r rrfr,r r rr73r.z(ShadowDatabaseTestsMixin.test_noSuchNamecCr6)zm I{getspnam} raises L{TypeError} when passed a L{bytes}, just like L{spwd.getspnam}. r1N)r*r2rrfr,r r rÚtest_getspnamBytes:r.z+ShadowDatabaseTestsMixin.test_getspnamBytescCsR|j}| ¡d}| |¡| ¡dfD]}| t|ƒt¡| t|ƒd¡qdS)zf The shadow user record returned by I{getspnam} and I{getspall} has a length. ré N)rrrfÚgetspallr:r;r<r)rr>r r'r r rr?As  þz*ShadowDatabaseTestsMixin.test_recordLengthc Csø|j}| ¡\ }}}}}}}} } | |¡| ¡dfD]^} | | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| | d|¡| | d| ¡| | d | ¡| t| ƒtt| ƒƒ¡| tt | d ¡qd 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. rrrr@rArBrCr8éryN) rrrfrzrr;rDr*rEr) rr>r r!rprqrrrsrtrurvr'r r rrFLs2 öôz-ShadowDatabaseTestsMixin.test_recordIndexableN) rGrHrIrJrwr7rxr?rFr r r rre s rec@rK) ÚShadowDatabaseTestsz& Tests for L{ShadowDatabase}. cCstƒ|_d|_dS)zE Create a L{ShadowDatabase} with no user data in it. rN)rrrMr,r r rrNts zShadowDatabaseTests.setUpc Cs¤|jd7_dt|jƒ}d|}d|}|jd}|jd}|jd}|jd}|jd}|jd } |jd } |j |||||||| | ¡ |||||||| | f S) rOrrPr r!rr@rArBrCr8rQ) rrTr r!rprqrrrsrtrurvr r rr{s       ÿz'ShadowDatabaseTests.getExistingUserInfoc Csâd}d}d}d}d}d}d}d}d} |j} |  ||||||||| ¡ |  |¡g|  ¡fD]B\} | | j|¡| | j|¡| | j|¡| | j|¡| | j |¡| | j |¡| | j |¡| | j |¡| | j | ¡q,d S) rUrVrWéé*éié r@iN)rrSrfrzrrgrhrirjrkrlrmrnro) rr r!rprqrrrsrtrurvr>r'r r rrZs,÷z ShadowDatabaseTests.test_addUserNr[r r r rr|os  r|c@sDeZdZdZedur dZn e ¡dkrdZneZdd„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 spwdrz1Cannot access shadow user database except as rootcCst|j ¡ƒ|_dSr])r^rrzr_r,r r rrN¾szSPwdModuleTests.setUpcCs t|jƒS)zF Read and return the next record from C{self._users}. )rbr_r,r r rrÁs z#SPwdModuleTests.getExistingUserInfo) rGrHrIrJÚspwdrdÚosÚgetuidrrNrr r r rr°s  r)rJr Ú_pwdÚ ImportErrorr‚Ú_spwdrƒÚoperatorrÚtwisted.python.compatrÚtwisted.python.fakepwdrrÚtwisted.trial.unittestrrrr+rrLr\rer|rr r r rÚs6  ÿ  ÿ   {9cA