Edit File: test_childutils.cpython-36.pyc
3 �k�`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 kr�ejd d� dS )� )�BytesION)�StringIO)� as_stringc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�ChildUtilsTestsc C s* ddl m} |ddi�}| j|d k � d S )Nr )�getRPCInterface�SUPERVISOR_SERVER_URLzhttp://localhost:9001)�supervisor.childutilsr Z assertTrue)�selfr Zrpc� r �%/usr/lib/python3.6/test_childutils.py�test_getRPCInterface s z$ChildUtilsTests.test_getRPCInterfacec C sF ddl m} |ddi�}| j|jd� | j|jd� | j|jd� d S )Nr )�getRPCTransportr zhttp://localhost:9001� )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�}||�}| j|jd� | j|jd� | j|jd� d S )Nr )r zhttp://localhost:9001ZchrismZabc123)r ZSUPERVISOR_USERNAMEZSUPERVISOR_PASSWORD)r r r r r r )r r �envr r r r �$test_getRPCTransport_with_uname_pass s z4ChildUtilsTests.test_getRPCTransport_with_uname_passc C s. ddl m} d}||�}| j|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}||�\}}| j|ddd�� | j|d� d S )Nr )� eventdataza:1 b:2 thedata r 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} tjd �}||�}| j|d � d S ) Nr )� get_asctime�� � � � � � z2009-01-18 22:14:07,000���) r% r& r' r( r) r* r r r+ )r r$ �timeZmktimer )r r$ Z timestampr r r r �test_get_asctime- s z ChildUtilsTests.test_get_asctimeN) �__name__� __module__� __qualname__r r r r r# r- r r r r r s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�!TestProcessCommunicationsProtocolc C sR ddl m} t� }|jd|� ddlm} |j}|j}| j|j � |d | � d S )Nr )�pcomms hello)�ProcessCommunicationEvent) r r2 r �send�supervisor.eventsr3 � BEGIN_TOKEN� END_TOKENr �getvalue)r r2 �stdoutr3 �begin�endr r r � test_send4 s z+TestProcessCommunicationsProtocol.test_sendc C sj ddl m} tj}zJt� }t_|jd� ddlm} |j}|j}| j |j � |d | � W d |t_X d S )Nr )r2 s hello)r3 ) r r2 �sysr9 r r5 r3 r6 r7 r r8 )r r2 �old�ior3 r: r; r r r � test_stdout= s z-TestProcessCommunicationsProtocol.test_stdoutc C sj ddl m} tj}zJt� }t_|jd� ddlm} |j}|j}| j |j � |d | � W d |t_X d S )Nr )r2 s hello)r3 ) r r2 r= �stderrr r5 r3 r6 r7 r r8 )r r2 r> r? r3 r: r; r r r � test_stderrJ s z-TestProcessCommunicationsProtocol.test_stderrN)r. r/ r0 r< r@ rB r r r r r1 3 s r1 c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) �TestEventListenerProtocolc C sf ddl m} G dd� d�}|� }t� }|j||�\}}| j|ddi� | j|d� | j|j� d� d S ) Nr )�listenerc @ s e Zd Zdd� Zdd� ZdS )z2TestEventListenerProtocol.test_wait.<locals>.Dummyc S s dS )Nzlen:5r )r r r r �readline[ s z;TestEventListenerProtocol.test_wait.<locals>.Dummy.readlinec W s dS )N�hellor )r Zignoredr r r �read] s z7TestEventListenerProtocol.test_wait.<locals>.Dummy.readN)r. r/ r0 rE rG r r r r �DummyZ s rH �len�5rF zREADY )r rD r �waitr r8 )r rD rH �stdinr9 r! r r r r � test_waitX s z#TestEventListenerProtocol.test_waitc C sF ddl m} ddlm} t|j�}t� }|j|� | j|j � |� d S )Nr )rD )�PEventListenerDispatcher) r rD �supervisor.dispatchersrN r ZREADY_FOR_EVENTS_TOKENr Zreadyr r8 )r rD rN �tokenr9 r r r � test_tokenf s z$TestEventListenerProtocol.test_tokenc C sJ ddl m} ddlm} t|j�}t� }|j|� | j|j � |d � d S )Nr )rD )rN z2 OK) r rD rO rN r �RESULT_TOKEN_STARTr �okr r8 )r rD rN r: r9 r r r �test_okn s z!TestEventListenerProtocol.test_okc C sJ ddl m} ddlm} t|j�}t� }|j|� | j|j � |d � d S )Nr )rD )rN z4 FAIL) r rD rO rN r rR r Zfailr r8 )r rD rN r: r9 r r r � test_failv s z#TestEventListenerProtocol.test_failc C s^ ddl m} ddlm} t|j�}t� }d}|j||� d|t|�|f }| j |j � |� d S )Nr )rD )rN zthe body data ya fool z%s%s %s) r rD rO rN r rR r r4 rI r r8 )r rD rN r: r9 �msgZexpectedr r r r< ~ s z#TestEventListenerProtocol.test_sendN)r. r/ r0 rM rQ rT rU r< r r r r rC W s rC c C s t jtjt �S )N)�unittestZ findTestCasesr= �modulesr. r r r r � test_suite� s rY �__main__)Z defaultTest)r? r r= r, rW Zsupervisor.compatr r ZTestCaser r1 rC rY r. �mainr r r r �<module> s +$2
Back to File Manager