o

�
br�@srdZd
dl
mZ
ed�
Zerd
dlmZ
d
dlmZ
d
dlm	Z	
d
dl
mZmZ
d
dl
mZ
Gd	d
�d
ej�ZdS)z,
Tests for L{twisted.conch.ssh.forwarding}.
�)
�
requireModule�cryptography)
�
forwarding)
�IPv6Address)
�deterministicResolvingReactor)�MemoryReactorClock�StringTransport)
�unittestc@s(eZ
dZd
Zes
dZdd�Zdd�ZdS)�TestSSHConnectForwardingChannelzH
    Unit and integration tests for L{SSHConnectForwardingChannel}.
    zCannot run without cryptographycCs>|
jd
d}|
j
d
}|�d�
}t|��d�
}|�|�

dS)z�
        Fake that connection was established for first connectTCP request made
        on C{reactor}.

        @param reactor: Reactor on which to fake the connection.
        @type  reactor: A reactor.
        r
�N)
�peerAddress)�
tcpClients�
connectors�
buildProtocolr�getDestination�makeConnection)�self�reactor�factory�	connector�protocol�	transport�r�D/usr/lib/python3/dist-packages/twisted/conch/test/test_forwarding.py�makeTCPConnections






z1TestSSHConnectForwardingChannel.makeTCPConnectionc
Csptj
d
d�
}
t�}t|dg
�|
_|
�d�

|�|�

|�|
j�

|�	|
j
tj�
|�t
ddd�|
j
j���
dS)z�
        When a hostname is sent as part of forwarding requests, it
        is resolved using HostnameEndpoint's resolver.
        )zfwd.example.org��)
�hostportz::1N�TCPr)r�SSHConnectForwardingChannelrr�_reactor�channelOpenr�successResultOf�_channelOpenDeferred�assertIsInstance�client�SSHForwardingClient�assertEqualrr�getPeer)r�sut�
memoryReactorrrr� test_channelOpenHostnameRequests+s






�z@TestSSHConnectForwardingChannel.test_channelOpenHostnameRequestsN)�__name__�
__module__�__qualname__�__doc__r�skiprr*rrrrr
s

r
N)r.�twisted.python.reflectrr�twisted.conch.sshr�twisted.internet.addressr�$twisted.internet.test.test_endpointsr�twisted.test.proto_helpersrr�
twisted.trialr	�TestCaser
rrrr�<module>
s