File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_newtls.cpython-310.pyc
o
�b� � @ s d Z ddlmZ ddlmZmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZ ddlmZ zddlmZ dd lmZ W n eyG d
ZY nw eZdd lmZ G d d
� d
ej�ZG dd� d�Zeej�G dd� de��Z G dd� deee �Z e � � e � � � d
S )z(
Tests for L{twisted.internet._newtls}.
� )�
interfaces)�ConnectableProtocol�runProtocolsWithReactor)�ReactorBuilder)�
TCPCreator)�ContextGeneratingMixin�
SSLCreator�StartTLSClientCreator�TLSMixin)�unittest)�_newtls)�tlsN)� implementerc @ s e Zd ZdZes
dZdd� ZdS )�BypassTLSTestsz4
Tests for the L{_newtls._BypassTLS} class.
z<Couldn't import _newtls, perhaps pyOpenSSL is old or missingc sf t � � g �G � �fdd�d�}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)�self� _connDone)�result� �C/usr/lib/python3/dist-packages/twisted/internet/test/test_newtls.py�loseConnection4 s zSBypassTLSTests.test_loseConnectionPassThrough.<locals>.FakeTransport.loseConnectionN)�__name__�
__module__� __qualname__r r ��defaultr r r �
FakeTransport3 s r N)�objectr �
_BypassTLSr � assertEqual)r r �bypass�
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 � d S r r �r r r r �pauseProducingH � zFakeProducer.pauseProducingc C r' r r r( r r r �resumeProducingK r* z FakeProducer.resumeProducingc C r' r r r( r r r �
stopProducingN r* zFakeProducer.stopProducingN)r r r r$ r) r+ r, r r r r r& C 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__Y s
zProducerProtocol.__init__c C sf t | jjtj�s td��| j�| jd� | j� | jjj
j
� | j� � | j� | jjj
� | j� � d S )Nz#TLSMemoryBIOProtocol not hooked up.T)
�
isinstance� transport�protocolr
�TLSMemoryBIOProtocol� RuntimeError�registerProducerr. r r � _producer�unregisterProducerr r( r r r �handshakeCompleted] s
z#ProducerProtocol.handshakeCompletedN)r r r r$ r/ r8 r r r r r- R s r- c @ s@ e Zd ZdZes
dZdd� 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 �6 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 � �z,ProducerTestsMixin.test_producerSSLFromStartc C r: )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 r; r r r �test_producerAfterStartTLS� r= 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 ��� �r!��| j jjj� ��| j jj� n��| j jjjj� ��| j jjj� | j �� | j �� d S r ) r1 r5 �startTLSr r2 r6 r. r7 r r( �r. r �
serverContext� streamingr r �connectionMade� s
z\ProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.RegisterTLSProtocol.connectionMadeN�r r r rC r r@ r r �RegisterTLSProtocol� s rE c s e Zd Z� fdd�ZdS )zJProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.StartTLSProtocolc s | j �� � d S r )r1 r? r( ��
clientContextr r rC � s zYProducerTestsMixin.startTLSAfterRegisterProducer.<locals>.StartTLSProtocol.connectionMadeNrD r rF r r �StartTLSProtocol� s rH N)�getClientContext�getServerContextr& r r r r )r rB rE rH r )rG r. r rA rB r � startTLSAfterRegisterProducer� s �z0ProducerTestsMixin.startTLSAfterRegisterProducerc C � | � d� dS )z�
When a streaming producer is registered, and then startTLS is called,
the producer is re-registered with the C{TLSMemoryBIOProtocol}.
TN�rK r( r r r �+test_startTLSAfterRegisterProducerStreaming� s z>ProducerTestsMixin.test_startTLSAfterRegisterProducerStreamingc C rL )z�
When a non-streaming producer is registered, and then startTLS is
called, the producer is re-registered with the
C{TLSMemoryBIOProtocol}.
FNrM r( r r r �.test_startTLSAfterRegisterProducerNonStreaming� s zAProducerTestsMixin.test_startTLSAfterRegisterProducerNonStreamingN) r r r r$ r r% r< r> rK rN rO r r r r r9 l s $ r9 )!r$ �twisted.internetr �&twisted.internet.test.connectionmixinsr r �#twisted.internet.test.reactormixinsr � twisted.internet.test.test_tcpr � twisted.internet.test.test_tlsr r r r
�
twisted.trialr r �__newtls�twisted.protocolsr
� ImportError�zope.interfacer �TestCaser r& �IHandshakeListenerr- r9 �globals�update�makeTestCaseClassesr r r r �<module> s* � !^