HEX

File: //proc/self/root/lib/python3/dist-packages/twisted/mail/test/__pycache__/test_pop3.cpython-38.pyc
U


W[n��@s�dZddlmZddlZddlZddlZddlmZddlm	Z	ddl
mZddlm
Z
ddlmZdd	lmZdd
lmZddlmZddlmZdd
lmZddlmZddlmZddlmZmZddlZddl Zddl!Zddl"Zddl#Zddl$ZGdd�dej%�Z&Gdd�dej'j(�Z)Gdd�d�Z*Gdd�d�Z+Gdd�dej,�Z-Gdd�dej%�Z.Gdd�dej/�Z0Gdd �d ej1�Z2Gd!d"�d"ej%�Z3eej4�Gd#d$�d$��Z5Gd%d&�d&�Z6d'd(�Z7Gd)d*�d*ej%�Z8Gd+d,�d,ej%�Z9Gd-d.�d.�Z:Gd/d0�d0ej%�Z;Gd1d2�d2�Z<d3Z=ej>e=e?d4�Z@d5ZAej>eAe?d4�ZBGd6d7�d7e<ej%�ZCGd8d9�d9e<ej%�ZDGd:d;�d;e2�ZEGd<d=�d=eC�ZFGd>d?�d?eD�ZGGd@dA�dAe2�ZHGdBdC�dCeC�ZIGdDdE�dEeD�ZJGdFdG�dGej%�ZKdS)Hz,
Test cases for Ltwisted.mail.pop3} module.
�)�print_functionN)�OrderedDict)�BytesIO)�implementer)�cred)�internet)�mail)�defer)�pop3)�loopback)�failure)�
intToBytes)�LineSendingProtocol)�unittest�utilc@s@eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dS)�UtilityTestszl
    Test the various helper functions and classes used by the POP3 server
    protocol implementation.
    cCs�g}tt�ddddddg��}t�|j|d�}t|�}|�|g�t|�|�|g�t|�|�|g�t|�|�|dddg�td�D]}t|�q�|�|ddddddddg�d	S)
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.
        Z012Z345�6�7�8�9��N)	�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_LineBufferingcCsBg}tdddg�}t�|j|d�}|D]}q&|�|dddg�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_FinishLineBufferingBsz%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
ZsuccessResponse�r r&r&r'�test_SuccessResponseFormatterOs�z*UtilityTests.test_SuccessResponseFormattercCsHtt�g��d}|�|d�tt�ddddg��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
ZformatStatResponser)r ZstatLiner&r&r'�test_StatLineFormatterYsz#UtilityTests.test_StatLineFormatterc	CsPtt�g��}|�|ddg�tt�ddddg��}|�|ddd	d
ddg�dS)
z}
        Test that the function which formats the lines in response to a LIST
        command does so appropriately.
        �+OK 0
�.
����ds+OK 4
s1 1
s2 2
s3 3
s4 100
N)r1r
ZformatListResponser)r �	listLinesr&r&r'�test_ListLineFormatterds�
��z#UtilityTests.test_ListLineFormattercCs�dddg}tt�g|j��}|�|ddg�tt�dddg|j��}|�|dd	d
ddg�tt�dd
dg|j��}|�|dd	ddg�d
S)zy
        Test that the function which formats lines in response to a UIDL
        command does so appropriately.
        �abc�defZghis+OK 
r4�{i�iOs1 abc
s2 def
s3 ghi
rN)r1r
ZformatUIDListResponse�__getitem__r)r Zuidsr9r&r&r'�test_UIDListLineFormatterus(
����
�z&UtilityTests.test_UIDListLineFormatterN)
�__name__�
__module__�__qualname__�__doc__r(r+r-r2r:r?r&r&r&r'r%s

rc@seZdZdZdZdd�ZdS)�
MyVirtualPOP3z2
    A virtual-domain-supporting POP3 server.
    �<moshez>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)ZlookupDomain�service�domains�authenticateUserAPOP�magic)r �user�digest�domainr&r&r'rI�s�z"MyVirtualPOP3.authenticateUserAPOPN�r@rArBrCrJrIr&r&r&r'rD�srDc@s0eZdZdZdd�Zdd�Zdd�Zdd	�Zd
S)�DummyDomainz-
    A virtual domain for a POP3 server.
    cCs
i|_dS�N��usersr,r&r&r'�__init__�szDummyDomain.__init__cCsg|j|<dS)zV
        Create a mailbox for a new user.

        @param name: The username.
        NrQ)r �namer&r&r'�addUser�szDummyDomain.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)rR�append)r rT�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}.
        cSsdSrPr&r&r&r&r'�<lambda>��z2DummyDomain.authenticateUserAPOP.<locals>.<lambda>)r
�IMailbox�ListMailboxrR)r rTrLrJrMr&r&r'rI�s
z DummyDomain.authenticateUserAPOPN)r@rArBrCrSrUrXrIr&r&r&r'rO�s
	
rOc@sBeZdZdZdd�Zddd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)r\z@
    A simple in-memory list implementation of L{IMailbox}.
    cCs
||_dS)z,
        @param list: The messages.
        N�r1)r r1r&r&r'rS�szListMailbox.__init__NcCs&|dkrdd�|jD�St|j|�S)��
        Get some message information.

        @param i: See L{pop3.IMailbox.listMessages}.
        @return: See L{pop3.IMailbox.listMessages}.
        NcSsg|]}t|��qSr&��len)�.0�lr&r&r'�
<listcomp>�sz,ListMailbox.listMessages.<locals>.<listcomp>)r1r`�r r$r&r&r'�listMessages�szListMailbox.listMessagescCst|j|�S�z�
        Get the message content.

        @param i: See L{pop3.IMailbox.getMessage}.
        @return: See L{pop3.IMailbox.getMessage}.
        )rr1rdr&r&r'�
getMessage�szListMailbox.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&rdr&r&r'�getUidl�szListMailbox.getUidlcCsd|j|<dS�zm
        Wipe the message at the given index.

        @param i: See L{pop3.IMailbox.deleteMessage}.
        rZNr]rdr&r&r'�
deleteMessage�szListMailbox.deleteMessagecCsdS)z=
        No-op.

        @see: L{pop3.IMailbox.sync}
        Nr&r,r&r&r'�syncszListMailbox.sync)N)
r@rArBrCrSrergrhrjrkr&r&r&r'r\�s


	r\c@s8eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�ZdS)
�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_WELCOMEZapop�r �liner&r&r'rnszMyPOP3Downloader.handle_WELCOMEcCs<|��}|d}|dkr(td||f��g|_|�d�dS)zc
        Require an I{OK} response to I{APOP}.

        @param line: The I{APOP} response.
        r�+OKzcode is: %s , parts is: %s r5N)�split�AssertionError�linesZretr)r rp�parts�coder&r&r'�handle_APOPszMyPOP3Downloader.handle_APOPcCs|j�|�dS)zh
        Record one line of message information.

        @param line: A I{RETR} response line.
        N)rtrVror&r&r'�handle_RETR_continue+sz%MyPOP3Downloader.handle_RETR_continuecCsd�|j�d|_|��dS)z:
        Record the received message information.
        �
N)�joinrtrW�quitr,r&r&r'�handle_RETR_end4sz MyPOP3Downloader.handle_RETR_endcCs |dd�dkrtd|��dS)zc
        Require an I{OK} response to I{QUIT}.

        @param line: The I{QUIT} response.
        Nr7rqscode is )rsror&r&r'�handle_QUIT<szMyPOP3Downloader.handle_QUITN)	r@rArBrCrnrwrxr|r}r&r&r&r'rls
	rlc@sXeZdZdZdZdeee��dZdd�Zdd�Z	d	d
�Z
ejdd�ge
_d
d�Z
dS)�	POP3Testsz!
    Tests for L{pop3.POP3}.
    s-Subject: urgent

Someone set up us the bomb!
s>+OK <moshez>
+OK Authentication succeeded
+OK 
1 0
.
+OK s;
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.
        rFshelloN)	r�protocol�Factory�factoryrHrOrUrXrWr,r&r&r'�setUp`s
zPOP3Tests.setUpcs>tddddg��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�
)rz�responser�expectedOutput)�ignoredr!��clientr r&r'�checkwsz&POP3Tests.test_messages.<locals>.check)rrDr�rGrZloopbackTCP�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)rrW�connectionLostr�Failure�	Exception)r��ZclientProtocolrr r&r'r��s�z&POP3Tests.test_loopback.<locals>.check)rDr�rGrlr�
loopbackAsyncr�)r r��dr&r�r'�
test_loopback}szPOP3Tests.test_loopbackz*twisted.mail.pop3.POP3Client is deprecated)rWcCsRtj��}i|_t�|jd<t�}||_|�tj	|j
dd�}|�|jdd�dS)zP
        Look up a user in a domain which this server does not support.
        stwistedmatrix.comsnobody@baz.comspasswordrzno such domain baz.comN)
rrr�rHrOrDrGZassertRaisesr
�	POP3ErrorrIr�args)r r�r��excr&r&r'�test_incorrectDomain�s
�zPOP3Tests.test_incorrectDomainN)r@rArBrCrWr
r`r�r�r�r�r�suppressr�r&r&r&r'r~Gs
��
�
r~c@seZdZdZdZdd�ZdS)�	DummyPOP3zF
    A simple POP3 server with a hard-coded mailbox for any user.
    rEcCstjtt�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}.
        cSsdSrPr&r&r&r&r'rY�rZz0DummyPOP3.authenticateUserAPOP.<locals>.<lambda>)r
r[�DummyMailbox�
ValueError)r rKZpasswordr&r&r'rI�s
zDummyPOP3.authenticateUserAPOPNrNr&r&r&r'r��sr�c@s@eZdZdZdgZdd�Zddd�Zdd	�Zd
d�Zdd
�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�|_||_dSrP)r��messages�
exceptionType)r r�r&r&r'rS�szDummyMailbox.__init__NcCs<|dkrdd�|jD�S|t|j�kr.|���t|j|�S)r^NcSsg|]}t|��qSr&r_)ra�mr&r&r'rc�sz-DummyMailbox.listMessages.<locals>.<listcomp>)r�r`r�rdr&r&r're�s
zDummyMailbox.listMessagescCst|j|�Srf)rr�rdr&r&r'rg�szDummyMailbox.getMessagecCs|t|j�kr|���t|�S)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}.
        )r`r�r�r
rdr&r&r'rh�szDummyMailbox.getUidlcCsd|j|<dSri)r�rdr&r&r'rj�szDummyMailbox.deleteMessage)N)
r@rArBrCr�rSrergrhrjr&r&r&r'r��s	


r�c@sheZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�Zdd�Zdd�Z
dd�ZdS)�AnotherPOP3Testsz(
    Additional L{pop3.POP3} tests.
    cCs,t�}t|�}t�||�}|�|j|||�S)a�
        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.

        @return: A L{Deferred} that fires when the lines have been delivered
            and the output checked.
        )r�rrr�r��
_cbRunTest)r rtr��dummyr�r�r&r&r'�runTest�szAnotherPOP3Tests.runTestcCs2|�d�|�d�|j��|�t�td���|S)Nr�r��rrzr�r�rr�r��r r�r�r�r�r&r&r'r�
s
��zAnotherPOP3Tests._cbRunTestcCs@|�ddddddddgdd	d
ddd
dddddddddd
dd
g�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��+OK <moshez>�+OK Authentication succeededs+OK 1s1 44�.�+OK s1 0s+OK 44sFrom: moshes	To: mosherZsHow are you, friend?s -ERR Bad message number arguments-ERR message deleted�r�r,r&r&r'�test_buffers<	���zAnotherPOP3Tests.test_buffercCs|�dddgddddg�S)z)
        Test the no-op command.
        sAPOP spiv dummysNOOPr�r�r�r�r�r,r&r&r'�	test_noop8s���zAnotherPOP3Tests.test_noopcCsHt�}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�rrr�r��challengersrrr�r��_cbTestAuthListing)r �pr�r�r&r&r'�test_authListingFs�
�z!AnotherPOP3Tests.test_authListingcCsL|�|jd�d��|�t|jdd��dddg�|�|jdd�dS)	Nr5rqr6rsAUTH1sAUTHLASTs
SECONDAUTHr�)�
assertTruer��
startswithr�sorted)r r�r�r&r&r'r�Xs
�z#AnotherPOP3Tests._cbTestAuthListingcCs.t�}tddg�}t�||�}|�|j||�S)z�
        L{pop3.POP3} handles a I{PASS} command before a I{USER} command with an
        error indicating the out-of-sequence operation.
        s	PASS foozr�)r�rrr�r��_cbTestIllegalPASS�r r�r�r�r&r&r'�test_illegalPASS_s�z!AnotherPOP3Tests.test_illegalPASScCs4d}|�|d�|j�d�|�t�td���dS�Ns4+OK <moshez>
-ERR USER required before PASS
+OK 
r�r�r�r�r&r&r'r�ms���z#AnotherPOP3Tests._cbTestIllegalPASScCs.t�}tddg�}t�||�}|�|j||�S)z|
        L{pop3.POP3} handles a I{PASS} command with a password equal to C{""}
        as it would any other value.
        sPASS r�)r�rrr�r��_cbTestEmptyPASSr�r&r&r'�test_emptyPASSvs�zAnotherPOP3Tests.test_emptyPASScCs4d}|�|d�|j�d�|�t�td���dSr�r�r�r&r&r'r��s���z!AnotherPOP3Tests._cbTestEmptyPASScCsVtd�}t|t�s|�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 �PASSzutf-8s�PASSr�r�s(-ERR bad protocol or server: POP3Error: r�r5)
�str�
isinstance�bytes�encoder�ZflushLoggedErrorsr
r�rr`)r �errorr��errorsr&r&r'�test_badUTF8CharactersInCommand�s"

����z0AnotherPOP3Tests.test_badUTF8CharactersInCommandN)r@rArBrCr�r�r�r�r�r�r�r�r�r�r�r&r&r&r'r��s&		r�c@sLeZdZdZdd�Zdd�Zeddg�Zdd	�Zd
Z	dd�Z
d
Zd
d�ZdS)�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.
    cCsdS)�W
        Return the hard-coded value.

        @return: L{pop3.IServerFactory}
        zTest Implementation Stringr&r,r&r&r'�cap_IMPLEMENTATION�sz$TestServerFactory.cap_IMPLEMENTATIONcCsdS)r��<r&r,r&r&r'�
cap_EXPIRE�szTestServerFactory.cap_EXPIRE)sSCHEME_1N)sSCHEME_2NcCsdS)r��xr&r,r&r&r'�cap_LOGIN_DELAY�sz!TestServerFactory.cap_LOGIN_DELAYTcCs|jS�r�)�puer,r&r&r'�perUserExpiration�sz#TestServerFactory.perUserExpirationcCs|jSr�)�puldr,r&r&r'�perUserLoginDelay�sz#TestServerFactory.perUserLoginDelayN)
r@rArBrCr�r�rr�r�r�r�r�r�r&r&r&r'r��s	r�c@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.
    r8�N)r@rArBrCZ
loginDelayZmessageExpirationr&r&r&r'r��sr�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)Ztestcase�s�capsr#r&r&r'�	contained�s
r�c@sPeZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dd�ZdS)�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
�POP3r�r�rr�FileWrapper�	transport�connectionMade�do_CAPA�listCapabilitiesr��getvalue�
splitlines�pcapsr��mbox�lpcapsr�rr�r��r r�r�r&r&r'r��s
zCapabilityTests.setUpcCst|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_UIDLszCapabilityTests.test_UIDLcCst|d|j|j|j�dS)zA
        The server can advertise the I{TOP} capability.
        sTOPNr�r,r&r&r'�test_TOPszCapabilityTests.test_TOPcCst|d|j|j|j�dS)zB
        The server can advertise the I{USER} capability.
        sUSERNr�r,r&r&r'�	test_USER"szCapabilityTests.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)szCapabilityTests.test_EXPIREcCst|d|j|j|j�dS)zE
        The server can advertise its implementation string.
        s)IMPLEMENTATION Test Implementation StringNr�r,r&r&r'�test_IMPLEMENTATION2s�z#CapabilityTests.test_IMPLEMENTATIONcCst|d|j|j|j�dS)zH
        The server can advertise the SASL schemes it supports.
        sSASL SCHEME_1 SCHEME_2Nr�r,r&r&r'�	test_SASL=s�zCapabilityTests.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_DELAYHsz CapabilityTests.test_LOGIN_DELAYN)r@rArBrCr�r�r�r�r�r�r�r�r&r&r&r'r��s	r�c@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)r�Fr�N)rr
r�r�r�r�r�rrr�r�r�r�r�r�r�r�r�r�r�r�r�rr�r�r�r&r&r'r�Vs
zGlobalCapabilitiesTests.setUpcCst|d|j|j|j�dS)zG
        I{EXPIRE} is in the server's advertised capabilities.
        s	EXPIRE 60Nr�r,r&r&r'r�nsz#GlobalCapabilitiesTests.test_EXPIREcCst|d|j|j|j�dS)zL
        I{LOGIN-DELAY} is in the server's advertised capabilities.
        sLOGIN-DELAY 120Nr�r,r&r&r'r�usz(GlobalCapabilitiesTests.test_LOGIN_DELAYN)r@rArBrCr�r�r�r&r&r&r'r�Rsr�c@seZdZdZdd�ZdS)�	TestRealmzI
    An L{IRealm} which knows about a single test account's mailbox.
    cGs(|dkrtjtt�dd�fSds$t�dS)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}.
        �testusercSsdSrPr&r&r&r&r'rY�rZz)TestRealm.requestAvatar.<locals>.<lambda>FN)r
r[r�r�rs)r ZavatarIdZmindZ
interfacesr&r&r'�
requestAvatar�szTestRealm.requestAvatarN)r@rArBrCr�r&r&r&r'r�}sr�c@seZdZdZdd�ZdS)�	SASLTestsz7
    Tests for L{pop3.POP3}'s SASL implementation.
    cCs0t��}t�|_dtjji|j_tj�	t
��|_tj��}|�
dd�|j�|�t�}tj�|�|_|��|�d�|�|���d�dk�|�d�|����dd	d
�}t�|�}t�d|����d�}|�t� d|��!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-MD5r�stestpasswordsCAPAs
SASL CRAM-MD5rs
AUTH CRAM-MD5r.r6N�asciis	testuser ryrqr�)'r
r�r�r�rZcredentialsZCramMD5Credentialsr�ZportalZPortalr�ZcheckersZ'InMemoryUsernamePasswordDatabaseDontUserUZregisterCheckerrrrr�r�r��lineReceivedr�r��findr��base64Zdecodestring�hmacZHMACZ	hexdigestr�Zencodestring�rstripr�r�rr�r�)r r�Zchr�Zchalr�r&r&r'�test_ValidLogin�s0�



� zSASLTests.test_ValidLoginN)r@rArBrCr�r&r&r&r'r��sr�c@s�eZdZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
d�Z
dd�Zdd�Zdd�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
r��mailboxTyper�r�r1Zschedule�
pop3Serverrrrr�r�r��seek�truncate�
pop3Transport�r r�r�r&r&r'r��s

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)r�r�rr�r�r,r&r&r'�tearDown�s
�zCommandMixin.tearDowncCs|jj��dS)zz
        Do some of the things that the reactor would take care of, if the
        reactor were actually running.
        N)r�r�Z_checkProducerr,r&r&r'�_flush�szCommandMixin._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�r�r�r�rrr�r�r�rr&r&r'�	test_LIST�s



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�r�r�r�rr�r�r�rr&r&r'�test_LISTWithBadArgument�s.
�


�


�
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 1r3rr�s+OK 
1 0
.
N)r�r�r�rr�r�r�rrr&r&r'r�s



zCommandMixin.test_UIDLcCs�|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 2Nrrr&r&r'�test_UIDLWithBadArgument-s.
�


�


�
z%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.
        sSTATrN)r�r�r�rrr�rr&r&r'�	test_STATKs

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?
.
rNrrr&r&r'�	test_RETRXs
�
zCommandMixin.test_RETRcCs�|j}|j}|�d�|�|��d�|�d�|�d�|�d�|�|��d�|�d�|�d�|�d�|�|��d�|�d�|�d�dS)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 arrsRETR 0r�Nrrr&r&r'�test_RETRWithBadArgumentms.
�


�


�
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)
r�r�r�r�rV�extraMessager�rrr�rr&r&r'r��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 1rsTOP 0 1sTOP 3 1N�r�r�r�r�rVr
r�rr�r�r�rr&r&r'�test_TOPWithBadArgument�sL
�


�


�


�


�
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.
        �LASTr3rNrrr&r&r'�	test_LAST�s
�
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�rr�+OK 2
N�r�r�r�r�rVr
r�rr�r�rr�rr&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 10rrrNrrr&r&r'�test_TopUpdatesHighest�s



�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 10rrrNrrr&r&r'�test_HighestOnlyProgressess




�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�sRSETrrr3Nrrr&r&r'�test_ResetClearsHighests




�z$CommandMixin.test_ResetClearsHighestN)r@rArBrCr
r�rrrrr�r	r
rrr�rrrrrrr&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.)rW�categoryzxtwisted.mail.pop3.IMailbox.getUidl may not raise IndexError for out-of-bounds message numbers: raise ValueError instead.c@sXeZdZdZeZeZdd�Ze	ge_
dd�Zege_
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.
    cCs
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�rr,r&r&r'rGsz/IndexErrorCommandTests.test_LISTWithBadArgumentcCs
t�|�S)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�r	r,r&r&r'r	Qsz/IndexErrorCommandTests.test_UIDLWithBadArgumentcCs
t�|�S)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�rr,r&r&r'r[sz.IndexErrorCommandTests.test_TOPWithBadArgumentcCs
t�|�S)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�rr,r&r&r'resz/IndexErrorCommandTests.test_RETRWithBadArgumentN)r@rArBrC�
IndexErrorr�r�r�r�_listMessageSuppressionr�r	�_getUidlSuppressionrrr&r&r&r'r>src@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)r@rArBrCr�r�r�r�r&r&r&r'rpsrc@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	Zsucceedr�re)r r%r&r&r're�s	z SyncDeferredMailbox.listMessages)N)r@rArBrCrer&r&r&r'r|src@seZdZdZeZdS)�"IndexErrorSyncDeferredCommandTestsz{
    Run all of the L{IndexErrorCommandTests} tests with a
    synchronous-Deferred returning IMailbox implementation.
    N�r@rArBrCrr�r&r&r&r'r�src@seZdZdZeZdS)�"ValueErrorSyncDeferredCommandTestsz{
    Run all of the L{ValueErrorCommandTests} tests with a
    synchronous-Deferred returning IMailbox implementation.
    Nr r&r&r&r'r!�sr!c@s"eZdZdZdd�Zddd�ZdS)�AsyncDeferredMailboxzo
    Mailbox which has a listMessages implementation which returns a Deferred
    which has not yet fired.
    cOsg|_tj|f|�|�dSrP)�waitingr�rS)r r)�kwr&r&r'rS�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	ZDeferredr#rVr�re)r r%r�r&r&r're�s	z!AsyncDeferredMailbox.listMessages)N)r@rArBrCrSrer&r&r&r'r"�sr"c@seZdZdZeZdd�ZdS)�#IndexErrorAsyncDeferredCommandTests�}
    Run all of the L{IndexErrorCommandTests} tests with an
    asynchronous-Deferred returning IMailbox implementation.
    cCs6|jjjr(|jjj��\}}|�|�qt�|�dS�zH
        Fire whatever Deferreds we've built up in our mailbox.
        N)r�r�r#�pop�callbackrr�r r�r)r&r&r'r�s
z*IndexErrorAsyncDeferredCommandTests._flushN�r@rArBrCr"r�rr&r&r&r'r%�sr%c@seZdZdZeZdd�ZdS)�#ValueErrorAsyncDeferredCommandTestsr&cCs6|jjjr(|jjj��\}}|�|�qt�|�dSr')r�r�r#r(r)rrr*r&r&r'r�s
z*ValueErrorAsyncDeferredCommandTests._flushNr+r&r&r&r'r,�sr,c@seZdZdZdd�ZdS)�
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)r@rArBrCr3r&r&r&r'r-�sr-)LrCZ
__future__rr�r�r�collectionsr�iorZzope.interfacerr.rrrZtwisted.internetr	Ztwisted.mailr
Ztwisted.protocolsrZtwisted.pythonrZtwisted.python.compatr
Ztwisted.test.proto_helpersrZ
twisted.trialrrZtwisted.cred.checkersZtwisted.cred.credentialsZtwisted.cred.portalZtwisted.internet.protocolZtwisted.mail.pop3Ztwisted.mail.protocolsZTestCaserZ	protocolsZVirtualPOP3rDrOr\rmrlr~r�r�ZMailboxr�r�ZIServerFactoryr�r�r�r�r�r�r�r�Z_listMessageDeprecationr��PendingDeprecationWarningrZ_getUidlDeprecationrrrrrr!r"r%r,r-r&r&r&r'�<module>s�i-=8W@16Y+&y����2