3

�k�`�Q�@s�dd
lZdd
l
Z
ddlmZ
ddlmZ
ddlmZ
ddlmZ
Gdd�de
j�ZGdd	�d	e
j�Z	Gd
d�de
j�Z
Gdd
�d
e
j�Zdd�Ze
dkr�e
jdd�

d
S)�N)
�DummyOptions)
�DummyPConfig)
�DummyProcess)
�
DummyEventc@sLeZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dS)�"EventSubscriptionNotificationTestsc
Csd
dlm
}

g|
jdd�<dS)Nr
)
�events)�
supervisorr�	callbacks)�selfr�r�!/usr/lib/python3.6/test_events.py�setUp
s

z(EventSubscriptionNotificationTests.setUpc
Csd
dlm
}

g|
jdd�<dS)Nr
)
r)rrr	)r
rrrr�tearDowns

z+EventSubscriptionNotificationTests.tearDownc
Cs,d
dlm
}

|
jdd�
|j|
jdg
�
dS)Nr
)
r)NN)rrZ	subscribe�assertEqualr	)r
rrrr�test_subscribes


z1EventSubscriptionNotificationTests.test_subscribec
CsBd
dlm
}

dddg|
jdd�<|
jdd�
|j|
jd	d
g�
dS)Nr
)
r�
��)rr)rr)rr)rr)rr)rrr	Zunsubscriber)r
rrrr�test_unsubscribes



z3EventSubscriptionNotificationTests.test_unsubscribec
Cs6d
dlm
}

dg
|
jdd�<|
j�
|j|
jg�
dS)Nr
)
r)NN)rrr	�clearr)r
rrrr�
test_clears



z-EventSubscriptionNotificationTests.test_clearc
sNd
dlm
}

g��f
dd�}t|fg
|
jdd�<|
jt��

|j�dg
�
dS)Nr
)
rc

s�jd
�

dS)Nr)
�append)
�event)
�
Lrr�callback&s
zEEventSubscriptionNotificationTests.test_notify_true.<locals>.callbackr)rrrr	�notifyr)r
rrr)
rr�test_notify_true#s




z3EventSubscriptionNotificationTests.test_notify_truec
sZd
dlm
}

g��f
dd�}Gdd�d�}||fg
|
jdd�<|
jt��

|j�g�
dS)Nr
)
rc

s�jd
�

dS)Nr)
r)
r)
rrrr/s
zFEventSubscriptionNotificationTests.test_notify_false.<locals>.callbackc
@seZ
dZd
S)zJEventSubscriptionNotificationTests.test_notify_false.<locals>.AnotherEventN)�__name__�
__module__�__qualname__rrrr�AnotherEvent1s
r )rrr	rrr)r
rrr r)
rr�test_notify_false,s




z4EventSubscriptionNotificationTests.test_notify_falsec
s^d
dlm
}

g��f
dd�}Gdd�dt�}t|fg
|
jdd�<|
j|��

|j�dg
�
dS)Nr
)
rc

s�jd
�

dS)Nr)
r)
r)
rrrr:s
zMEventSubscriptionNotificationTests.test_notify_via_subclass.<locals>.callbackc
@seZ
dZd
S)zSEventSubscriptionNotificationTests.test_notify_via_subclass.<locals>.ASubclassEventN)rrrrrrr�ASubclassEvent<s
r"r)rrrr	rr)r
rrr"r)
rr�test_notify_via_subclass7s




z;EventSubscriptionNotificationTests.test_notify_via_subclassN)rrrr
rrrrrr!r#rrrrr	s
	rc@s�eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd+d,�Zd-d.�Zd/d0�Zd1S)2�TestEventTypesc
CsFd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessLogEventrrr)�supervisor.eventsr%r�process�pid�data)r
r%�instrrr�test_ProcessLogEvent_attributesDs





z.TestEventTypes.test_ProcessLogEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r%)
�Event)r&r%r,�
assertTrue�
issubclass)r
r%r,rrr� test_ProcessLogEvent_inheritenceKs



z/TestEventTypes.test_ProcessLogEvent_inheritencec
CsTd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessLogStdoutEventrrr�stdout)r&r0rr'r(r)�channel)r
r0r*rrr�%test_ProcessLogStdoutEvent_attributesRs





z4TestEventTypes.test_ProcessLogStdoutEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r0)
r%)r&r0r%r-r.)r
r0r%rrr�&test_ProcessLogStdoutEvent_inheritenceZs



z5TestEventTypes.test_ProcessLogStdoutEvent_inheritencec
CsTd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessLogStderrEventrrr�stderr)r&r5rr'r(r)r2)r
r5r*rrr�%test_ProcessLogStderrEvent_attributesas





z4TestEventTypes.test_ProcessLogStderrEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r5)
r%)r&r5r%r-r.)r
r5r%rrr�&test_ProcessLogStderrEvent_inheritenceis



z5TestEventTypes.test_ProcessLogStderrEvent_inheritencec
CsFd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessCommunicationEventrrr)r&r9rr'r(r))r
r9r*rrr�)test_ProcessCommunicationEvent_attributesps





z8TestEventTypes.test_ProcessCommunicationEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r9)
r,)r&r9r,r-r.)r
r9r,rrr�*test_ProcessCommunicationEvent_inheritencews



z9TestEventTypes.test_ProcessCommunicationEvent_inheritencec
CsTd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessCommunicationStdoutEventrrrr1)r&r<rr'r(r)r2)r
r<r*rrr�/test_ProcessCommunicationStdoutEvent_attributes~s





z>TestEventTypes.test_ProcessCommunicationStdoutEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r<)
r9)r&r<r9r-r.)r
r<r9rrr�0test_ProcessCommunicationStdoutEvent_inheritence�s





z?TestEventTypes.test_ProcessCommunicationStdoutEvent_inheritencec
CsTd
dlm
}

|
ddd�}|j|jd�
|j|jd�
|j|jd�
|j|jd�
dS)Nr
)
�ProcessCommunicationStderrEventrrrr6)r&r?rr'r(r)r2)r
r?r*rrr�/test_ProcessCommunicationStderrEvent_attributes�s





z>TestEventTypes.test_ProcessCommunicationStderrEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
r?)
r9)r&r?r9r-r.)r
r?r9rrr�0test_ProcessCommunicationStderrEvent_inheritence�s





z?TestEventTypes.test_ProcessCommunicationStderrEvent_inheritencec
Cs6d
dlm
}

|
dd�}|j|jd�
|j|jd�
dS)Nr
)
�RemoteCommunicationEventrr)r&rBr�typer))r
rBr*rrr�(test_RemoteCommunicationEvent_attributes�s




z7TestEventTypes.test_RemoteCommunicationEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
rB)
r,)r&rBr,r-r.)r
rBr,rrr�)test_RemoteCommunicationEvent_inheritence�s



z8TestEventTypes.test_RemoteCommunicationEvent_inheritencec
CsXd
dlm
}

t�}t|ddd�}t|�
}t�}|
||�}|j|j|�
|j|j|�
dS)Nr
)
�EventRejectedEvent�process1z
/bin/process1)	r&rFrrrrrr'r)r
rF�options�pconfig1r'Zrejected_eventrrrr�"test_EventRejectedEvent_attributes�s








z1TestEventTypes.test_EventRejectedEvent_attributesc
Cs,d
dlm
}

d
dlm}
|jt|
|��

dS)Nr
)
rF)
r,)r&rFr,�assertFalser.)r
rFr,rrr�3test_EventRejectedEvent_does_not_inherit_from_event�s



zBTestEventTypes.test_EventRejectedEvent_does_not_inherit_from_eventc
Cs4d
dlm
}

x"|
j|
j|
jfD]}|j|�

qWdS)Nr
)
r)rr�SupervisorStateChangeEvent�SupervisorRunningEventZSupervisorStoppingEvent�$_test_one_SupervisorStateChangeEvent)r
r�klassrrr�$test_all_SupervisorStateChangeEvents�s




z3TestEventTypes.test_all_SupervisorStateChangeEventscCs d
dlm
}
|jt|
|��

dS)Nr
)
rM)r&rMr-r.)r
rPrMrrrrO�s

z3TestEventTypes._test_one_SupervisorStateChangeEventc
	CsLd
dlm
}

x:|
j|
j|
j|
j|
j|
j|
j|
j	|
j
f	D]}|j|�

q6WdS)Nr
)
r)rr�ProcessStateEvent�ProcessStateStoppedEvent�ProcessStateExitedEvent�ProcessStateFatalEvent�ProcessStateBackoffEvent�ProcessStateRunningEvent�ProcessStateUnknownEvent�ProcessStateStoppingEvent�ProcessStateStartingEvent�_test_one_ProcessStateEvent)r
rrPrrr�test_all_ProcessStateEvents�s










z*TestEventTypes.test_all_ProcessStateEventscCs�d
dlm
}
d
dlm}
|jt|
|��

t�}t|ddd�}t|�
}|
||j	�}|j
|j|�
|j
|j|j	�
|j
|j
d�
dS)Nr
)
�
ProcessStates)
rRrGz
/bin/process1T)�supervisor.statesr]r&rRr-r.rrr�STARTINGrr'�
from_state�expected)r
rPr]rRrHrIr'r*rrrr[�s









z*TestEventTypes._test_one_ProcessStateEventc
Cs8d
dlm
}

x&|
j|
j|
j|
jfD]}|j|�

q"WdS)Nr
)
r)rr�	TickEvent�
Tick5Event�Tick60Event�
Tick3600Event�_test_one_TickEvent)r
rrPrrr�test_all_TickEvents�s





z"TestEventTypes.test_all_TickEventscCsFd
dlm
}
|jt|
|��

|
dd�}|j|jd�
|j|jd�
dS)Nr
)
rbrr)r&rbr-r.r�whenZsupervisord)r
rPrbr*rrrrf�s





z"TestEventTypes._test_one_TickEventc
Cs&d
dlm
}

|
d�
}|j|jd�
dS)Nr
)
�ProcessGroupAddedEvent�	myprocess)r&rir�group)r
rir*rrr�&test_ProcessGroupAddedEvent_attributes�s


z5TestEventTypes.test_ProcessGroupAddedEvent_attributesc
Cs&d
dlm
}

|
d�
}|j|jd�
dS)Nr
)
�ProcessGroupRemovedEventrj)r&rmrrk)r
rmr*rrr�(test_ProcessGroupRemovedEvent_attributes�s


z7TestEventTypes.test_ProcessGroupRemovedEvent_attributesN)rrrr+r/r3r4r7r8r:r;r=r>r@rArDrErJrLrQrOr\r[rgrfrlrnrrrrr$Cs0

	
r$c@s�eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�ZdS) �TestSerializationscCsv|
jd
�
}|d}d}i}t
|�
dkr.|d}|rnytdd�|j�D�
�
}Wn tk
rl


td|�
�
YnX||fS)N�

r
�rc
Ssg|]}
|
jdd
��qS)�
:r)
�split)�.0�
xrrr�
<listcomp>

sz3TestSerializations._deserialize.<locals>.<listcomp>z'headerdata %r could not be deserialized)rs�len�dict�
ValueError�AssertionError)r
Z
serializationr)Z
headerdata�payload�headersrrr�_deserialize

s














zTestSerializations._deserializec
s�t�}
t
|
d
d
d��t��
}ddlm}
G�f
dd�d�}||_||dd�}|j|j��
\}}|j|d	d
|�
|j|d
d
|�
|j|dd|�
|j|d�
dS)
NrGz
/bin/process1r
)
r0c
seZ
dZ�Zd
S)z=TestSerializations.test_plog_stdout_event.<locals>.DummyGroupN)rrr�configr)
rIrr�
DummyGroup
s
rr�yo�processname�	groupnamer(�
1)	rrrr&r0rkr}r{r)r
rHrGr0rrr|r{r)
rIr�test_plog_stdout_event
s










z)TestSerializations.test_plog_stdout_eventc
s�t�}
t
|
d
d
d��t��
}ddlm}
G�f
dd�d�}||_||dd�}|j|j��
\}}|j|d	d
|�
|j|d
d
|�
|j|dd|�
|j|d�
dS)
NrGz
/bin/process1r
)
r5c
seZ
dZ�Zd
S)z=TestSerializations.test_plog_stderr_event.<locals>.DummyGroupN)rrrr~r)
rIrrr%
s
rrr�r�r�r(r�)	rrrr&r5rkr}r{r)r
rHrGr5rrr|r{r)
rIr�test_plog_stderr_event 
s










z)TestSerializations.test_plog_stderr_eventc
s�t�}
t
|
d
d
d��t��
}ddlm}
G�f
dd�d�}||_||dd�}|j|j��
\}}|j|d	d
|�
|j|d
d
|�
|j|dd|�
|j|d�
dS)
NrGz
/bin/process1r
)
r<c
seZ
dZ�Zd
S)z>TestSerializations.test_pcomm_stdout_event.<locals>.DummyGroupN)rrrr~r)
rIrrr4
s
rrr�r�r�r(r�)	rrrr&r<rkr}r{r)r
rHrGr<rrr|r{r)
rIr�test_pcomm_stdout_event/
s










z*TestSerializations.test_pcomm_stdout_eventc
s�t�}
t
|
d
d
d��t��
}G�f
dd�d�}||_ddlm}
||dd�}|j|j��
\}}|j|d	d
|�
|j|d
d
|�
|j|dd|�
|j|d�
dS)
NrGz
/bin/process1c
seZ
dZ�Zd
S)z>TestSerializations.test_pcomm_stderr_event.<locals>.DummyGroupN)rrrr~r)
rIrrrB
s
rr
)
r?rr�r�r�r(r�)	rrrrkr&r?r}r{r)r
rHrGrr?rr|r{r)
rIr�test_pcomm_stderr_event>
s










z*TestSerializations.test_pcomm_stderr_eventc
CsJd
dlm
}

|
dd�}|j|j��
\}}|j|dd|�
|j|d�
dS)Nr
)
rB�fooZbarrC)r&rBr}r{r)r
rBrr|r{rrr�test_remote_comm_eventM
s






z)TestSerializations.test_remote_comm_eventc
CsFd
dlm
}

|
d�
}|j|j��
\}}|j|dd�
|j|d�
dS)Nr
)
rir�r�rq)r&rir}r{r)r
rirr|r{rrr�test_process_group_added_eventT
s





z1TestSerializations.test_process_group_added_eventc
CsFd
dlm
}

|
d�
}|j|j��
\}}|j|dd�
|j|d�
dS)Nr
)
rmr�r�rq)r&rmr}r{r)r
rmrr|r{rrr� test_process_group_removed_event[
s





z3TestSerializations.test_process_group_removed_eventc

s�d
dlm
}

d
dlm}
x�|j|jfD]�}t�}t|ddd��G�f
dd�d�}t��
}||_	|||
j
�}|j|j��
\}}	|j
t|�
d�
|j
|d	d�
|j
|d
d�
|j
|dd�
|j
|	d
�
q&WdS)Nr
)
r])
rrGz
/bin/process1c
seZ
dZ�Zd
S)zUTestSerializations.test_process_state_events_without_extra_values.<locals>.DummyGroupN)rrrr~r)
rIrrrl
s
rrr�r�r`r_rq)r^r]rrrUrXrrrrkr_r}r{rrw)
r
r]rrPrHrrGrr|r{r)
rIr�.test_process_state_events_without_extra_valuesb
s$















zATestSerializations.test_process_state_events_without_extra_valuesc

s�d
dlm
}

d
dlm}
x�|j|j|jfD]�}t�}t|ddd��G�f
dd�d�}t	��
}||_
d|_|||
j�}|j
|j��
\}}	|jt|�
d	�
|j|d
d�
|j|dd�
|j|dd
�
|j|dd�
|j|	d�
q*WdS)Nr
)
r])
rrGz
/bin/process1c
seZ
dZ�Zd
S)zITestSerializations.test_process_state_events_with_pid.<locals>.DummyGroupN)rrrr~r)
rIrrr�
s
rr�r�r�r`r_r(r�rq)r^r]rrrWrSrYrrrrkr(r_r}r{rrw)
r
r]rrPrHrrGrr|r{r)
rIr�"test_process_state_events_with_pidx
s*


















z5TestSerializations.test_process_state_events_with_pidc

sL
d
dlm
}

d
dlm}
�
x,|j|jfD�
]}t�}t|ddd��G�f
dd�d�}t��
}||_	|||
j
�}|j|j��
\}}	|j
t|�
d�
|j
|d	d�
|j
|d
d�
|j
|dd�
|j
|d
d�
|j
|	d�
d|_|||
j
�}|j|j��
\}}	|j
|d
d�
d|_|||
j
�}|j|j��
\}}	|j
|d
d�
q(WdS)Nr
)
r])
rrGz
/bin/process1c
seZ
dZ�Zd
S)zUTestSerializations.test_process_state_events_starting_and_backoff.<locals>.DummyGroupN)rrrr~r)
rIrrr�
s
rr�r�r�r`r_Ztries�
0rqrr�r�
2)r^r]rrrZrVrrrrkr_r}r{rrwZbackoff)
r
r]rrPrHrrGrr|r{r)
rIr�.test_process_state_events_starting_and_backoff�
s6
























zATestSerializations.test_process_state_events_starting_and_backoffc
	s�d
dlm
}

d
dlm}
t�}t|ddd��t��
}G�f
dd�d�}||_d|_|
j	||j
d	d
�}|j|j��
\}}|j
t|�
d�
|j
|dd�
|j
|d
d�
|j
|dd�
|j
|dd�
|j
|dd�
|j
|d�
dS)Nr
)
r)
r]rGz
/bin/process1c
seZ
dZ�Zd
S)zOTestSerializations.test_process_state_exited_event_expected.<locals>.DummyGroupN)rrrr~r)
rIrrr�
s
rrT)
ra�r�r�r(r�r`r_rarq)rrr^r]rrrrkr(rTr_r}r{rrw)	r
rr]rHrGrrr|r{r)
rIr�(test_process_state_exited_event_expected�
s&

















z;TestSerializations.test_process_state_exited_event_expectedc
	s�d
dlm
}

d
dlm}
t�}t|ddd��t��
}G�f
dd�d�}||_d|_|
j	||j
d	d
�}|j|j��
\}}|j
t|�
d�
|j
|dd�
|j
|d
d�
|j
|dd�
|j
|dd�
|j
|dd�
|j
|d�
dS)Nr
)
r)
r]rGz
/bin/process1c
seZ
dZ�Zd
S)zQTestSerializations.test_process_state_exited_event_unexpected.<locals>.DummyGroupN)rrrr~r)
rIrrr�
s
rrF)
rar�r�r�r(r�r`r_rar�rq)rrr^r]rrrrkr(rTr_r}r{rrw)	r
rr]rHrGrrr|r{r)
rIr�*test_process_state_exited_event_unexpected�
s&

















z=TestSerializations.test_process_state_exited_event_unexpectedc
CsBd
dlm
}

|
j�}|j|j��
\}}|j|i�
|j|d�
dS)Nr
)
rrq)rrrNr}r{r)r
rrr|r{rrr�test_supervisor_sc_event�
s





z+TestSerializations.test_supervisor_sc_eventc
Csbd
dlm
}

xP|
j|
j|
jfD]<}|dd�}|j|j��
\}}|j|ddi
�
|j|d�
qWdS)Nr
)
rrrrhr�rq)rrrcrdrer}r{r)r
rrPrr|r{rrr�test_tick_events�
s








z#TestSerializations.test_tick_eventsN)rrrr}r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrro
s
roc@seZ
dZd
d�Zdd�ZdS)�TestUtilityFunctionsc
Cs<d
dlm
}

x*|
jjj�D]\}}|j|
j|�
|�
qWdS)Nr
)
r)rr�
EventTypes�__dict__�itemsrZgetEventNameByType)r
r�name�valuerrr�test_getEventNameByType�
s


z,TestUtilityFunctions.test_getEventNameByTypec

Cs`d
dlm
}

|jt|
jd��

Gdd�d|
j�}z"|
jd|�
|j|
jj|k�

Wd|
j`XdS)Nr
)
r�FOOc
@seZ
dZd
S)z4TestUtilityFunctions.test_register.<locals>.FooEventN)rrrrrrr�FooEvent�
s
r�)	rrrK�hasattrr�r,�registerr-r�)r
rr�rrr�
test_register�
s




z"TestUtilityFunctions.test_registerN)rrrr�r�rrrrr��
s
r�cCstj
tjt�
S)
N)�unittestZ
findTestCases�sys�modulesrrrrr�
test_suites
r��__main__)
ZdefaultTest)r�r�Zsupervisor.tests.baserrrrZTestCaserr$ror�r�r�mainrrrr�<module>
s



:>p