File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_newtls.cpython-38.pyc
U
W[% � @ s" d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZmZ ddl
mZ dd l
mZ dd
lmZ z dd lmZ dd lmZ W n ek
r� d
ZY nX ddlmZ G dd� dej�ZG dd� de �Z eej �G dd� de ��Z G dd� de ee�Z e!� �"e �#� � d
S )z(
Tests for L{twisted.internet._newtls}.
� )�division�absolute_import)�unittest)�
interfaces)�ReactorBuilder)�ConnectableProtocol�runProtocolsWithReactor)�
SSLCreator�TLSMixin)�StartTLSClientCreator)�ContextGeneratingMixin)�
TCPCreator)�tls)�_newtlsN)� implementerc @ s e Zd ZdZesdZdd� ZdS )�BypassTLSTestsz4
Tests for the L{_newtls._BypassTLS} class.
z<Couldn't import _newtls, perhaps pyOpenSSL is old or missingc sh t � � g �G � �fdd�dt �}t�||� �}|�� | ��� g� t � }|�|� | ��� |g� dS )z�
C{_BypassTLS.loseConnection} calls C{loseConnection} on the base
class, while preserving any default argument in the base class'
C{loseConnection} implementation.
c s e Zd Z� f�fdd� ZdS )zDBypassTLSTests.test_loseConnectionPassThrough.<locals>.FakeTransportc s � � |� d S �N)�append)�selfZ _connDone)�result� �C/usr/lib/python3/dist-packages/twisted/internet/test/test_newtls.py�loseConnection- s zSBypassTLSTests.test_loseConnectionPassThrough.<locals>.FakeTransport.loseConnectionN)�__name__�
__module__� __qualname__r r ��defaultr r r �
FakeTransport, s r N)�objectr Z
_BypassTLSr � assertEqual)r r ZbypassZ
notDefaultr r r � test_loseConnectionPassThrough# s
z-BypassTLSTests.test_loseConnectionPassThroughN)r r r �__doc__r �skipr! r r r r r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � FakeProducerz'
A producer that does nothing.
c C s d S r r �r r r r �pauseProducingB s zFakeProducer.pauseProducingc C s d S r r r% r r r �resumeProducingF s z FakeProducer.resumeProducingc C s d S r r r% r r r �
stopProducingJ s zFakeProducer.stopProducingN)r r r r" r& r' r( r r r r r$ = s r$ c @ s e Zd ZdZdd� Zdd� ZdS )�ProducerProtocolzy
Register a producer, unregister it, and verify the producer hooks up to
innards of C{TLSMemoryBIOProtocol}.
c C s || _ || _d S r )�producerr )r r* r r r r �__init__V s zProducerProtocol.__init__c C sf t | jjtj�std��| j�| jd� | j� | jjj
j
� | j� � | j� | jjj
� | j� � d S )Nz#TLSMemoryBIOProtocol not hooked up.T)
�
isinstance� transport�protocolr ZTLSMemoryBIOProtocol� RuntimeError�registerProducerr* r r � _producer�unregisterProducerr r% r r r �handshakeCompleted[ s �
z#ProducerProtocol.handshakeCompletedN)r r r r" r+ r3 r r r r r) O s r) c @ s@ e Zd ZdZesdZdd� Zdd� Zdd� Zd d
� Z d d � Z
d
S )�ProducerTestsMixinzM
Test the new TLS code integrates C{TLSMemoryBIOProtocol} correctly.
z)Could not import twisted.internet._newtlsc C s6 g }t � }t| t� t||�t� � | �||dg� dS )z�
C{registerProducer} and C{unregisterProducer} on TLS transports
created as SSL from the get go are passed to the
C{TLSMemoryBIOProtocol}, not the underlying transport directly.
N)r$ r r r) r r �r r r* r r r �test_producerSSLFromStartt s �z,ProducerTestsMixin.test_producerSSLFromStartc C s6 g }t � }t| t� t||�t� � | �||dg� dS )z�
C{registerProducer} and C{unregisterProducer} on TLS transports
created by C{startTLS} are passed to the C{TLSMemoryBIOProtocol}, not
the underlying transport directly.
N)r$ r r r) r r r5 r r r �test_producerAfterStartTLS� s �z-ProducerTestsMixin.test_producerAfterStartTLSc sp | � � � | �� �g �t� �G ����fdd�dt�}G � fdd�dt�}t| |� |� t� � | ����g� dS )z�
When a producer is registered, and then startTLS is called,
the producer is re-registered with the C{TLSMemoryBIOProtocol}.
c s e Zd Z� ���fdd�ZdS )zMProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.RegisterTLSProtocolc s� | j �� �� | j ��� �rB��| j jjj� ��| j jj� n&��| j jjjj� ��| j jjj� | j �� | j �� d S r ) r- r0 �startTLSr r. r1 r* r2 r r% �r* r �
serverContext� streamingr r �connectionMade� s �
z\ProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.RegisterTLSProtocol.connectionMadeN�r r r r< r r9 r r �RegisterTLSProtocol� s r> c s e Zd Z� fdd�ZdS )zJProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.StartTLSProtocolc s | j �� � d S r )r- r8 r% ��
clientContextr r r< � s zYProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.StartTLSProtocol.connectionMadeNr= r r? r r �StartTLSProtocol� s rA N)ZgetClientContextZgetServerContextr$ r r r
r )r r; r> rA r )r@ r* r r: r; r � startTLSAfterRegisterProducer� s �z0ProducerTestsMixin.startTLSAfterRegisterProducerc C s | � d� dS )z�
When a streaming producer is registered, and then startTLS is called,
the producer is re-registered with the C{TLSMemoryBIOProtocol}.
TN�rB r% r r r �+test_startTLSAfterRegisterProducerStreaming� s z>ProducerTestsMixin.test_startTLSAfterRegisterProducerStreamingc C s | � d� dS )z�
When a non-streaming producer is registered, and then startTLS is
called, the producer is re-registered with the
C{TLSMemoryBIOProtocol}.
FNrC r% r r r �.test_startTLSAfterRegisterProducerNonStreaming� s zAProducerTestsMixin.test_startTLSAfterRegisterProducerNonStreamingN) r r r r" r r# r6 r7 rB rD rE r r r r r4 l s %r4 )$r" Z
__future__r r Z
twisted.trialr Ztwisted.internetr Z#twisted.internet.test.reactormixinsr Z&twisted.internet.test.connectionmixinsr r Z twisted.internet.test.test_tlsr r
r r Z twisted.internet.test.test_tcpr
Ztwisted.protocolsr r � ImportErrorZzope.interfacer ZTestCaser r r$ ZIHandshakeListenerr) r4 �globals�updateZmakeTestCaseClassesr r r r �<module> s*
" \