o ¯b“¸ã@sÜdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddlZ ddlZ ddlZ ddlZ ddlZ ddl mZmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"Gdd„de!j#ƒZ$Gdd„dej%j&ƒZ'Gdd„dƒZ(Gdd„dƒZ)Gdd„dej*ƒZ+Gdd„de!j#ƒZ,Gdd„dej-ƒZ.Gdd„de.ƒZ/Gdd„dej0ƒZ1Gd d!„d!e!j#ƒZ2e ej3ƒGd"d#„d#ƒƒZ4Gd$d%„d%ƒZ5d&d'„Z6Gd(d)„d)e!j#ƒZ7Gd*d+„d+e!j#ƒZ8Gd,d-„d-ƒZ9Gd.d/„d/e!j#ƒZ:Gd0d1„d1ƒZ;d2Zd3Z?d4Z@e"j=e@e>d3ZAGd5d6„d6e;e!j#ƒZBGd7d8„d8e;e!j#ƒZCGd9d:„d:e1ƒZDGd;d<„d„d>eCƒZFGd?d@„d@e1ƒZGGdAdB„dBeBƒZHGdCdD„dDeCƒZIGdEdF„dFe!j#ƒZJGdGdH„dHe!j#ƒZKdS)Iz- Test cases for L{twisted.mail.pop3} module. éN)Ú OrderedDict)Úmd5)ÚBytesIO)Ú implementer)ÚcredÚinternetÚmail)Údefer)Úpop3)Úloopback)Úfailure)ÚLineSendingProtocol)ÚunittestÚutilc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)Ú UtilityTestszl Test the various helper functions and classes used by the POP3 server protocol implementation. cCs¤g}tt gd¢¡ƒ}t |j|d¡}t|ƒ}| |g¡t|ƒ| |g¡t|ƒ| |g¡t|ƒ| |gd¢¡tdƒD]}t|ƒqA| |gd¢¡dS)zº Test creating a LineBuffer and feeding it some lines. The lines should build up in its internal buffer for a while and then get spat out to the writer. )Ú012Ú345Ú6Ú7Ú8Ú9é)rrré)rrrrrrrrN) ÚiterÚ itertoolsÚcycler Ú_IteratorBufferÚextendÚ assertEqualÚnextÚrange)ÚselfÚoutputÚinputÚcÚiÚn©r'ú=/usr/lib/python3/dist-packages/twisted/mail/test/test_pop3.pyÚtest_LineBuffering's     zUtilityTests.test_LineBufferingcCs>g}tgd¢ƒ}t |j|d¡}|D]}q| |gd¢¡dS)z„ Test that a LineBuffer flushes everything when its iterator is exhausted, and itself raises StopIteration. )ÚaÚbr$rN)rr rrr)r!r"r#r$r%r'r'r(Útest_FinishLineBuffering<s  z%UtilityTests.test_FinishLineBufferingcCs| t d¡d¡dS)zb Test that the thing that spits out POP3 'success responses' works right. sGreat.s +OK Great. N)rr ÚsuccessResponse©r!r'r'r(Útest_SuccessResponseFormatterHsz*UtilityTests.test_SuccessResponseFormattercCsDtt g¡ƒd}| |d¡tt gd¢¡ƒd}| |d¡dS)zX Test that the function which formats stat lines does so appropriately. éÿÿÿÿs +OK 0 0 )é ériu's +OK 4 10142 N)Úlistr ÚformatStatResponser)r!ÚstatLiner'r'r(Útest_StatLineFormatterOs z#UtilityTests.test_StatLineFormattercCsDtt g¡ƒ}| |ddg¡tt gd¢¡ƒ}| |gd¢¡dS)z} Test that the function which formats the lines in response to a LIST command does so appropriately. ó+OK 0 ó. )ééééd)s+OK 4 s1 1 s2 2 s3 3 s4 100 r8N)r3r ÚformatListResponser)r!Ú listLinesr'r'r(Útest_ListLineFormatterYsþz#UtilityTests.test_ListLineFormattercCszgd¢}tt g|j¡ƒ}| |ddg¡tt gd¢|j¡ƒ}| |gd¢¡tt gd¢|j¡ƒ}| |gd¢¡dS) zy Test that the function which formats lines in response to a UIDL command does so appropriately. )ÚabcÚdefÚghió+OK r8)é{i¯éO)rCó1 abc s2 def ó3 ghi r8)rNrE)rCrFrGr8N)r3r ÚformatUIDListResponseÚ __getitem__r)r!Úuidsr>r'r'r(Útest_UIDListLineFormattergsÿz&UtilityTests.test_UIDListLineFormatterN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r)r,r/r6r?rKr'r'r'r(r!s  rc@óeZdZdZdZdd„ZdS)Ú MyVirtualPOP3z2 A virtual-domain-supporting POP3 server. ócCs(| |¡\}}|jjd |||j|¡S)a7 Authenticate against a user against a virtual domain. @param user: The username. @param digest: The digested password. @return: A three-tuple like the one returned by L{IRealm.requestAvatar}. The mailbox will be for the user given by C{user}. óbaz.com)Ú lookupDomainÚserviceÚdomainsÚauthenticateUserAPOPÚmagic)r!ÚuserÚdigestÚdomainr'r'r(rW€s  ÿz"MyVirtualPOP3.authenticateUserAPOPN©rLrMrNrOrXrWr'r'r'r(rQyó rQc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) Ú DummyDomainz- A virtual domain for a POP3 server. cCs i|_dS©N©Úusersr.r'r'r(Ú__init__–s zDummyDomain.__init__cCsg|j|<dS)zV Create a mailbox for a new user. @param name: The username. Nr`)r!Únamer'r'r(ÚaddUser™ózDummyDomain.addUsercCs|j| |¡dS)zš Add a message to the mailbox of the named user. @param name: The username. @param message: The contents of the message. N)raÚappend)r!rcÚmessager'r'r(Ú addMessage¡szDummyDomain.addMessagecCstjt|j|ƒdd„fS)ae Succeed with a L{ListMailbox}. @param name: The name of the user authenticating. @param digest: ignored @param magic: ignored @param domain: ignored @return: A three-tuple like the one returned by L{IRealm.requestAvatar}. The mailbox will be for the user given by C{name}. cSódSr_r'r'r'r'r(Ú·óz2DummyDomain.authenticateUserAPOP..)r ÚIMailboxÚ ListMailboxra)r!rcrZrXr[r'r'r(rWªs z DummyDomain.authenticateUserAPOPN)rLrMrNrOrbrdrhrWr'r'r'r(r^‘s  r^c@sBeZdZdZdd„Zddd„Zdd„Zd d „Zd d „Zd d„Z dS)rmz@ A simple in-memory list implementation of L{IMailbox}. cCs ||_dS)z, @param list: The messages. N©r3)r!r3r'r'r(rb¿ó zListMailbox.__init__NcCs&|dur dd„|jDƒSt|j|ƒS)ú™ Get some message information. @param i: See L{pop3.IMailbox.listMessages}. @return: See L{pop3.IMailbox.listMessages}. NcSóg|]}t|ƒ‘qSr'©Úlen)Ú.0Úlr'r'r(Ú Íóz,ListMailbox.listMessages..)r3rs©r!r%r'r'r(Ú listMessagesÅszListMailbox.listMessagescCót|j|ƒS©z Get the message content. @param i: See L{pop3.IMailbox.getMessage}. @return: See L{pop3.IMailbox.getMessage}. )rr3rxr'r'r(Ú getMessageÐózListMailbox.getMessagecCs|S)z¡ Construct a UID by using the given index value. @param i: See L{pop3.IMailbox.getUidl}. @return: See L{pop3.IMailbox.getUidl}. r'rxr'r'r(ÚgetUidlÙszListMailbox.getUidlcCód|j|<dS©zm Wipe the message at the given index. @param i: See L{pop3.IMailbox.deleteMessage}. óNrnrxr'r'r(Ú deleteMessageârezListMailbox.deleteMessagecCódS)z= No-op. @see: L{pop3.IMailbox.sync} Nr'r.r'r'r(ÚsyncêrkzListMailbox.syncr_) rLrMrNrOrbryr|r~r‚r„r'r'r'r(rmºs    rmc@s8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ÚMyPOP3DownloaderzE A POP3 client which downloads all messages from the server. cCstj ||¡| dd¡dS)zK Authenticate. @param line: The welcome response. s hello@baz.comsworldN)r Ú POP3ClientÚhandle_WELCOMEÚapop©r!Úliner'r'r(r‡÷szMyPOP3Downloader.handle_WELCOMEcCsB| ¡}|d}|dkrtd|›d|›dƒ‚g|_| d¡dS)zc Require an I{OK} response to I{APOP}. @param line: The I{APOP} response. ró+OKz code is: z , parts is: ú r9N)ÚsplitÚAssertionErrorÚlinesÚretr)r!rŠÚpartsÚcoder'r'r(Ú handle_APOPs zMyPOP3Downloader.handle_APOPcCs|j |¡dS)zh Record one line of message information. @param line: A I{RETR} response line. N)rrfr‰r'r'r(Úhandle_RETR_continue óz%MyPOP3Downloader.handle_RETR_continuecCsd |j¡d|_| ¡dS)z: Record the received message information. ó N)ÚjoinrrgÚquitr.r'r'r(Úhandle_RETR_ends z MyPOP3Downloader.handle_RETR_endcCs |dd…dkrtd|ƒ‚dS)zc Require an I{OK} response to I{QUIT}. @param line: The I{QUIT} response. Nr;r‹scode is )rŽr‰r'r'r(Ú handle_QUITs ÿzMyPOP3Downloader.handle_QUITN) rLrMrNrOr‡r“r”r™ršr'r'r'r(r…òs   r…c@sReZdZdZdZdeeƒfZdd„Zdd„Zdd „Z e j d d ge _ d d „Z dS)Ú POP3Testsz! Tests for L{pop3.POP3}. s-Subject: urgent Someone set up us the bomb! s{+OK +OK Authentication succeeded +OK 1 0 . +OK %d Subject: urgent Someone set up us the bomb! . +OK cCsNtj ¡|_i|j_tƒ|jjd<|jjd d¡|jjd d|j¡dS)zC Set up a POP3 server with virtual domain support. rSshelloN) rÚprotocolÚFactoryÚfactoryrVr^rdrhrgr.r'r'r(ÚsetUpAs zPOP3Tests.setUpcs:tgd¢ƒ‰tƒ}ˆj|_‡‡fdd„}t |ˆ¡ |¡S)zL Messages can be downloaded over a loopback TCP connection. )sAPOP hello@baz.com worldóUIDLóRETR 1óQUITcs"d ˆj¡d}ˆ |ˆj¡dS)Nó )r—ÚresponserÚexpectedOutput)Úignoredr"©Úclientr!r'r(ÚcheckZsz&POP3Tests.test_messages..check)r rQržrUr Ú loopbackTCPÚ addCallback)r!Úserverr©r'r§r(Ú test_messagesKsÿzPOP3Tests.test_messagescs:tƒ‰ˆjˆ_tƒ‰‡‡‡fdd„}t ˆˆ¡}| |¡S)zH Messages can be downloaded over a loopback connection. cs(ˆ ˆjˆj¡ˆ t tdƒ¡¡dS)NúTest harness disconnect)rrgÚconnectionLostr ÚFailureÚ Exception)r¦©ÚclientProtocolrœr!r'r(r©hs ÿz&POP3Tests.test_loopback..check)rQržrUr…r Ú loopbackAsyncr«)r!r©Údr'r²r(Ú test_loopback`s   zPOP3Tests.test_loopbackz*twisted.mail.pop3.POP3Client is deprecated)rgcCsRtj ¡}i|_tƒ|jd<tƒ}||_| tj |j dd¡}|  |j dd¡dS)zP Look up a user in a domain which this server does not support. stwistedmatrix.comsnobody@baz.comspasswordrzno such domain baz.comN) rrœrrVr^rQrUÚ assertRaisesr Ú POP3ErrorrWrÚargs)r!ržr¬Úexcr'r'r(Útest_incorrectDomainus   ÿzPOP3Tests.test_incorrectDomainN) rLrMrNrOrgrsr¥rŸr­r¶rÚsuppressr»r'r'r'r(r›&s ÿô  ÿ r›c@rP)Ú DummyPOP3zF A simple POP3 server with a hard-coded mailbox for any user. rRcCstjttƒdd„fS)zÎ Succeed with a L{DummyMailbox}. @param user: ignored @param password: ignored @return: A three-tuple like the one returned by L{IRealm.requestAvatar}. cSrir_r'r'r'r'r(rj–rkz0DummyPOP3.authenticateUserAPOP..©r rlÚ DummyMailboxÚ ValueError)r!rYÚpasswordr'r'r(rWŒs zDummyPOP3.authenticateUserAPOPNr\r'r'r'r(r½…r]r½c@óeZdZdZdd„ZdS)Ú DummyPOP3AuthzL Class to test successful authentication in twisted.mail.pop3.POP3. cCs6tj tƒ¡|_tj ¡}| ||¡|j |¡dSr_)rÚportalÚPortalÚ TestRealmÚcheckersÚ'InMemoryUsernamePasswordDatabaseDontUserdÚregisterChecker)r!rYrÁÚchr'r'r(rbžs  zDummyPOP3Auth.__init__N)rLrMrNrOrbr'r'r'r(rÙó rÃc@s@eZdZdZdgZdd„Zddd„Zdd „Zd d „Zd d „Z dS)r¿zý An in-memory L{pop3.IMailbox} implementation. @ivar messages: A sequence of L{bytes} defining the messages in this mailbox. @ivar exceptionType: The type of exception to raise when an out-of-bounds index is addressed. s,From: moshe To: moshe How are you, friend? cCstjdd…|_||_dSr_)r¿ÚmessagesÚ exceptionType)r!rÍr'r'r(rb²s zDummyMailbox.__init__NcCs<|dur dd„|jDƒS|t|jƒkr| ¡‚t|j|ƒS)rpNcSrqr'rr)rtÚmr'r'r(rv¾rwz-DummyMailbox.listMessages..)rÌrsrÍrxr'r'r(ry¶s zDummyMailbox.listMessagescCrzr{)rrÌrxr'r'r(r|Ãr}zDummyMailbox.getMessagecCs |t|jƒkr | ¡‚d|fS)zÇ Construct a UID which is simply the string representation of the given index. @param i: See L{pop3.IMailbox.getUidl}. @return: See L{pop3.IMailbox.getUidl}. s%d)rsrÌrÍrxr'r'r(r~Ìs zDummyMailbox.getUidlcCrr€)rÌrxr'r'r(r‚ØrezDummyMailbox.deleteMessager_) rLrMrNrOrÌrbryr|r~r‚r'r'r'r(r¿¥s    r¿c@säeZdZdZd5dd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z ddggfdd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%d&„Zd'd(„Zd)d*„Zd+d,„Zd-d.„Zd/d0„Zd1d2„Zd3d4„ZdS)6ÚAnotherPOP3Testsz( Additional L{pop3.POP3} tests. NcCs4|r|ntƒ}t|ƒ}t ||¡}| |j|||¡S)aO Assert that when C{lines} are delivered to L{pop3.POP3} it responds with C{expectedOutput}. @param lines: A sequence of L{bytes} representing lines to deliver to the server. @param expectedOutput: A sequence of L{bytes} representing the expected response from the server. @param protocolInstance: Instance of L{twisted.mail.pop3.POP3} or L{None}. If L{None}, a new DummyPOP3 will be used. @return: A L{Deferred} that fires when the lines have been delivered and the output checked. )r½r r r´r«Ú _cbRunTest)r!rr¥ÚprotocolInstanceÚdummyr¨rµr'r'r(ÚrunTestæs zAnotherPOP3Tests.runTestcCs2| d |¡d |j¡¡| t tdƒ¡¡|S)Nr£r®)rr—r¤r¯r r°r±)r!r¦r¨rÒr¥r'r'r(rÐüszAnotherPOP3Tests._cbRunTestcCó| gd¢gd¢¡S)a4 Test a lot of different POP3 commands in an extremely pipelined scenario. This test may cover legitimate behavior, but the intent and granularity are not very good. It would likely be an improvement to split it into a number of smaller, more focused tests. )sAPOP moshez dummyóLISTr r¡óRETR 2sDELE 1r¡r¢)ó +OK ó+OK Authentication succeededs+OK 1s1 44ó.ó+OK s1 0rÙs+OK 44s From: moshes To: moshersHow are you, friend?rÙs -ERR Bad message number argumentrÚs-ERR message deletedrÚ©rÓr.r'r'r(Ú test_buffers  õzAnotherPOP3Tests.test_buffercCrÔ)z) Test the no-op command. )sAPOP spiv dummysNOOPr¢)r×rØrÚrÚrÛr.r'r'r(Ú test_noop+sþzAnotherPOP3Tests.test_noopcCs>d}| ddgdd|dg¡}| tj¡}| t|ƒd¡|S)zi Sending a command with invalid UTF-8 characters will raise a L{pop3.POP3Error}. s&not authenticated yet: cannot do PASSsPASSr¢r×s(-ERR bad protocol or server: POP3Error: rÚr9)rÓÚflushLoggedErrorsr r¸rrs)r!ÚerrorrµÚerrorsr'r'r(Útest_badUTF8CharactersInCommand4sýþ z0AnotherPOP3Tests.test_badUTF8CharactersInCommandcCsHtƒ}tj ¡|_ddddœ|j_tddgƒ}t ||¡}|  |j |¡S)zš L{pop3.POP3} responds to an I{AUTH} command with a list of supported authentication types based on its factory's C{challengers}. N)sAuth1s secondAuthsauthLastsAUTHr¢) r½rrœrržÚ challengersr r r´r«Ú_cbTestAuthListing)r!Úpr¨rµr'r'r(Útest_authListingFs þÿ z!AnotherPOP3Tests.test_authListingcCsJ| |jd d¡¡| t|jdd…ƒgd¢¡| |jdd¡dS)Nr9r‹r:r)sAUTH1sAUTHLASTs SECONDAUTHrÙ)Ú assertTruer¤Ú startswithrÚsorted)r!r¦r¨r'r'r(rãXs ÿz#AnotherPOP3Tests._cbTestAuthListingc Cs~|s|}|s|}gd¢}||kr||krgd¢}d d|g¡d d|g¡g}||dg7}||dg7}|j||t||ƒdS) aÝ Test a login with PASS. If L{real_user} matches L{tried_user} and L{real_password} matches L{tried_password}, a successful login will be expected. Otherwise an unsuccessful login will be expected. @type real_user: L{bytes} @param real_user: The user to test. @type real_password: L{bytes} @param real_password: The password of the test user. @type tried_user: L{bytes} or L{None} @param tried_user: The user to call USER with. If None, real_user will be used. @type tried_password: L{bytes} or L{None} @param tried_password: The password to call PASS with. If None, real_password will be used. @type after_auth_input: L{list} of l{bytes} @param after_auth_input: Extra protocol input after authentication. @type after_auth_output: L{list} of l{bytes} @param after_auth_output: Extra protocol output after authentication. )r×ó+OK USER accepted, send PASSs-ERR Authentication failed)r×rérØó óUSERóPASSr¢rÚ)rÑ)r—rÓrÃ) r!Ú real_userÚ real_passwordÚ tried_userÚtried_passwordÚafter_auth_inputÚafter_auth_outputr¤Ú fullInputr'r'r(Úrun_PASS_s"$  þýzAnotherPOP3Tests.run_PASScCs| d d|g¡dggd¢¡S)z Test protocol violation produced by calling PASS before USER. @type password: L{bytes} @param password: A password to test. rêrìr¢)r×s-ERR USER required before PASSrÚ)rÓr—)r!rÁr'r'r(Úrun_PASS_before_USER sþz%AnotherPOP3Tests.run_PASS_before_USERcCó | d¡S)z> Test PASS before USER with a wrong password. ófooz©rõr.r'r'r(Útest_illegal_PASS_before_USER«roz.AnotherPOP3Tests.test_illegal_PASS_before_USERcCrö)z? Test PASS before USER with an empty password. rrør.r'r'r(Útest_empty_PASS_before_USER±roz,AnotherPOP3Tests.test_empty_PASS_before_USERcCrö)zI Test PASS before USER with an password that is a space. rêrør.r'r'r(Útest_one_space_PASS_before_USER·roz0AnotherPOP3Tests.test_one_space_PASS_before_USERcCrö)zK Test PASS before USER with a password containing a space. ó fooz barzrør.r'r'r(Útest_space_PASS_before_USER½roz,AnotherPOP3Tests.test_space_PASS_before_USERcCrö)zS Test PASS before USER with a password containing multiple spaces. ófooz barz asdfrør.r'r'r(Ú%test_multiple_spaces_PASS_before_USERÃroz6AnotherPOP3Tests.test_multiple_spaces_PASS_before_USERcCrö)zS Test PASS before USER with a password containing tabs and spaces. ófooz barz crazy@! rør.r'r'r(Ú&test_other_whitespace_PASS_before_USERÉroz7AnotherPOP3Tests.test_other_whitespace_PASS_before_USERcCó | dd¡S)z1 Test PASS with a good password. ótestuserr÷©rôr.r'r'r(Útest_good_PASSÏó zAnotherPOP3Tests.test_good_PASScCr)ú? Test PASS with a password containing a space. rrürr.r'r'r(Útest_space_PASSÕrz AnotherPOP3Tests.test_space_PASScCr)rrrþrr.r'r'r(Útest_multiple_spaces_PASSÛrz*AnotherPOP3Tests.test_multiple_spaces_PASScCr©zG Test PASS with a password containing tabs and spaces. rrrr.r'r'r(Útest_other_whitespace_PASSárz+AnotherPOP3Tests.test_other_whitespace_PASScCó|jddddS)z. Test PASS with a wrong user. rr÷s wronguser)rïrr.r'r'r(Útest_pass_wrong_userçóz%AnotherPOP3Tests.test_pass_wrong_usercCr )z2 Test PASS with a wrong password. rr÷sbarz©rðrr.r'r'r(Útest_wrong_PASSírz AnotherPOP3Tests.test_wrong_PASScCr )rrrüs foozbarz rrr.r'r'r(Útest_wrong_space_PASSórz&AnotherPOP3Tests.test_wrong_space_PASScCr )rrrþs foozbarz rrr.r'r'r(Útest_wrong_multiple_spaces_PASSùsÿz0AnotherPOP3Tests.test_wrong_multiple_spaces_PASScCrr rr.r'r'r(Ú test_wrong_other_whitespace_PASSrz1AnotherPOP3Tests.test_wrong_other_whitespace_PASScCs4dg}d ddg¡g}|jdd||d |jtj¡S)zM After logging in, test a dummy command that is not defined. s DUMMY COMMANDrês'-ERR bad protocol or server: POP3Error:sUnknown protocol command: DUMMYró testpassword)rñrò)r—rôr«rÞr r¸)r!Ú extra_inputÚ extra_outputr'r'r(Útest_wrong_commandsþÿÿ ü ûz#AnotherPOP3Tests.test_wrong_commandr_)rLrMrNrOrÓrÐrÜrÝrárårãrôrõrùrúrûrýrÿrrrr r r rrrrrr'r'r'r(rÏás> *   ùA  rÏc@sLeZdZdZdd„Zdd„ZeddgƒZdd „Zd Z d d „Z d Z d d„Z dS)ÚTestServerFactoryz÷ A L{pop3.IServerFactory} implementation, for use by the test suite, with some behavior controlled by the values of (settable) public attributes and other behavior based on values hard-coded both here and in some test methods. cCrƒ)úW Return the hard-coded value. @return: L{pop3.IServerFactory} zTest Implementation Stringr'r.r'r'r(Úcap_IMPLEMENTATION&óz$TestServerFactory.cap_IMPLEMENTATIONcCrƒ)ré<r'r.r'r'r(Ú cap_EXPIRE.rzTestServerFactory.cap_EXPIRE)sSCHEME_1N)sSCHEME_2NcCrƒ)réxr'r.r'r'r(Úcap_LOGIN_DELAY8rz!TestServerFactory.cap_LOGIN_DELAYTcCó|jS©r)Úpuer.r'r'r(ÚperUserExpirationBóz#TestServerFactory.perUserExpirationcCr r!)Úpuldr.r'r'r(ÚperUserLoginDelayLr$z#TestServerFactory.perUserLoginDelayN) rLrMrNrOrrrrârr"r#r%r&r'r'r'r(rs  rc@seZdZdZdZdZdS)Ú TestMailboxz× An incomplete L{IMailbox} implementation with certain per-user values hard-coded and known by tests in this module. This is useful for testing the server's per-user capability implementation. r<éN)rLrMrNrOÚ loginDelayÚmessageExpirationr'r'r'r(r'Us r'cGs|D]}| ||¡qdS)aG Assert that the given capability is included in all of the capability sets. @param testcase: A L{unittest.TestCase} to use to make assertions. @param s: The capability for which to check. @type s: L{bytes} @param caps: The capability sets in which to check. @type caps: L{tuple} of iterable N)ÚassertIn)ÚtestcaseÚsÚcapsr$r'r'r(Ú containedcs ÿr/c@sPeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dS)ÚCapabilityTestsz@ Tests for L{pop3.POP3}'s per-user capability handling. cCs–tƒ}t ¡}tƒ|_tj |¡|_|  ¡|  ¡|  ¡|_ |  ¡ ¡|_tƒ}tƒ|_tj |¡|_|  ¡|  ¡ ¡|_| t tdƒ¡¡dS)ú> Create a POP3 server with some capabilities. r®N)rr ÚPOP3rržrrœÚ FileWrapperÚ transportÚconnectionMadeÚdo_CAPAÚlistCapabilitiesr.ÚgetvalueÚ splitlinesÚpcapsr'ÚmboxÚlpcapsr¯r r°r±©r!r-rär'r'r(rŸys zCapabilityTests.setUpcCót|d|j|j|jƒdS)zB The server can advertise the I{UIDL} capability. r N©r/r.r:r<r.r'r'r(Ú test_UIDLózCapabilityTests.test_UIDLcCr>)zA The server can advertise the I{TOP} capability. sTOPNr?r.r'r'r(Útest_TOP•rAzCapabilityTests.test_TOPcCr>)zB The server can advertise the I{USER} capability. rëNr?r.r'r'r(Ú test_USER›rAzCapabilityTests.test_USERcCs$t|d|j|jƒt|d|jƒdS)zj The server can advertise its per-user expiration as well as a global expiration. sEXPIRE 60 USERs EXPIRE 25Nr?r.r'r'r(Ú test_EXPIRE¡ózCapabilityTests.test_EXPIREcCr>)zE The server can advertise its implementation string. s)IMPLEMENTATION Test Implementation StringNr?r.r'r'r(Útest_IMPLEMENTATION©sûz#CapabilityTests.test_IMPLEMENTATIONcCr>)zH The server can advertise the SASL schemes it supports. sSASL SCHEME_1 SCHEME_2Nr?r.r'r'r(Ú test_SASLµrAzCapabilityTests.test_SASLcCs$t|d|j|jƒ| d|j¡dS)zc The can advertise a per-user login delay as well as a global login delay. sLOGIN-DELAY 120 USERsLOGIN-DELAY 100N)r/r.r:r+r<r.r'r'r(Útest_LOGIN_DELAY»rEz CapabilityTests.test_LOGIN_DELAYN) rLrMrNrOrŸr@rBrCrDrFrGrHr'r'r'r(r0ts r0c@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚGlobalCapabilitiesTestsz> Tests for L{pop3.POP3}'s global capability handling. cCs¦tƒ}t ¡}tƒ|_d|j_|j_tj  |¡|_ |  ¡|  ¡|  ¡|_| ¡ ¡|_tƒ}tƒ|_tj  |¡|_ |  ¡| ¡ ¡|_| t tdƒ¡¡dS)r1Fr®N)rr r2rržr"r%rrœr3r4r5r6r7r.r8r9r:r'r;r<r¯r r°r±r=r'r'r(rŸÉs zGlobalCapabilitiesTests.setUpcCr>)zG I{EXPIRE} is in the server's advertised capabilities. s EXPIRE 60Nr?r.r'r'r(rDàrAz#GlobalCapabilitiesTests.test_EXPIREcCr>)zL I{LOGIN-DELAY} is in the server's advertised capabilities. sLOGIN-DELAY 120Nr?r.r'r'r(rHærAz(GlobalCapabilitiesTests.test_LOGIN_DELAYN)rLrMrNrOrŸrDrHr'r'r'r(rIÄs  rIc@rÂ)rÆzI An L{IRealm} which knows about a single test account's mailbox. cGs |dkrtjttƒdd„fSJ‚)aE Retrieve a mailbox for I{testuser} or fail. @param avatarId: See L{IRealm.requestAvatar}. @param mind: See L{IRealm.requestAvatar}. @param interfaces: See L{IRealm.requestAvatar}. @raises: L{AssertionError} when requesting an C{avatarId} other than I{testuser}. rcSrir_r'r'r'r'r(rjþrkz)TestRealm.requestAvatar..r¾)r!ÚavatarIdÚmindÚ interfacesr'r'r(Ú requestAvataròs zTestRealm.requestAvatarN)rLrMrNrOrMr'r'r'r(rÆírËrÆc@rÂ)Ú SASLTestsz7 Tests for L{pop3.POP3}'s SASL implementation. cCs.t ¡}tƒ|_dtjji|j_tj  t ƒ¡|_tj   ¡}|  dd¡|j |¡tƒ}tj |¡|_| ¡| d¡| | ¡ d¡dk¡| d¡| ¡ ¡dd d …}t |¡}tjd|td  ¡  d ¡}| t !d |¡¡| |j"¡| | ¡ ¡d d¡dk¡| #t$ %t&dƒ¡¡d S)z A CRAM-MD5-based SASL login attempt succeeds if it uses a username and a hashed password known to the server's credentials checker. sCRAM-MD5rrsCAPAs SASL CRAM-MD5rs AUTH CRAM-MD5r0r:N)Ú digestmodÚasciis testuser r‹r®)'r r2rržrÚ credentialsÚCramMD5CredentialsrârÄrÅrÆrÇrÈrdrÉrrrœr3r4r5Ú lineReceivedrær8Úfindr9Úbase64Ú b64decodeÚhmacÚHMACrÚ hexdigestÚencodeÚ b64encoder;r¯r r°r±)r!rärÊr-Úchalr¤r'r'r(Útest_ValidLogins*      ÿ  zSASLTests.test_ValidLoginN)rLrMrNrOr]r'r'r'r(rNrËrNc@sœeZdZdZdZdd„Zdd„Zdd„Zd d „Zd d „Z d d„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd „Zd!d"„Zd#d$„Zd%S)&Ú CommandMixinzI Tests for all the commands a POP3 server is allowed to receive. s1From: guy To: fellow More message text for you. cCs\t ¡}| |j¡|_t|_||_tƒ}t j   |¡|_ |  ¡| d¡| d¡||_dS)zŽ Make a POP3 server protocol instance hooked up to a simple mailbox and a transport that buffers output to a BytesIO. rN)r r2Ú mailboxTyperÍr;r3ÚscheduleÚ pop3Serverrrrœr3r4r5ÚseekÚtruncateÚ pop3Transport©r!rär-r'r'r(rŸ4s   zCommandMixin.setUpcCs|j t tdƒ¡¡dS)zo Disconnect the server protocol so it can clean up anything it might need to clean up. r®N)rar¯r r°r±r.r'r'r(ÚtearDownEs ÿzCommandMixin.tearDowncCs|jj ¡dS)zz Do some of the things that the reactor would take care of, if the reactor were actually running. N)rar4Ú_checkProducerr.r'r'r(Ú_flushNr•zCommandMixin._flushcCsh|j}|j}| d¡| ¡| | ¡d¡| d¡| d¡| d¡| ¡| | ¡d¡dS)zé Test the two forms of list: with a message index number, which should return a short-form response, and without a message index number, which should return a long-form response, one line per message. sLIST 1ó +OK 1 44 rrÕs+OK 1 1 44 . N©rardrSrhrr8rbrcrer'r'r(Ú test_LISTVs    zCommandMixin.test_LISTcCsš|j}|j}| d¡| | ¡d¡| d¡| d¡| d¡| | ¡d¡| d¡| d¡| d¡| | ¡d¡| d¡| d¡dS) zo Test that non-integers and out-of-bound integers produce appropriate error responses. sLIST as -ERR Invalid message-number: a rsLIST 0s -ERR Invalid message-number: 0 sLIST 2s -ERR Invalid message-number: 2 N©rardrSrr8rbrcrer'r'r(Útest_LISTWithBadArgumentió        z%CommandMixin.test_LISTWithBadArgumentcCs`|j}|j}| d¡| | ¡d¡| d¡| d¡| d¡| ¡| | ¡d¡dS)zy Test the two forms of the UIDL command. These are just like the two forms of the LIST command. sUIDL 1r7rr s+OK 1 0 . N)rardrSrr8rbrcrhrer'r'r(r@€s    zCommandMixin.test_UIDLcCóš|j}|j}| d¡| | ¡d¡| d¡| d¡| d¡| | ¡d¡| d¡| d¡| d¡| | ¡d¡| d¡| d¡dS)z€ Test that UIDL with a non-integer or an out-of-bounds integer produces the appropriate error response. sUIDL aó"-ERR Bad message number argument rsUIDL 0sUIDL 2Nrlrer'r'r(Útest_UIDLWithBadArgument‘rnz%CommandMixin.test_UIDLWithBadArgumentcCs2|j}|j}| d¡| ¡| | ¡d¡dS)z¦ Test the single form of the STAT command, which returns a short-form response of the number of messages in the mailbox and their total size. sSTATriN)rardrSrhrr8rer'r'r(Ú test_STAT¨s  zCommandMixin.test_STATcCsF|j}|j}| d¡| ¡| | ¡d¡| d¡| d¡dS)z- Test downloading a message. r¡s;+OK 44 From: moshe To: moshe How are you, friend? . rNrjrer'r'r(Ú test_RETR´s þ zCommandMixin.test_RETRcCro)z» Test that trying to download a message with a bad argument, either not an integer or an out-of-bounds integer, fails with the appropriate error response. sRETR arprsRETR 0rÖNrlrer'r'r(Útest_RETRWithBadArgumentÉs        z%CommandMixin.test_RETRWithBadArgumentcCsB|j}|j}|jj |j¡| d¡| ¡| |  ¡d¡dS)zQ Test downloading the headers and part of the body of a message. sTOP 1 0s9+OK Top of message follows From: moshe To: moshe . N) rardr;rÌrfÚ extraMessagerSrhrr8rer'r'r(rBás þzCommandMixin.test_TOPcCs|j}|j}|jj |j¡| d¡| | ¡d¡|  d¡|  d¡| d¡| | ¡d¡|  d¡|  d¡| d¡| | ¡d¡|  d¡|  d¡| d¡| | ¡d¡|  d¡|  d¡| d¡| | ¡d¡|  d¡|  d¡d S) a$ Test that trying to download a message with a bad argument, either a message number which isn't an integer or is an out-of-bounds integer or a number of lines which isn't an integer or is a negative integer, fails with the appropriate error response. sTOP 1 as-ERR Bad line count argument rsTOP 1 -1sTOP a 1rpsTOP 0 1sTOP 3 1N© rardr;rÌrfrurSrr8rbrcrer'r'r(Útest_TOPWithBadArgumentôs.              z$CommandMixin.test_TOPWithBadArgumentcCsN|j}|j}|jj |j¡| d¡| | ¡d¡|  d¡|  d¡dS)z“ Test the exceedingly pointless LAST command, which tells you the highest message index which you have already downloaded. óLASTr7rNrvrer'r'r(Ú test_LASTs  zCommandMixin.test_LASTcCst|j}|j}|jj |j¡| d¡| ¡| d¡|  d¡| d¡|  |  ¡d¡| d¡|  d¡dS)zM Test that issuing a RETR command updates the LAST response. rÖrrxó+OK 2 N© rardr;rÌrfrurSrhrbrcrr8rer'r'r(Útest_RetrieveUpdatesHighest&s     z(CommandMixin.test_RetrieveUpdatesHighestcCs`|j}|j}|jj |j¡| d¡| ¡| d¡|  d¡| d¡|  |  ¡d¡dS)zL Test that issuing a TOP command updates the LAST response. sTOP 2 10rrxrzNr{rer'r'r(Útest_TopUpdatesHighest7s    z#CommandMixin.test_TopUpdatesHighestcCsr|j}|j}|jj |j¡| d¡| ¡| d¡| ¡| d¡|  d¡| d¡|  |  ¡d¡dS)z Test that downloading a message with a smaller index than the current LAST response doesn't change the LAST response. rÖsTOP 1 10rrxrzNr{rer'r'r(Útest_HighestOnlyProgressesFs     z'CommandMixin.test_HighestOnlyProgressescCsj|j}|j}|jj |j¡| d¡| ¡| d¡| d¡|  d¡| d¡|  |  ¡d¡dS)zH Test that issuing RSET changes the LAST response to 0. rÖsRSETrrxr7Nr{rer'r'r(Útest_ResetClearsHighestXs     z$CommandMixin.test_ResetClearsHighestN)rLrMrNrOrurŸrfrhrkrmr@rqrrrsrtrBrwryr|r}r~rr'r'r'r(r^(s(  $ r^z}twisted.mail.pop3.IMailbox.listMessages may not raise IndexError for out-of-bounds message numbers: raise ValueError instead.)rgÚcategoryzxtwisted.mail.pop3.IMailbox.getUidl may not raise IndexError for out-of-bounds message numbers: raise ValueError instead.c@sXeZdZdZeZeZdd„Ze ge_ dd„Z e ge _ dd„Z e ge _ dd „Ze ge_ d S) ÚIndexErrorCommandTestsz» Run all of the command tests against a mailbox which raises IndexError when an out of bounds request is made. This behavior will be deprecated shortly and then removed. cCó t |¡S)z· An attempt to get metadata about a message with a bad argument fails with an I{ERR} response even if the mailbox implementation raises L{IndexError}. )r^rmr.r'r'r(rm†ó z/IndexErrorCommandTests.test_LISTWithBadArgumentcCr‚)z· An attempt to look up the UID of a message with a bad argument fails with an I{ERR} response even if the mailbox implementation raises L{IndexError}. )r^rqr.r'r'r(rqrƒz/IndexErrorCommandTests.test_UIDLWithBadArgumentcCr‚)zµ An attempt to download some of a message with a bad argument fails with an I{ERR} response even if the mailbox implementation raises L{IndexError}. )r^rwr.r'r'r(rwšrƒz.IndexErrorCommandTests.test_TOPWithBadArgumentcCr‚)z­ An attempt to download a message with a bad argument fails with an I{ERR} response even if the mailbox implementation raises L{IndexError}. )r^rtr.r'r'r(rt¤rƒz/IndexErrorCommandTests.test_RETRWithBadArgumentN)rLrMrNrOÚ IndexErrorrÍr¿r_rmÚ_listMessageSuppressionr¼rqÚ_getUidlSuppressionrwrtr'r'r'r(r|s rc@seZdZdZeZeZdS)ÚValueErrorCommandTestsa Run all of the command tests against a mailbox which raises ValueError when an out of bounds request is made. This is the correct behavior and after support for mailboxes which raise IndexError is removed, this will become just C{CommandTestCase}. N)rLrMrNrOrÀrÍr¿r_r'r'r'r(r‡¯sr‡c@seZdZdZddd„ZdS)ÚSyncDeferredMailboxzo Mailbox which has a listMessages implementation which returns a Deferred which has already fired. NcCst t ||¡¡S)zÏ Synchronously list messages. @type n: L{int} or L{None} @param n: The 0-based index of the message. @return: A L{Deferred} which already has a message list result. )r Úsucceedr¿ry)r!r&r'r'r(ryÁs z SyncDeferredMailbox.listMessagesr_)rLrMrNrOryr'r'r'r(rˆ»srˆc@óeZdZdZeZdS)Ú"IndexErrorSyncDeferredCommandTestsz{ Run all of the L{IndexErrorCommandTests} tests with a synchronous-Deferred returning IMailbox implementation. N©rLrMrNrOrˆr_r'r'r'r(r‹Íór‹c@rŠ)Ú"ValueErrorSyncDeferredCommandTestsz{ Run all of the L{ValueErrorCommandTests} tests with a synchronous-Deferred returning IMailbox implementation. NrŒr'r'r'r(rŽÖrrŽc@s"eZdZdZdd„Zddd„ZdS)ÚAsyncDeferredMailboxzo Mailbox which has a listMessages implementation which returns a Deferred which has not yet fired. cOs"g|_tj|g|¢Ri|¤ŽdSr_)Úwaitingr¿rb)r!r*Úkwr'r'r(rbåszAsyncDeferredMailbox.__init__NcCs$t ¡}|j |t ||¡f¡|S)zÎ Record a new unfired L{Deferred} in C{self.waiting} and return it. @type n: L{int} or L{None} @param n: The 0-based index of the message. @return: The L{Deferred} )r ÚDeferredrrfr¿ry)r!r&rµr'r'r(ryés z!AsyncDeferredMailbox.listMessagesr_)rLrMrNrOrbryr'r'r'r(rßsrc@óeZdZdZeZdd„ZdS)Ú#IndexErrorAsyncDeferredCommandTestsú} Run all of the L{IndexErrorCommandTests} tests with an asynchronous-Deferred returning IMailbox implementation. cCó>|jjjr|jjj ¡\}}| |¡|jjjst |¡dS©zH Fire whatever Deferreds we've built up in our mailbox. N)rar;rÚpopÚcallbackrrh©r!rµr*r'r'r(rhó   þz*IndexErrorAsyncDeferredCommandTests._flushN©rLrMrNrOrr_rhr'r'r'r(r”øó r”c@r“)Ú#ValueErrorAsyncDeferredCommandTestsr•cCr–r—)rar;rr˜r™r‡rhršr'r'r(rhr›z*ValueErrorAsyncDeferredCommandTests._flushNrœr'r'r'r(rž rržc@rÂ)Ú POP3MiscTestsz} Miscellaneous tests more to do with module/package structure than anything to do with the Post Office Protocol. cCs(tjj}|jD] }| t||ƒ¡qdS)z This test checks that all names listed in twisted.mail.pop3.__all__ are actually present in the module. N)Útwistedrr Ú__all__ræÚhasattr)r!ÚmodÚattrr'r'r(Útest_all"s ÿzPOP3MiscTests.test_allN)rLrMrNrOr¥r'r'r'r(rŸs rŸc@rÂ)ÚPOP3ClientDeprecationTestszI Tests for the now deprecated L{twisted.mail.pop3client} module. cCs\ddlm}|j|jgd}| |ddt¡| |ddd¡| t|ƒd¡|dS) zò A deprecation warning is emitted when directly importing the now deprected pop3client module. This test might fail is some other code has already imported it. No code should use the deprected module. r)Ú pop3client)ÚoffendingFunctionsr€rgzXtwisted.mail.pop3client was deprecated in Twisted 21.2.0. Use twisted.mail.pop3 instead.r9N)Ú twisted.mailr§Ú flushWarningsÚtest_deprecationrÚDeprecationWarningrs)r!r§Ú warningsShownr'r'r(r«1s  þz+POP3ClientDeprecationTests.test_deprecationN)rLrMrNrOr«r'r'r'r(r¦,rËr¦)LrOrUrWrÚ collectionsrÚhashlibrÚiorÚzope.interfacerÚtwisted.cred.checkersr Útwisted.cred.credentialsÚtwisted.cred.portalÚtwisted.internet.protocolÚtwisted.mail.pop3Útwisted.mail.protocolsrrrÚtwisted.internetr r©r Útwisted.protocolsr Útwisted.pythonr Útwisted.test.proto_helpersr Ú twisted.trialrrÚTestCaserÚ protocolsÚ VirtualPOP3rQr^rmr†r…r›r2r½rÃÚMailboxr¿rÏÚIServerFactoryrr'r/r0rIrÆrNr^Ú_listMessageDeprecationr¼ÚPendingDeprecationWarningr…Ú_getUidlDeprecationr†rr‡rˆr‹rŽrr”ržrŸr¦r'r'r'r(Ús€         X)84_ <>7P)&Dÿÿÿÿ3