CasperSecurity
o
�bK � @ st d Z ddlmZmZmZ ddlmZ ddlmZ ddl m
Z
G dd� de
j �Z G dd � d �Z
G d
d � d e
e
j �Zd S )
z2
Test cases for twisted.protocols.postfix module.
� )�Dict�List�Tuple)�postfix)�StringTransport)�unittestc @ s e Zd Zg d�Zdd� ZdS )�PostfixTCPMapQuoteTests))� foor )s foo bars foo%20bar)s foo bars foo%09bar)s foo
bars foo%0Abars foo%0abar)s foo
bars foo%0D%0Abars foo%0D%0abars foo%0d%0Abars foo%0d%0abar)s foo s foo%20)s foos %20fooc C sV | j D ]%}|d }|dd � }| �t�|�|d � |D ] }| �t�|�|� q qd S )Nr � )�data� assertEqualr �quote�unquote)�self�entry�raw�quoted�q� r �;/usr/lib/python3/dist-packages/twisted/test/test_postfix.py�testData! s
��z PostfixTCPMapQuoteTests.testDataN)�__name__�
__module__� __qualname__r r r r r r r s r c @ sR e Zd ZU i Zeeef ed< g Zee eef ed< dd� Z
dd� Z dd� Z d S )
�PostfixTCPMapServerTestCaser �chatc
C s� t �| j�}t� }t �� }||_||_|�|� | jD ] \}}|� |� | �
|� � |d|||� � f � |� � q|�
d� dS )z�
Test that I{get} and I{put} commands are responded to correctly by
L{postfix.PostfixTCPMapServer} when its factory is an instance of
L{postifx.PostfixTCPMapDictServerFactory}.
z For %r, expected %r but got %rN)r � PostfixTCPMapDictServerFactoryr r �PostfixTCPMapServer�service�factory�makeConnectionr � lineReceivedr �value�clear�
setTimeout�r r � transport�protocol�input�expected_outputr r r � test_chat4 s"
��
z%PostfixTCPMapServerTestCase.test_chatc C s� t �| j�}t� }t �� }||_||_|�|� | jD ] \}}|� |� | �
|� � |d� |||� � �� |�
� q|�d� dS )z�
Test that I{get} and I{put} commands are responded to correctly by
L{postfix.PostfixTCPMapServer} when its factory is an instance of
L{postifx.PostfixTCPMapDeferringDictServerFactory}.
z$For {!r}, expected {!r} but got {!r}N)r �'PostfixTCPMapDeferringDictServerFactoryr r r r r r r r! r r" �formatr# r$ r% r r r �test_deferredChatM s"
��
z-PostfixTCPMapServerTestCase.test_deferredChatc C sF G dd� d�}t �� }|� |_t� |_|�d� | �|j�� d� dS )z^
If the factory throws an exception,
error code 400 must be returned.
c @ s e Zd ZdZdd� ZdS )zCPostfixTCPMapServerTestCase.test_getException.<locals>.ErrorFactoryzE
Factory that raises an error on key lookup.
c S s t d��)NzThis is a test error)� Exception)r �keyr r r �getr s zGPostfixTCPMapServerTestCase.test_getException.<locals>.ErrorFactory.getN)r r r �__doc__r0 r r r r � ErrorFactorym s r2 s get examples 400 This is a test error
N)r r r r r& r! r r" )r r2 �serverr r r �test_getExceptiong s
z-PostfixTCPMapServerTestCase.test_getExceptionN)
r r r r r �bytes�__annotations__r r r r* r- r4 r r r r r + s
r c @ s e Zd Zddd�Zg d�ZdS )�
ValidTestss
ThisIs Foos bar really is found
)r s bar)
)s gets& 400 Command 'get' takes 1 parameters.
)s get foo bar� 500
)s put�& 400 Command 'put' takes 2 parameters.
)s put foor9 )s put foo bar baz� 500 put is not implemented yet.
)s put foo barr: )s get foos 200 ThisIs%20Foo
)s get bars' 200 %20bar%20really%20is%20found%0D%0A
)s get bazr8 )r s 400 unknown command
N)r r r r r r r r r r7 | s
� r7 N)r1 �typingr r r �twisted.protocolsr �twisted.test.proto_helpersr �
twisted.trialr �TestCaser r r7 r r r r �<module> s Q