← Back
Editing: test_pop3client.cpython-38.pyc
U W[�U � @ s, d dl Z d dlZd dlmZ d dlmZmZmZmZm Z d dl m Z d dl m Z d dl mZ d dlmZ d dlmZmZ d d lmZ d d lmZ d d lmZ d d lmZ zd d l m Z m Z W n e k r� d Z Z Y nX G dd� de�Z ddddddd�Z!d2dd�Z"dd� Z#G dd� dej$�Z%G dd� d�Z&G d d � d �Z'G d d � d ej$�Z(G d d!� d!ej$�Z)G d"d#� d#ej$�Z*G d$d%� d%e �Z+G d&d'� d'�Z,G d(d)� d)ej-�Z.G d*d+� d+ej$�Z/G d,d-� d-e,ej$�Z0e dk�r�e/fD ] Z1d.e1_2�q�n&e �3ed�dk�re/fD ] Z1d/e1_2�q d dl4Z5G d0d1� d1ej$�Z6dS )3� N)�directlyProvides)�reactor�defer�error�protocol� interfaces)�AdvancedPOP3Client)� InsecureAuthenticationDisallowed)�ServerErrorResponse)�pop3testserver)�basic�loopback)�log)� intToBytes)�StringTransport)�unittest)�ClientTLSContext�ServerTLSContextc @ s e Zd Zdd� ZdS )�#StringTransportWithConnectionLosingc C s | j �t�� � d S �N)r �connectionLostr ZConnectionDone��self� r �C/usr/lib/python3/dist-packages/twisted/mail/test/test_pop3client.py�loseConnection s z2StringTransportWithConnectionLosing.loseConnectionN)�__name__� __module__� __qualname__r r r r r r s r s 180s LOGIN)s TOPs LOGIN-DELAYs UIDL� STLSs USERs SASLTc C s8 t � }t|_t� }||_|�|� | r0|�d� ||fS )Ns +OK Hello! )� POP3Client�capCacheZ _capCacher r ZmakeConnection� dataReceived)�greet�p�tr r r �setUp$ s r& c C s | fdd�S )Nc S s |� S r r )�result�fr r r �<lambda>7 � zstrip.<locals>.<lambda>r )r( r r r �strip6 s r+ c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� ZdS ) �POP3ClientLoginTestsc sD t dd�\}}d|_|�dd�}|�d� � �|t��� fdd��S ) NF�r# T� username� passwords -ERR Offline for maintenance c s � � | jd d�S )Nr s Offline for maintenance�� assertEqual�args��excr r r r) C s �z;POP3ClientLoginTests.testNegativeGreeting.<locals>.<lambda>)r& �allowInsecureLogin�loginr" � assertFailurer � addCallback�r r$ r% �dr r r �testNegativeGreeting< s � �z)POP3ClientLoginTests.testNegativeGreetingc C s< t � \}}|�d�}| �|�� d� |�d� |�| jd�S )Nr. � USER username s +OK send password s send password)r& �userr1 �valuer" r8 r9 r r r � testOkUserG s z POP3ClientLoginTests.testOkUserc sH t � \}}|�d�}� �|�� d� |�d� � �|t��� fdd��S )Nr. r< s -ERR account suspended c s � � | jd d�S )Nr s account suspendedr0 r3 r r r r) V r* z2POP3ClientLoginTests.testBadUser.<locals>.<lambda>)r& r= r1 r> r" r7 r r8 r9 r r r � testBadUserO s � �z POP3ClientLoginTests.testBadUserc C s< t � \}}|�d�}| �|�� d� |�d� |�| jd�S )Nr/ � PASS password s +OK you're in! s you're in!)r& �passwordr1 r> r" r8 r9 r r r � testOkPassY s z POP3ClientLoginTests.testOkPassc sH t � \}}|�d�}� �|�� d� |�d� � �|t��� fdd��S )Nr/ rA s -ERR go away c s � � | jd d�S )Nr s go awayr0 r3 r r r r) h r* z2POP3ClientLoginTests.testBadPass.<locals>.<lambda>)r& rB r1 r> r" r7 r r8 r9 r r r � testBadPassa s � �z POP3ClientLoginTests.testBadPassc C s^ t � \}}d|_|�dd�}| �|�� d� |�d� | �|�� d� |�d� |�| jd�S ) NTr. r/ r< s +OK go ahead � USER username PASS password s +OK password accepted s password accepted)r& r5 r6 r1 r> r" r8 r9 r r r � testOkLogink s z POP3ClientLoginTests.testOkLoginc sj t � \}}d|_|�dd�}� �|�� d� |�d� � �|�� d� |�d� � �|t��� fdd ��S ) NTr. r/ r< s +OK waiting on you rE � -ERR bogus login c s � � | jd d�S �Nr s bogus loginr0 r3 r r r r) � r* z;POP3ClientLoginTests.testBadPasswordLogin.<locals>.<lambda>� r& r5 r6 r1 r> r" r7 r r8 r9 r r r �testBadPasswordLoginv s � �z)POP3ClientLoginTests.testBadPasswordLoginc sP t � \}}d|_|�dd�}� �|�� d� |�d� � �|t��� fdd��S )NTr. r/ r< rG c s � � | jd d�S rH r0 r3 r r r r) � r* z;POP3ClientLoginTests.testBadUsernameLogin.<locals>.<lambda>rI r9 r r r �testBadUsernameLogin� s � �z)POP3ClientLoginTests.testBadUsernameLoginc C s* t dd�\}}|�d� | �|jd � d S )NFr- s" +OK lalala this has no challenge �r& r" r1 ZserverChallenge�r r$ r% r r r �testServerGreeting� s z'POP3ClientLoginTests.testServerGreetingc C s* t dd�\}}|�d� | �|jd� d S )NFr- s +OK <here is the challenge> s <here is the challenge>rL rM r r r � testServerGreetingWithChallenge� s z4POP3ClientLoginTests.testServerGreetingWithChallengec C sL t dd�\}}|�d� |�dd�}| �|�� d� |�d� |�| jd�S ) NFr- s" +OK <challenge string goes here> r. r/ s0 APOP username f34f1e464d0d7927607753129cabe39a s +OK Welcome! s Welcome!)r& r" r6 r1 r> r8 r9 r r r �testAPOP� s � z POP3ClientLoginTests.testAPOPc C s> t dd�\}}|�d� |�dd�}| �|�� � | �|t�S )NFr- � +OK Howdy r. r/ )r& r" r6 Z assertFalser> r7 r r9 r r r � testInsecureLoginRaisesException� s �z5POP3ClientLoginTests.testInsecureLoginRaisesExceptionc C sp t dd�\}}t|tj� |�d� |�dd�}| �|�� d� |�� |�d� | �|�� d� |�d� |S ) z� If a server doesn't offer APOP but the transport is secured using SSL or TLS, a plaintext login should be allowed, not rejected with an InsecureAuthenticationDisallowed exception. Fr- rQ r. r/ r< s +OK rA ) r& r r Z ISSLTransportr" r6 r1 r> �clearr9 r r r � testSSLTransportConsideredSecure� s z5POP3ClientLoginTests.testSSLTransportConsideredSecureN)r r r r; r? r@ rC rD rF rJ rK rN rO rP rR rT r r r r r, ; s r, c @ s e Zd Zdd� Zdd� ZdS )� ListConsumerc C s i | _ d S r ��datar r r r �__init__� s zListConsumer.__init__c C s |\}}| j �|g ��|� d S r )rW � setdefault�append)r r' �itemr> r r r �consume� s zListConsumer.consumeN�r r r rX r\ r r r r rU � s rU c @ s e Zd Zdd� Zdd� ZdS )�MessageConsumerc C s g | _ d S r rV r r r r rX � s zMessageConsumer.__init__c C s | j �|� d S r )rW rZ �r �liner r r r\ � s zMessageConsumer.consumeNr] r r r r r^ � s r^ c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�POP3ClientListTestsc C sJ t � \}}|�� }| �|�� d� |�d� |�d� |�| jdddg�S )N� LIST � +OK Here it comes s 1 3 2 2 3 1 . � � � )r& �listSizer1 r> r" r8 r9 r r r � testListSize� s z POP3ClientListTests.testListSizec C s� t � \}}t� }|j}|�|�}| �|�� d� |�d� |�d� | �|jdgdgdgd�� |�d� | �|jdgdgdgdgdgdgd �� |�d � |�| j |�S ) Nrb rc s 1 3 2 2 3 1 rd re rf )r rf re s 5 3 6 2 7 1 )r rf re � � � � . ) r& rU r\ rg r1 r> r" rW r8 �assertIdentical�r r$ r% �cr( r: r r r �testListSizeWithConsumer� s � z,POP3ClientListTests.testListSizeWithConsumerc sF t � \}}|�� }� �|�� d� |�d� � �|t��� fdd��S )Nrb �! -ERR Fatal doom server exploded c s � � | jd d�S �Nr s Fatal doom server explodedr0 r3 r r r r) � s �z8POP3ClientListTests.testFailedListSize.<locals>.<lambda>)r& rg r1 r> r" r7 r r8 r9 r r r �testFailedListSize� s � �z&POP3ClientListTests.testFailedListSizec C sJ t � \}}|�� }| �|�� d� |�d� |�d� |�| jdddg�S )N� UIDL rc s 1 abc 2 def 3 ghi . � abcs defs ghi)r& �listUIDr1 r> r" r8 r9 r r r � testListUID� s z POP3ClientListTests.testListUIDc C sx t � \}}t� }|j}|�|�}| �|�� d� |�d� |�d� | �|jdgdgdgd�� |�d� |�| j |�S ) Nrt rc s 1 xyz 2 abc 5 mno s xyzru s mno)r rf ri rl ) r& rU r\ rv r1 r> r" rW r8 rm rn r r r �testListUIDWithConsumer s z+POP3ClientListTests.testListUIDWithConsumerc sF t � \}}|�� }� �|�� d� |�d� � �|t��� fdd��S )Nrt rq c s � � | jd d�S rr r0 r3 r r r r) s �z7POP3ClientListTests.testFailedListUID.<locals>.<lambda>)r& rv r1 r> r" r7 r r8 r9 r r r �testFailedListUID s � �z%POP3ClientListTests.testFailedListUIDN) r r r rh rp rs rw rx ry r r r r ra � s ra c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�POP3ClientMessageTestsc C s^ t � \}}|�d�}| �|�� d� |�d� |�d� |�d� |�d� |�| jddg�S ) N� � RETR 8 � +OK Message incoming � La la la here is message text s" ..Further message text tra la la rl � La la la here is message texts .Further message text tra la la�r& �retriever1 r> r" r8 r9 r r r � testRetrieve s ��z#POP3ClientMessageTests.testRetrievec C s` t � \}}t� }|j}|�d|�}| �|�� d� |�d� |�d� |�d� |�| j||�S )Nr{ r| r} r~ s ..Further message text . ) r& r^ r\ r� r1 r> r" r8 �_cbTestRetrieveWithConsumerrn r r r �testRetrieveWithConsumer+ s z/POP3ClientMessageTests.testRetrieveWithConsumerc C s" | � ||� | �|jddg� d S )Nr s .Further message text�rm r1 rW �r r' r( ro r r r r� 7 s �z2POP3ClientMessageTests._cbTestRetrieveWithConsumerc C sb t � \}}|jddd�}| �|�� d� |�d� |�d� |�d� |�d� |�| jd d g�S ) Nr{ re ��lines� TOP 8 2 � +OK 2 lines on the way � Line the first! Woop � Line the last! Bye rl � Line the first! Woop� Line the last! Byer� r9 r r r �testPartialRetrieve= s ��z*POP3ClientMessageTests.testPartialRetrievec C sn t � \}}t� }|j}|jd|dd�}| �|�� d� |�d� |�d� |�d� |�d� |�| j||�S ) Nr{ re r� r� r� r� r� rl ) r& r^ r\ r� r1 r> r" r8 �"_cbTestPartialRetrieveWithConsumerrn r r r � testPartialRetrieveWithConsumerK s z6POP3ClientMessageTests.testPartialRetrieveWithConsumerc C s" | � ||� | �|jddg� d S )Nr� r� r� r� r r r r� X s �z9POP3ClientMessageTests._cbTestPartialRetrieveWithConsumerc sH t � \}}|�d�}� �|�� d� |�d� � �|t��� fdd��S )Nr s RETR 1 rq c s � � | jd d�S rr r0 r3 r r r r) e s �z;POP3ClientMessageTests.testFailedRetrieve.<locals>.<lambda>)r& r� r1 r> r" r7 r r8 r9 r r r �testFailedRetrieve^ s � �z)POP3ClientMessageTests.testFailedRetrievec s� t � \� }� �fdd�td�D �}tdd�D ]x}��|�� dt|� d � |�� � �d� � �d t|� d � � �d t|� d � ��|�� d � � �d � q,tj|d d�S )z� Issue three retrieve calls immediately without waiting for any to succeed and make sure they all do succeed eventually. c sD g | ]<}� � |���jd t|d � d dt|d � d g��qS )� First line of rf � .� Second line of )r� r8 r1 r )�.0�i�r$ r r r � <listcomp>o s � ��zCPOP3ClientMessageTests.test_concurrentRetrieves.<locals>.<listcomp>rd rf ri s RETR s r� r� rl r� r* T)ZfireOnOneErrback) r& �ranger1 r> r rS r" r Z DeferredList)r r% Zmessagesr� r r� r �test_concurrentRetrievesi s � z/POP3ClientMessageTests.test_concurrentRetrievesN) r r r r� r� r� r� r� r� r� r� r r r r rz s rz c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dS )�POP3ClientMiscTestsc C sb t � \}}|jdd�}| �|�� d� |�d� |�d� |�| jd d d dddgddgdgd ��S ) Nr �ZuseCache� CAPA s +OK Capabilities on the way s! X Y Z A 1 2 3 B 1 2 C 1 . � 1� 2� 3)� X� Y� Z� A� B� C�r& Z capabilitiesr1 r> r" r8 r9 r r r �testCapability� s ��z"POP3ClientMiscTests.testCapabilityc C s> t � \}}|jdd�}| �|�� d� |�d� |�| ji �S )Nr r� r� � -ERR This server is lame! r� r9 r r r �testCapabilityError� s z'POP3ClientMiscTests.testCapabilityErrorc C s: t � \}}|�� }| �|�� d� |�d� |�| jd�S )N� STAT s +OK 1 1212 )rf i� )r& �statr1 r> r" r8 r9 r r r �testStat� s z POP3ClientMiscTests.testStatc sF t � \}}|�� }� �|�� d� |�d� � �|t��� fdd��S )Nr� r� c s � � | jd d�S �Nr s This server is lame!r0 r3 r r r r) � r* z3POP3ClientMiscTests.testStatError.<locals>.<lambda>)r& r� r1 r> r" r7 r r8 r9 r r r � testStatError� s � �z!POP3ClientMiscTests.testStatErrorc C s: t � \}}|�� }| �|�� d� |�d� |�| jd�S )N� NOOP s +OK No-op to you too! s No-op to you too!)r& �noopr1 r> r" r8 r9 r r r �testNoop� s z POP3ClientMiscTests.testNoopc sF t � \}}|�� }� �|�� d� |�d� � �|t��� fdd��S )Nr� r� c s � � | jd d�S r� r0 r3 r r r r) � r* z3POP3ClientMiscTests.testNoopError.<locals>.<lambda>)r& r� r1 r> r" r7 r r8 r9 r r r � testNoopError� s � �z!POP3ClientMiscTests.testNoopErrorc C s: t � \}}|�� }| �|�� d� |�d� |�| jd�S )N� RSET s +OK Reset state s Reset state)r& �resetr1 r> r" r8 r9 r r r �testRset� s z POP3ClientMiscTests.testRsetc sF t � \}}|�� }� �|�� d� |�d� � �|t��� fdd��S )Nr� r� c s � � | jd d�S r� r0 r3 r r r r) � r* z3POP3ClientMiscTests.testRsetError.<locals>.<lambda>)r& r� r1 r> r" r7 r r8 r9 r r r � testRsetError� s � �z!POP3ClientMiscTests.testRsetErrorc C s< t � \}}|�d�}| �|�� d� |�d� |�| jd�S )Nrd � DELE 4 s +OK Hasta la vista s Hasta la vista)r& �deleter1 r> r" r8 r9 r r r � testDelete� s z POP3ClientMiscTests.testDeletec sH t � \}}|�d�}� �|�� d� |�d� � �|t��� fdd��S )Nrd r� s -ERR Winner is not you. c s � � | jd d�S )Nr s Winner is not you.r0 r3 r r r r) � r* z5POP3ClientMiscTests.testDeleteError.<locals>.<lambda>)r& r� r1 r> r" r7 r r8 r9 r r r �testDeleteError� s � �z#POP3ClientMiscTests.testDeleteErrorN) r r r r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zddd�Zdd� ZdS )� SimpleClientNc C s || _ d| _d S )NT)�deferredr5 )r r� �contextFactoryr r r rX � s zSimpleClient.__init__c C s | j �d � d S r )r� �callback)r Z challenger r r �serverGreeting� s zSimpleClient.serverGreeting)N)r r r rX r� r r r r r� � s r� c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zdd � Zd d � Z dS ) �POP3HelperMixinNc C s: t �� }tj| jd�| _t|| jd�| _d| j_ || _ d S )N)r� � ) r �Deferredr ZPOP3TestServer� serverCTX�serverr� � clientCTX�client�timeout� connected)r r: r r r r& � s �zPOP3HelperMixin.setUpc C s | ` | `| `d S r )r� r� r� r r r r �tearDown� s zPOP3HelperMixin.tearDownc C s | j j�� d S r )r� � transportr )r �ignorer r r � _cbStopClient s z POP3HelperMixin._cbStopClientc C s | j j�� | jj�� |S r )r� r� r r� )r �failurer r r � _ebGeneral s zPOP3HelperMixin._ebGeneralc C s t j| j| jdd�S )NF)Znoisy)r Z loopbackTCPr� r� r r r r r s zPOP3HelperMixin.loopback) r r r r� r� r& r� r� r� r r r r r r� � s r� c @ s e Zd ZG dd� dej�ZdS )�TLSServerFactoryc @ s$ e Zd ZdZg Zdd� Zdd� ZdS )zTLSServerFactory.protocolNc C s8 g | j _| jd d � | _| j�d�D ]}| �|� q$d S )Nr )�factory�input�output�pop�sendLiner_ r r r �connectionMade s z(TLSServerFactory.protocol.connectionMadec sB � j j�|� � fdd�� j�d�D � |dkr>� j�� j� d S )Nc s g | ]}� � |��qS r )r� )r� �lr r r r� s z:TLSServerFactory.protocol.lineReceived.<locals>.<listcomp>r r )r� r� rZ r� r� r� �startTLS�contextr_ r r r � lineReceived s z&TLSServerFactory.protocol.lineReceived)r r r r� r� r� r� r r r r r s r N)r r r r Z LineReceiverr r r r r r� s r� c @ s e Zd ZdZdd� ZdS )� POP3TLSTestsz= Tests for POP3Client's support for TLS connections. c s t � �dgdddgdgddgdgg�j_t� �j_tjd�dd�}��|j� |� � j }|� � j }t � � �tt � � t� ����fdd�}|�_t�� }�fd d �|_t�|||�� �fd d �}�fd d�}��fdd�}� �fdd�} �j�|� �j�|� �j�|� �j�| � �jS )zd POP3Client.startTLS starts a TLS session over its existing TCP connection. s +OKr r� r z 127.0.0.1)Z interfacec s t ��| � � �d � d S r )r� r r� )�reason)�connLostDeferred�cpr r r > s z2POP3TLSTests.test_startTLS.<locals>.connectionLostc s � S r r r �r� r r r) C r* z,POP3TLSTests.test_startTLS.<locals>.<lambda>c s t �d� � �� S )Nz!Connected to server; starting TLS)r �msgr� �Zignoredr� r r � cbConnectedG s z/POP3TLSTests.test_startTLS.<locals>.cbConnectedc s t �d� � �� S )NzStarted TLS; disconnecting)r r� �quitr� r� r r � cbStartedTLSK s z0POP3TLSTests.test_startTLS.<locals>.cbStartedTLSc s$ t �d� � ��jddddg� d S )Nz.Disconnected; asserting correct input receiveds CAPAr s QUIT)r r� r1 r� )Zign)r �sfr r �cbDisconnectedO s �z2POP3TLSTests.test_startTLS.<locals>.cbDisconnectedc s t �d� � �� �S )NzEAsserted correct input; disconnecting client and shutting down server)r r� Z disconnect�r' )�connr� r r �cleanupU s z+POP3TLSTests.test_startTLS.<locals>.cleanup)r� r r� r r� r Z listenTCPZ addCleanupZ stopListeningZgetHostZhost�portr r� r� r r Z ClientFactoryZ connectTCPr� r8 ZaddBoth) r r� �H�Pr Zcfr� r� r� r� r )r� r� r� r r� r � test_startTLS) s8 � zPOP3TLSTests.test_startTLSN)r r r �__doc__r� r r r r r� $ s r� c @ s e Zd Zdd� ZdS )�POP3TimeoutTestsc s� � ��fdd�}�fdd�� �fdd���fdd�}d �j _d t_||g}t�jjtt|�� �j��j� �j� �j � �� � S ) Nc s& �j �dd�} | �� � | ��� | S )NZtest�twisted)r� r6 r8 � addErrback)r: ��loggedInr �timedOutr r r6 f s z+POP3TimeoutTests.testTimeout.<locals>.loginc s � � d� d S )Nz%Successfully logged in!? Impossible!)Zfailr� r r r r� l s z.POP3TimeoutTests.testTimeout.<locals>.loggedInc s | � tj� � �d � d S r )Ztrapr � TimeoutErrorr� )r� r r r r� p s z.POP3TimeoutTests.testTimeout.<locals>.timedOutc s � j �� S r )r� r� r r r r r� t s z*POP3TimeoutTests.testTimeout.<locals>.quitg{�G�z�?T) r� r� r ZTIMEOUT_RESPONSE�mapr� r8 r+ r� r� r� r )r r6 r� �methodsr r� r � testTimeoute s z POP3TimeoutTests.testTimeoutN)r r r r� r r r r r� d s r� zOpenSSL not presentzReactor doesn't support SSLc @ s e Zd ZdZdd� Zdd� ZdS )� POP3ClientModuleStructureTestszt Miscellaneous tests more to do with module/package structure than anything to do with the POP3 client. c C s | � tjjjg � dS )z� twisted.mail.pop3client.__all__ should be empty because all classes should be imported through twisted.mail.pop3. N)r1 r� �mailZ pop3client�__all__r r r r �test_all� s z'POP3ClientModuleStructureTests.test_allc C s^ dd� t �tjd t j�D �}|D ]6}|dksD| �ttjj |�� q"| �ttjj d�� q"dS )a Every public class in twisted.mail.pop3client should be available as a member of twisted.mail.pop3 with the exception of twisted.mail.pop3client.POP3Client which should be available as twisted.mail.pop3.AdvancedClient. c S s$ g | ] }|d d dks|d �qS )r �_r )r� ro r r r r� � s �z>POP3ClientModuleStructureTests.test_import.<locals>.<listcomp>ztwisted.mail.pop3clientr r N) �inspectZ getmembers�sys�modulesZisclassZ assertTrue�hasattrr� r� Zpop3)r Z publicClassesZpcr r r � test_import� s � �z*POP3ClientModuleStructureTests.test_importN)r r r r� r� r r r r r r� � s r� )T)7r r� Zzope.interfacer Ztwisted.internetr r r r r Ztwisted.mail.pop3r r r r Ztwisted.mail.testr Ztwisted.protocolsr r Ztwisted.pythonr Ztwisted.python.compatr Ztwisted.test.proto_helpersr Z twisted.trialr Ztwisted.test.ssl_helpersr r � ImportErrorr r! r& r+ ZTestCaser, rU r^ ra rz r� r� r� Z ServerFactoryr� r� r� Zcase�skipZ IReactorSSLZtwisted.mail.pop3clientr� r� r r r r �<module> sX � Gh` "@!
Save File
Cancel