CasperSecurity
o
VƧ\k � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ G dd� dej�Z G dd� dej�Z
G d d
� d
ej�Z d d � Z e
d
krIejd d� dS dS )� )�BytesION)�StringIO)� as_stringc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�ChildUtilsTestsc C s* ddl m} |ddi�}| �|d u� d S )Nr )�getRPCInterface�SUPERVISOR_SERVER_URL�http://localhost:9001)�supervisor.childutilsr �
assertTrue)�selfr �rpc� r
�B/usr/lib/python3/dist-packages/supervisor/tests/test_childutils.py�test_getRPCInterface s z$ChildUtilsTests.test_getRPCInterfacec C sF ddl m} |ddi�}| �|jd� | �|jd� | �|jd� d S )Nr ��getRPCTransportr r � �r r � assertEqual�username�password� serverurl)r r �tr
r
r �"test_getRPCTransport_no_uname_pass s
z2ChildUtilsTests.test_getRPCTransport_no_uname_passc C sN ddl m} dddd�}||�}| �|jd� | �|jd� | �|jd� d S )Nr r r �chrism�abc123)r �SUPERVISOR_USERNAME�SUPERVISOR_PASSWORDr )r r �envr r
r
r �$test_getRPCTransport_with_uname_pass s �z4ChildUtilsTests.test_getRPCTransport_with_uname_passc C s. ddl m} d}||�}| �|ddd�� d S )Nr )� get_headersza:1 b:2�1�2��a�b)r r r )r r �line�resultr
r
r �test_get_headers s z ChildUtilsTests.test_get_headersc C s> ddl m} d}||�\}}| �|ddd�� | �|d� d S )Nr )� eventdataza:1 b:2
thedata
r! r" r# zthedata
)r r) r )r r) �payload�headers�datar
r
r �test_eventdata& s
z ChildUtilsTests.test_eventdatac C s. ddl m} t�d�}||�}| �|d� d S )Nr )� get_asctime) i� � � � � � r r ���z2009-01-18 22:14:07,000)r r. �time�mktimer )r r. � timestampr' r
r
r �test_get_asctime- s
z ChildUtilsTests.test_get_asctimeN) �__name__�
__module__� __qualname__r r r r( r- r8 r
r
r
r r s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�!TestProcessCommunicationsProtocolc C sR ddl m} t� }|�d|� ddlm} |j}|j}| �|� � |d | � d S �Nr )�pcomms hello)�ProcessCommunicationEvent)
r r> r �send�supervisor.eventsr? � BEGIN_TOKEN� END_TOKENr �getvalue)r r> �stdoutr? �begin�endr
r
r � test_send4 s z+TestProcessCommunicationsProtocol.test_sendc C �n ddl m} tj}z)t� }t_|�d� ddlm} |j}|j}| � |�
� |d | � W |t_d S |t_w r= ) r r> �sysrE r rA r? rB rC r rD �r r> �old�ior? rF rG r
r
r � test_stdout= �
z-TestProcessCommunicationsProtocol.test_stdoutc C rI r= ) r r> rJ �stderrr rA r? rB rC r rD rK r
r
r � test_stderrJ rO z-TestProcessCommunicationsProtocol.test_stderrN)r9 r: r; rH rN rQ r
r
r
r r< 3 s
r< c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestEventListenerProtocolc C sf ddl m} G dd� d�}|� }t� }|�||�\}}| �|ddi� | �|d� | �|�� d� d S ) Nr ��listenerc @ s e Zd Zdd� Zdd� ZdS )z2TestEventListenerProtocol.test_wait.<locals>.Dummyc S � dS )Nzlen:5r
)r r
r
r �readline[ � z;TestEventListenerProtocol.test_wait.<locals>.Dummy.readlinec W rU )N�hellor
)r �ignoredr
r
r �read] rW z7TestEventListenerProtocol.test_wait.<locals>.Dummy.readN)r9 r: r; rV rZ r
r
r
r �DummyZ s r[ �len�5rX zREADY
)r rT r �waitr rD )r rT r[ �stdinrE r+ r* r
r
r � test_waitX s z#TestEventListenerProtocol.test_waitc C sF ddl m} ddlm} t|j�}t� }|�|� | �|� � |� d S )Nr rS ��PEventListenerDispatcher)
r rT �supervisor.dispatchersrb r �READY_FOR_EVENTS_TOKENr �readyr rD )r rT rb �tokenrE r
r
r �
test_tokenf s
z$TestEventListenerProtocol.test_tokenc C �J ddl m} ddlm} t|j�}t� }|�|� | �|� � |d � d S )Nr rS ra z2
OK)
r rT rc rb r �RESULT_TOKEN_STARTr �okr rD �r rT rb rF rE r
r
r �test_okn �
z!TestEventListenerProtocol.test_okc C rh )Nr rS ra z4
FAIL)
r rT rc rb r ri r �failr rD rk r
r
r � test_failv rm z#TestEventListenerProtocol.test_failc C s^ ddl m} ddlm} t|j�}t� }d}|�||� d|t|�|f }| � |�
� |� d S )Nr rS ra zthe body data ya fool
z%s%s
%s) r rT rc rb r ri r r@ r\ r rD )r rT rb rF rE �msg�expectedr
r
r rH ~ s
z#TestEventListenerProtocol.test_sendN)r9 r: r; r` rg rl ro rH r
r
r
r rR W s rR c C s t �tjt �S )N)�unittest�
findTestCasesrJ �modulesr9 r
r
r
r �
test_suite� s ru �__main__)� defaultTest)rM r rJ r5 rr �supervisor.compatr r �TestCaser r< rR ru r9 �mainr
r
r
r �<module> s +$2�