Edit File: test_events.cpython-36.pyc
3 �k�`�Q � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ G dd� dej�ZG dd � d ej�Z G d d � d ej�Z G d d � d ej�Z dd� Z e dkr�ejdd� dS )� N)� DummyOptions)� DummyPConfig)� DummyProcess)� DummyEventc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )�"EventSubscriptionNotificationTestsc C s ddl m} g |jd d �< d S )Nr )�events)� supervisorr � callbacks)�selfr � r �!/usr/lib/python3.6/test_events.py�setUp s z(EventSubscriptionNotificationTests.setUpc C s ddl m} g |jd d �< d S )Nr )r )r r r )r r r r r �tearDown s z+EventSubscriptionNotificationTests.tearDownc C s, ddl m} |jd d � | j|jdg� d S )Nr )r )NN)r r Z subscribe� assertEqualr )r r r r r �test_subscribe s z1EventSubscriptionNotificationTests.test_subscribec C sB ddl m} dddg|jd d �< |jdd� | j|jd d g� d S ) Nr )r � � � )r r )r r )r r )r r )r r )r r r Z unsubscriber )r r r r r �test_unsubscribe s z3EventSubscriptionNotificationTests.test_unsubscribec C s6 ddl m} dg|jd d �< |j� | j|jg � d S )Nr )r )NN)r r r �clearr )r r r r r � test_clear s z-EventSubscriptionNotificationTests.test_clearc sN ddl m} g � � fdd�}t|fg|jd d �< |jt� � | j� dg� d S )Nr )r c s � j d� d S )Nr )�append)�event)�Lr r �callback& s zEEventSubscriptionNotificationTests.test_notify_true.<locals>.callbackr )r r r r �notifyr )r r r r )r r �test_notify_true# s z3EventSubscriptionNotificationTests.test_notify_truec sZ ddl m} g � � fdd�}G dd� d�}||fg|jd d �< |jt� � | j� g � d S )Nr )r c s � j d� d S )Nr )r )r )r r r r / s zFEventSubscriptionNotificationTests.test_notify_false.<locals>.callbackc @ s e Zd ZdS )zJEventSubscriptionNotificationTests.test_notify_false.<locals>.AnotherEventN)�__name__� __module__� __qualname__r r r r � AnotherEvent1 s r )r r r r r r )r r r r r )r r �test_notify_false, s z4EventSubscriptionNotificationTests.test_notify_falsec s^ ddl m} g � � fdd�}G dd� dt�}t|fg|jd d �< |j|� � | j� dg� d S )Nr )r c s � j d� d S )Nr )r )r )r r r r : s zMEventSubscriptionNotificationTests.test_notify_via_subclass.<locals>.callbackc @ s e Zd ZdS )zSEventSubscriptionNotificationTests.test_notify_via_subclass.<locals>.ASubclassEventN)r r r r r r r �ASubclassEvent< s r" r )r r r r r r )r r r r" r )r r �test_notify_via_subclass7 s z;EventSubscriptionNotificationTests.test_notify_via_subclassN) r r r r r r r r r r! r# r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1S )2�TestEventTypesc C sF ddl m} |ddd�}| j|jd� | j|jd� | j|jd� d S )Nr )�ProcessLogEventr r r )�supervisor.eventsr% r �process�pid�data)r r% �instr r r � test_ProcessLogEvent_attributesD s z.TestEventTypes.test_ProcessLogEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r% )�Event)r& r% r, � assertTrue� issubclass)r r% r, r r r � test_ProcessLogEvent_inheritenceK s z/TestEventTypes.test_ProcessLogEvent_inheritencec C sT ddl m} |ddd�}| j|jd� | j|jd� | j|jd� | j|jd� d S )Nr )�ProcessLogStdoutEventr r r �stdout)r& r0 r r' r( r) �channel)r r0 r* r r r �%test_ProcessLogStdoutEvent_attributesR s z4TestEventTypes.test_ProcessLogStdoutEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r0 )r% )r& r0 r% r- r. )r r0 r% r r r �&test_ProcessLogStdoutEvent_inheritenceZ s z5TestEventTypes.test_ProcessLogStdoutEvent_inheritencec C sT ddl m} |ddd�}| j|jd� | j|jd� | j|jd� | j|jd� d S )Nr )�ProcessLogStderrEventr r r �stderr)r& r5 r r' r( r) r2 )r r5 r* r r r �%test_ProcessLogStderrEvent_attributesa s z4TestEventTypes.test_ProcessLogStderrEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r5 )r% )r& r5 r% r- r. )r r5 r% r r r �&test_ProcessLogStderrEvent_inheritencei s z5TestEventTypes.test_ProcessLogStderrEvent_inheritencec C sF ddl m} |ddd�}| j|jd� | j|jd� | j|jd� d S )Nr )�ProcessCommunicationEventr r r )r& r9 r r' r( r) )r r9 r* r r r �)test_ProcessCommunicationEvent_attributesp s z8TestEventTypes.test_ProcessCommunicationEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r9 )r, )r& r9 r, r- r. )r r9 r, r r r �*test_ProcessCommunicationEvent_inheritencew s z9TestEventTypes.test_ProcessCommunicationEvent_inheritencec C sT ddl m} |ddd�}| j|jd� | j|jd� | j|jd� | j|jd� d S )Nr )� ProcessCommunicationStdoutEventr r r r1 )r& r< r r' r( r) r2 )r r< r* r r r �/test_ProcessCommunicationStdoutEvent_attributes~ s z>TestEventTypes.test_ProcessCommunicationStdoutEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r< )r9 )r& r< r9 r- r. )r r< r9 r r r �0test_ProcessCommunicationStdoutEvent_inheritence� s z?TestEventTypes.test_ProcessCommunicationStdoutEvent_inheritencec C sT ddl m} |ddd�}| j|jd� | j|jd� | j|jd� | j|jd� d S )Nr )� ProcessCommunicationStderrEventr r r r6 )r& r? r r' r( r) r2 )r r? r* r r r �/test_ProcessCommunicationStderrEvent_attributes� s z>TestEventTypes.test_ProcessCommunicationStderrEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )r? )r9 )r& r? r9 r- r. )r r? r9 r r r �0test_ProcessCommunicationStderrEvent_inheritence� s z?TestEventTypes.test_ProcessCommunicationStderrEvent_inheritencec C s6 ddl m} |dd�}| j|jd� | j|jd� d S )Nr )�RemoteCommunicationEventr r )r& rB r �typer) )r rB r* r r r �(test_RemoteCommunicationEvent_attributes� s z7TestEventTypes.test_RemoteCommunicationEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )rB )r, )r& rB r, r- r. )r rB r, r r r �)test_RemoteCommunicationEvent_inheritence� s z8TestEventTypes.test_RemoteCommunicationEvent_inheritencec C sX ddl m} t� }t|ddd�}t|�}t� }|||�}| j|j|� | j|j|� d S )Nr )�EventRejectedEvent�process1z /bin/process1) r& rF r r r r r r' r )r rF �options�pconfig1r' Zrejected_eventr r r r �"test_EventRejectedEvent_attributes� s z1TestEventTypes.test_EventRejectedEvent_attributesc C s, ddl m} ddl m} | jt||�� d S )Nr )rF )r, )r& rF r, � assertFalser. )r rF r, r r r �3test_EventRejectedEvent_does_not_inherit_from_event� s zBTestEventTypes.test_EventRejectedEvent_does_not_inherit_from_eventc C s4 ddl m} x"|j|j|jfD ]}| j|� q W d S )Nr )r )r r �SupervisorStateChangeEvent�SupervisorRunningEventZSupervisorStoppingEvent�$_test_one_SupervisorStateChangeEvent)r r �klassr r r �$test_all_SupervisorStateChangeEvents� s z3TestEventTypes.test_all_SupervisorStateChangeEventsc C s ddl m} | jt||�� d S )Nr )rM )r& rM r- r. )r rP rM r r r rO � s z3TestEventTypes._test_one_SupervisorStateChangeEventc C sL ddl m} x:|j|j|j|j|j|j|j|j |j f D ]}| j |� q6W d S )Nr )r ) r r �ProcessStateEvent�ProcessStateStoppedEvent�ProcessStateExitedEvent�ProcessStateFatalEvent�ProcessStateBackoffEvent�ProcessStateRunningEvent�ProcessStateUnknownEvent�ProcessStateStoppingEvent�ProcessStateStartingEvent�_test_one_ProcessStateEvent)r r rP r r r �test_all_ProcessStateEvents� s z*TestEventTypes.test_all_ProcessStateEventsc C s� ddl m} ddlm} | jt||�� t� }t|ddd�}t|�}|||j �}| j |j |� | j |j |j � | j |j d� d S )Nr )� ProcessStates)rR rG z /bin/process1T)�supervisor.statesr] r& rR r- r. r r r �STARTINGr r' � from_state�expected)r rP r] rR rH rI r' r* r r r r[ � s z*TestEventTypes._test_one_ProcessStateEventc C s8 ddl m} x&|j|j|j|jfD ]}| j|� q"W d S )Nr )r )r r � TickEvent� Tick5Event� Tick60Event� Tick3600Event�_test_one_TickEvent)r r rP r r r �test_all_TickEvents� s z"TestEventTypes.test_all_TickEventsc C sF ddl m} | jt||�� |dd�}| j|jd� | j|jd� d S )Nr )rb r r )r& rb r- r. r �whenZ supervisord)r rP rb r* r r r rf � s z"TestEventTypes._test_one_TickEventc C s& ddl m} |d�}| j|jd� d S )Nr )�ProcessGroupAddedEvent� myprocess)r& ri r �group)r ri r* r r r �&test_ProcessGroupAddedEvent_attributes� s z5TestEventTypes.test_ProcessGroupAddedEvent_attributesc C s& ddl m} |d�}| j|jd� d S )Nr )�ProcessGroupRemovedEventrj )r& rm r rk )r rm r* r r r �(test_ProcessGroupRemovedEvent_attributes� s z7TestEventTypes.test_ProcessGroupRemovedEvent_attributesN)r r r r+ r/ r3 r4 r7 r8 r: r; r= r> r@ rA rD rE rJ rL rQ rO r\ r[ rg rf rl rn r r r r r$ C s0 r$ c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd S ) �TestSerializationsc C sv |j d�}|d }d}i }t|�dkr.|d }|rnytdd� |j � D ��}W n tk rl td| ��Y nX ||fS )N� r � r c S s g | ]}|j d d��qS )�:r )�split)�.0�xr r r � <listcomp> s z3TestSerializations._deserialize.<locals>.<listcomp>z'headerdata %r could not be deserialized)rs �len�dict� ValueError�AssertionError)r Z serializationr) Z headerdata�payload�headersr r r � _deserialize s z TestSerializations._deserializec s� t � }t|ddd�� t� �}ddlm} G � fdd�d�}||_||dd�}| j|j� �\}}| j|d d|� | j|d d|� | j|d d |� | j|d� d S ) NrG z /bin/process1r )r0 c s e Zd Z� ZdS )z=TestSerializations.test_plog_stdout_event.<locals>.DummyGroupN)r r r �configr )rI r r � DummyGroup s r r �yo� processname� groupnamer( �1) r r r r& r0 rk r} r{ r )r rH rG r0 r r r| r{ r )rI r �test_plog_stdout_event s z)TestSerializations.test_plog_stdout_eventc s� t � }t|ddd�� t� �}ddlm} G � fdd�d�}||_||dd�}| j|j� �\}}| j|d d|� | j|d d|� | j|d d |� | j|d� d S ) NrG z /bin/process1r )r5 c s e Zd Z� ZdS )z=TestSerializations.test_plog_stderr_event.<locals>.DummyGroupN)r r r r~ r )rI r r r % s r r r� r� r� r( r� ) r r r r& r5 rk r} r{ r )r rH rG r5 r r r| r{ r )rI r �test_plog_stderr_event s z)TestSerializations.test_plog_stderr_eventc s� t � }t|ddd�� t� �}ddlm} G � fdd�d�}||_||dd�}| j|j� �\}}| j|d d|� | j|d d|� | j|d d |� | j|d� d S ) NrG z /bin/process1r )r< c s e Zd Z� ZdS )z>TestSerializations.test_pcomm_stdout_event.<locals>.DummyGroupN)r r r r~ r )rI r r r 4 s r r r� r� r� r( r� ) r r r r& r< rk r} r{ r )r rH rG r< r r r| r{ r )rI r �test_pcomm_stdout_event/ s z*TestSerializations.test_pcomm_stdout_eventc s� t � }t|ddd�� t� �}G � fdd�d�}||_ddlm} ||dd�}| j|j� �\}}| j|d d|� | j|d d|� | j|d d |� | j|d� d S ) NrG z /bin/process1c s e Zd Z� ZdS )z>TestSerializations.test_pcomm_stderr_event.<locals>.DummyGroupN)r r r r~ r )rI r r r B s r r )r? r r� r� r� r( r� ) r r r rk r& r? r} r{ r )r rH rG r r? r r| r{ r )rI r �test_pcomm_stderr_event> s z*TestSerializations.test_pcomm_stderr_eventc C sJ ddl m} |dd�}| j|j� �\}}| j|d d|� | j|d� d S )Nr )rB �fooZbarrC )r& rB r} r{ r )r rB r r| r{ r r r �test_remote_comm_eventM s z)TestSerializations.test_remote_comm_eventc C sF ddl m} |d�}| j|j� �\}}| j|d d� | j|d� d S )Nr )ri r� r� rq )r& ri r} r{ r )r ri r r| r{ r r r � test_process_group_added_eventT s z1TestSerializations.test_process_group_added_eventc C sF ddl m} |d�}| j|j� �\}}| j|d d� | j|d� d S )Nr )rm r� r� rq )r& rm r} r{ r )r rm r r| r{ r r r � test_process_group_removed_event[ s z3TestSerializations.test_process_group_removed_eventc s� ddl m} ddlm} x�|j|jfD ]�}t� }t|ddd�� G � fdd�d�}t� �}||_ |||j �}| j |j � �\}} | j t|�d� | j |d d� | j |d d� | j |d d � | j | d � q&W d S )Nr )r] )r rG z /bin/process1c s e Zd Z� ZdS )zUTestSerializations.test_process_state_events_without_extra_values.<locals>.DummyGroupN)r r r r~ r )rI r r r l s r r r� r� r` r_ rq )r^ r] r r rU rX r r r rk r_ r} r{ r rw ) r r] r rP rH r rG r r| r{ r )rI r �.test_process_state_events_without_extra_valuesb s$ zATestSerializations.test_process_state_events_without_extra_valuesc s� ddl m} ddlm} x�|j|j|jfD ]�}t� }t|ddd�� G � fdd�d�}t � �}||_ d|_ |||j �}| j |j� �\}} | jt|�d � | j|d d� | j|d d� | j|d d � | j|d d� | j| d� q*W d S )Nr )r] )r rG z /bin/process1c s e Zd Z� ZdS )zITestSerializations.test_process_state_events_with_pid.<locals>.DummyGroupN)r r r r~ r )rI r r r � s r r � r� r� r` r_ r( r� rq )r^ r] r r rW rS rY r r r rk r( r_ r} r{ r rw ) r r] r rP rH r rG r r| r{ r )rI r �"test_process_state_events_with_pidx s* z5TestSerializations.test_process_state_events_with_pidc sL ddl m} ddlm} �x,|j|jfD �]}t� }t|ddd�� G � fdd�d�}t� �}||_ |||j �}| j |j � �\}} | j t|�d� | j |d d� | j |d d� | j |d d � | j |d d� | j | d� d|_|||j �}| j |j � �\}} | j |d d� d|_|||j �}| j |j � �\}} | j |d d� q(W d S )Nr )r] )r rG z /bin/process1c s e Zd Z� ZdS )zUTestSerializations.test_process_state_events_starting_and_backoff.<locals>.DummyGroupN)r r r r~ r )rI r r r � s r r� r� r� r` r_ Ztries�0rq r r� r �2)r^ r] r r rZ rV r r r rk r_ r} r{ r rw Zbackoff) r r] r rP rH r rG r r| r{ r )rI r �.test_process_state_events_starting_and_backoff� s6 zATestSerializations.test_process_state_events_starting_and_backoffc s� ddl m} ddlm} t� }t|ddd�� t� �}G � fdd�d�}||_d|_|j ||j d d �}| j |j � �\}}| j t|�d � | j |d d� | j |d d� | j |d d� | j |d d� | j |d d� | j |d� d S )Nr )r )r] rG z /bin/process1c s e Zd Z� ZdS )zOTestSerializations.test_process_state_exited_event_expected.<locals>.DummyGroupN)r r r r~ r )rI r r r � s r r T)ra � r� r� r( r� r` r_ ra rq )r r r^ r] r r r rk r( rT r_ r} r{ r rw ) r r r] rH rG r r r| r{ r )rI r �(test_process_state_exited_event_expected� s& z;TestSerializations.test_process_state_exited_event_expectedc s� ddl m} ddlm} t� }t|ddd�� t� �}G � fdd�d�}||_d|_|j ||j d d �}| j |j � �\}}| j t|�d � | j |d d� | j |d d� | j |d d� | j |d d� | j |d d� | j |d� d S )Nr )r )r] rG z /bin/process1c s e Zd Z� ZdS )zQTestSerializations.test_process_state_exited_event_unexpected.<locals>.DummyGroupN)r r r r~ r )rI r r r � s r r F)ra r� r� r� r( r� r` r_ ra r� rq )r r r^ r] r r r rk r( rT r_ r} r{ r rw ) r r r] rH rG r r r| r{ r )rI r �*test_process_state_exited_event_unexpected� s& z=TestSerializations.test_process_state_exited_event_unexpectedc C sB ddl m} |j� }| j|j� �\}}| j|i � | j|d� d S )Nr )r rq )r r rN r} r{ r )r r r r| r{ r r r �test_supervisor_sc_event� s z+TestSerializations.test_supervisor_sc_eventc C sb ddl m} xP|j|j|jfD ]<}|dd�}| j|j� �\}}| j|ddi� | j|d� q W d S )Nr )r r r rh r� rq )r r rc rd re r} r{ r )r r rP r r| r{ r r r �test_tick_events� s z#TestSerializations.test_tick_eventsN)r r r r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r ro s ro c @ s e Zd Zdd� Zdd� ZdS )�TestUtilityFunctionsc C s< ddl m} x*|jjj� D ]\}}| j|j|�|� qW d S )Nr )r )r r � EventTypes�__dict__�itemsr ZgetEventNameByType)r r �name�valuer r r �test_getEventNameByType� s z,TestUtilityFunctions.test_getEventNameByTypec C s` ddl m} | jt|jd�� G dd� d|j�}z"|jd|� | j|jj|k� W d |j`X d S )Nr )r �FOOc @ s e Zd ZdS )z4TestUtilityFunctions.test_register.<locals>.FooEventN)r r r r r r r �FooEvent� s r� ) r r rK �hasattrr� r, �registerr- r� )r r r� r r r � test_register� s z"TestUtilityFunctions.test_registerN)r r r r� r� r r r r r� � s r� c C s t jtjt �S )N)�unittestZ findTestCases�sys�modulesr r r r r � test_suite s r� �__main__)Z defaultTest)r� r� Zsupervisor.tests.baser r r r ZTestCaser r$ ro r� r� r �mainr r r r �<module> s : > p
Back to File Manager