Edit File: test_dispatchers.cpython-36.pyc
3 �k�`�� � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z G dd � d e j �Z G d d � d e j �Z G d d � d e j �ZG dd� de j �ZG dd� de j �Zdd� Zedkr�e jdd� dS )� N)�as_bytes)� DummyOptions)� DummyProcess)� DummyPConfig)� DummyLogger)� DummyEventc @ sF e Zd Zdd� Zdd� Zdd� Zdd d �Zd d � Zdd� Zdd� Z dS )�PDispatcherTestsc C s ddl m} |� d S )Nr )�clear)�supervisor.eventsr )�selfr � r �&/usr/lib/python3.6/test_dispatchers.py�setUp s zPDispatcherTests.setUpc C s ddl m} |� d S )Nr )r )r r )r r r r r �tearDown s zPDispatcherTests.tearDownc C s ddl m} |S )Nr )� PDispatcher)�supervisor.dispatchersr )r r r r r �_getTargetClass s z PDispatcherTests._getTargetClassN�stdoutr c C s | j � |||�S )N)r )r �process�channel�fdr r r �_makeOne s zPDispatcherTests._makeOnec C s | j � }| jt|j� d S )N)r � assertRaises�NotImplementedError�readable)r �instr r r � test_readable s z PDispatcherTests.test_readablec C s | j � }| jt|j� d S )N)r r r �writable)r r r r r � test_writable! s z PDispatcherTests.test_writablec C s | j � }| j|j� d � d S )N)r � assertEqual�flush)r r r r r � test_flush% s zPDispatcherTests.test_flush)Nr r ) �__name__� __module__� __qualname__r r r r r r r! r r r r r s r c @ s e Zd Zdd� Zdd� Zdd� ZdGdd �Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFS )H�POutputDispatcherTestsc C s ddl m} |� d S )Nr )r )r r )r r r r r r * s z POutputDispatcherTests.setUpc C s ddl m} |� d S )Nr )r )r r )r r r r r r . s z POutputDispatcherTests.tearDownc C s ddl m} |S )Nr )�POutputDispatcher)r r& )r r& r r r r 2 s z&POutputDispatcherTests._getTargetClassr c C s. ddl m} |j|jd�}| j� ||| d�S )Nr )�events)r �stderr)� supervisorr' � ProcessCommunicationStdoutEventZ ProcessCommunicationStderrEventr )r r r r' r r r r 6 s z POutputDispatcherTests._makeOnec C s8 t � }t|dd�}t|�}| j|�}| j|j� d� d S )N�process1z /bin/process1F)r r r r r r )r �options�configr � dispatcherr r r r = s z$POutputDispatcherTests.test_writablec C s> t � }t|dd�}t|�}| j|�}d|_| j|j� d� d S )Nr+ z /bin/process1FT)r r r r �closedr r )r r, r- r r. r r r �test_readable_openD s z)POutputDispatcherTests.test_readable_openc C s> t � }t|dd�}t|�}| j|�}d|_| j|j� d� d S )Nr+ z /bin/process1TF)r r r r r/ r r )r r, r- r r. r r r �test_readable_closedL s z+POutputDispatcherTests.test_readable_closedc C s6 t � }t|dd�}t|�}| j|�}| jt|j� d S )Nr+ z /bin/process1)r r r r r r �handle_write_event)r r, r- r r. r r r �test_handle_write_eventT s z.POutputDispatcherTests.test_handle_write_eventc C sP t � }d|_t|dddd�}t|�}| j|�}| j|j� d � | j|jd� d S )Ns abcr+ z /bin/process1�d )�stdout_capture_maxbytes)r � readfd_resultr r r r �handle_read_event� output_buffer)r r, r- r r. r r r �test_handle_read_event[ s z-POutputDispatcherTests.test_handle_read_eventc C sh t � }d|_t|dddd�}t|�}| j|�}| j|j� | j|j� d � | j|j d� | j |j� d S )N� r+ z /bin/process1r4 )r5 ) r r6 r r r Z assertFalser/ r r7 r8 � assertTrue)r r, r- r r. r r r �%test_handle_read_event_no_data_closese s z<POutputDispatcherTests.test_handle_read_event_no_data_closesc C sh t � }t|dd�}t|�}| j|�}y td��W n |j� Y nX |jjd }| j|j d�|� d S )N�testz/test�foor z,uncaptured python exception, closing channel) r r r r � ValueError� handle_error�logger�datar; � startswith)r r, r- r r. �resultr r r �test_handle_errorq s z(POutputDispatcherTests.test_handle_errorc s� t � }t|ddddd�}t|�}d|_| j|�}d|_dd � |j_g � � fd d �}d d lm } |j |j j |� |j � | jt� �d� � d }| j|j|� | j|jd� | j|jd� d S )Nr+ z /bin/process1z/tmp/fooi� )�stdout_logfiler5 i� Tc S s dS )N�hallooor r r r r �<lambda>� s zLPOutputDispatcherTests.test_toggle_capturemode_sends_event.<locals>.<lambda>c s � j | � d S )N)�append)�event)�Lr r �doit� s zHPOutputDispatcherTests.test_toggle_capturemode_sends_event.<locals>.doitr )r' � rG )r r r �pidr Z capturemode� capturelog�getvaluer) r' � subscribe� EventTypesZPROCESS_COMMUNICATIONZtoggle_capturemoder �lenr rB )r r, r- r r. rL r' rJ r )rK r �#test_toggle_capturemode_sends_event~ s& z:POutputDispatcherTests.test_toggle_capturemode_sends_eventc C s� t � }t|dddd�}t|�}| j|�}|j� | j|jjd jd� | j|jjd j d� | j|j jd jd� | j|j jd j d� d S )Nr+ z /bin/process1z/tmp/foo)rF r T) r r r r � removelogsr � normallog�handlers�reopened�removed�childlog)r r, r- r r. r r r �test_removelogs� s z&POutputDispatcherTests.test_removelogsc C s` t � }t|dddd�}t|�}| j|�}|j� | j|jjd jd� | j|j jd jd� d S )Nr+ z /bin/process1z/tmp/foo)rF r T) r r r r � reopenlogsr rZ rW rX rV )r r, r- r r. r r r �test_reopenlogs� s z&POutputDispatcherTests.test_reopenlogsc C s� t � }ddlm} |jj|_t|dddd�}t|�}| j|�}d|_ |j � | j |j j dg� | j |jj d d� | j |j d � d S ) Nr )�loggersr+ z /bin/process1z/tmp/foo)rF �az'process1' stdout output: ar: )r r) r^ � LevelsByName�TRAC�loglevelr r r r8 � record_outputr rZ rB rA )r r, r^ r- r r. r r r �&test_record_output_log_non_capturemode� s z=POutputDispatcherTests.test_record_output_log_non_capturemodec s� t � }t|dddd�}t|�}| j|d�}d|_g � � fdd�}d d lm} |j|jj |� |j � | j t � �d � � d }| j |j |� | j |jd� d S ) Nr+ z /bin/process1T)�stdout_events_enabledr s hello from stdoutc s � j | � d S )N)rI )rJ )rK r r rL � s zWPOutputDispatcherTests.test_record_output_emits_stdout_event_when_enabled.<locals>.doitr )r' rM )r r r r r8 r) r' rQ rR �PROCESS_LOG_STDOUTrc r rS r rB )r r, r- r r. rL r' rJ r )rK r �2test_record_output_emits_stdout_event_when_enabled� s zIPOutputDispatcherTests.test_record_output_emits_stdout_event_when_enabledc sx t � }t|dddd�}t|�}| j|d�}d|_g � � fdd�}d d lm} |j|jj |� |j � | j t � �d � d S ) Nr+ z /bin/process1F)re r s hello from stdoutc s � j | � d S )N)rI )rJ )rK r r rL � s z`POutputDispatcherTests.test_record_output_does_not_emit_stdout_event_when_disabled.<locals>.doitr )r' ) r r r r r8 r) r' rQ rR rf rc r rS )r r, r- r r. rL r' r )rK r �;test_record_output_does_not_emit_stdout_event_when_disabled� s zRPOutputDispatcherTests.test_record_output_does_not_emit_stdout_event_when_disabledc s� t � }t|dddd�}t|�}| j|d�}d|_g � � fdd�}d d lm} |j|jj |� |j � | j t � �d � � d }| j |j |� | j |jd� d S ) Nr+ z /bin/process1T)�stderr_events_enabledr( s hello from stderrc s � j | � d S )N)rI )rJ )rK r r rL � s zWPOutputDispatcherTests.test_record_output_emits_stderr_event_when_enabled.<locals>.doitr )r' rM )r r r r r8 r) r' rQ rR �PROCESS_LOG_STDERRrc r rS r rB )r r, r- r r. rL r' rJ r )rK r �2test_record_output_emits_stderr_event_when_enabled� s zIPOutputDispatcherTests.test_record_output_emits_stderr_event_when_enabledc sx t � }t|dddd�}t|�}| j|d�}d|_g � � fdd�}d d lm} |j|jj |� |j � | j t � �d � d S ) Nr+ z /bin/process1F)ri r( s hello from stderrc s � j | � d S )N)rI )rJ )rK r r rL � s z`POutputDispatcherTests.test_record_output_does_not_emit_stderr_event_when_disabled.<locals>.doitr )r' ) r r r r r8 r) r' rQ rR rj rc r rS )r r, r- r r. rL r' r )rK r �;test_record_output_does_not_emit_stderr_event_when_disabled� s zRPOutputDispatcherTests.test_record_output_does_not_emit_stderr_event_when_disabledc C sx t � }ddlm} |jj|_t|ddddd�}t|�}| j|�}d|_ |j � | j |j j dg� | j |jj d d � d S ) Nr )r^ r+ z /bin/process1z/tmp/foor4 )rF r5 s! stdout string longer than a tokenz;'process1' stdout output: stdout string longer than a token)r r) r^ r` ra rb r r r r8 rc r rZ rB rA )r r, r^ r- r r. r r r �7test_record_output_capturemode_string_longer_than_token s zNPOutputDispatcherTests.test_record_output_capturemode_string_longer_than_tokenc C sZ t � }t|ddddd�}t|�}| j|�}d|_|j� | j|jjg � | j|jd� d S )Nr+ z /bin/process1z/tmp/foor4 )rF r5 r_ ) r r r r r8 rc r rZ rB )r r, r- r r. r r r �;test_record_output_capturemode_string_not_longer_than_token s zRPOutputDispatcherTests.test_record_output_capturemode_string_not_longer_than_tokenc sd ddl m} ddl m} g � � fdd�}|||� |j}|j}|d | }t� }ddlm} ||_d} t|d d | d d �} t | �} | j | �} z�|| _ | j � | j tjj| �d� | j t| j �d� | j t� �d � � d } ddl m} | j | j|� | j | j| � | j | jd� | j | jd� W d y"| jj� | jj� tj| � W n tt fk �r\ Y nX X d S )Nr )�ProcessCommunicationEvent)rQ c s � j | � d S )N)rI )rJ )r' r r rL - s zJPOutputDispatcherTests.test_stdout_capturemode_single_buffer.<locals>.doits hello)� getLoggerz/tmp/logr+ z /bin/process1i� )rF r5 rM )r* r ) r ro rQ � BEGIN_TOKEN� END_TOKENr �supervisor.loggersrp r r r r8 rc r �os�path�getsizerS r* � __class__r r rB rO �closerZ �remove�OSError�IOError)r ro rQ rL rq rr rB r, rp �logfiler- r r. rJ r* r )r' r �%test_stdout_capturemode_single_buffer' sF z<POutputDispatcherTests.test_stdout_capturemode_single_bufferc ( s� ddl m} ddl m} g � � fdd�}|||� dd l}tt|d|j��}t|j�d }|j}|j }|| | | | } d} | j | �} | d | } | d | } | d }t � }dd l m } ||_ d }t|d d|dd�}t|�}| j|�}�z�| |_|j� dd� |jjD � t|d��}| j|j� |� W d Q R X | j|j| t|�d � � | jt� �d� | j| 7 _|j� | jt� �d� dd� |jjD � t|d��}| j|j� |� W d Q R X | j|j| t|�d � � | jt� �d� | j|7 _|j� dd� |jjD � t|d��}| j|j� |d � W d Q R X | jt� �d � � d }ddl m} | j|j|� | j|j|� | j|j d� | j|j |� W d y"|j j � |jj � t j!|� W n t"t#fk �r� Y nX X d S )Nr )ro )rQ c s � j | � d S )N)rI )rJ )r' r r rL W s zMPOutputDispatcherTests.test_stdout_capturemode_multiple_buffers.<locals>.doit�letters� � :rM � )rp z/tmp/logr+ z /bin/process1i' )rF r5 c S s g | ] }|j � �qS r )r )�.0�xr r r � <listcomp>u s zSPOutputDispatcherTests.test_stdout_capturemode_multiple_buffers.<locals>.<listcomp>�rbc S s g | ] }|j � �qS r )r )r� r� r r r r� ~ s c S s g | ] }|j � �qS r )r )r� r� r r r r� � s )r* r )$r ro rQ �stringr �getattrZ ascii_letters�digitsrq rr �splitr rs rp r r r r8 rc rZ rW �openr �readrS r* rw r r rB rO rx rt ry rz r{ )r ro rQ rL r� r~ r� rq rr rB ZcolonZbroken�first�secondZthirdr, rp r| r- r r. �frJ r* r )r' r �(test_stdout_capturemode_multiple_buffersS sv z?POutputDispatcherTests.test_stdout_capturemode_multiple_buffersc C s� t � }d|_t|dddd�}t|�}| j|�}d}d}||_|j� | jt|j j �d� | j|j j d |� d |_||_|j� | jt|j j �d � | j|j j d |� d S ) NTr+ z /bin/process1z/tmp/foo)rF s4 [34mHello world... this is longer than a token![0ms+ Hello world... this is longer than a token!rM r Fr� ) r � strip_ansir r r r8 rc r rS rZ rB )r r, r- r r. �ansi�noansir r r �test_strip_ansi� s" z&POutputDispatcherTests.test_strip_ansic C s| t � }t|dd�}t|�}| j|�}| j|j|� | j|jd� | j|jd� | j|jd � | j|j d � | j|j d � d S )Nr+ z /bin/process1r r ) r r r r r r r r rO rV rZ )r r, r- r r. r r r �test_ctor_no_logfiles� s z,POutputDispatcherTests.test_ctor_no_logfilesc C s� t � }t|dddd�}t|�}| j|�}| j|j|� | j|jd� | j|jd� | j|jd � | j|j j t � | j|j |j � d S )Nr+ z /bin/process1z/tmp/foo)rF r r ) r r r r r r r r rO rV rw r rZ )r r, r- r r. r r r �test_ctor_logfile_only� s z-POutputDispatcherTests.test_ctor_logfile_onlyc C s� t � }t|dddd�}t|�}| j|�}| j|j|� | j|jd� | j|jd� | j|jj t � | j|j d � | j|j d � d S )Nr+ z /bin/process1i, )r5 r r ) r r r r r r r r rO rw r rV rZ )r r, r- r r. r r r �test_ctor_capturelog_only� s z0POutputDispatcherTests.test_ctor_capturelog_onlyc C st ddl m} t� }t|dd|d�d�}t|�}| j|�}| j|j|� | j|jd� | j|j d� | j|j d � d S )Nr )� logfile_namer+ z /bin/process1� )rF r ) �supervisor.datatypesr� r r r r r r r r rV )r r� r, r- r r. r r r �(test_ctor_stdout_logfile_is_empty_string� s z?POutputDispatcherTests.test_ctor_stdout_logfile_is_empty_stringc C s~ ddl m}m} t� }t|dd|d�|d�d�}t|�}| j|�}| j|j|� | j|j d� | j|j d� | j|j d � d S ) Nr )�booleanr� r+ z /bin/process1�NONE�false)rF � stdout_syslogr ) r� r� r� r r r r r r r r rV )r r� r� r, r- r r. r r r �5test_ctor_stdout_logfile_none_and_stdout_syslog_false� s zLPOutputDispatcherTests.test_ctor_stdout_logfile_none_and_stdout_syslog_falsec C s� ddl m}m} ddlm}m} ddlm} |� }|j|_ t |dd|d�|d�d �}t |�}| j |�} | j | j|� | j | jd � | j | jd� | j t| jj�d � | j | jjd j|� d S ) Nr )r� r� )r` � SyslogHandler)� ServerOptionsr+ z /bin/process1r� �true)rF r� r rM )r� r� r� rs r` r� �supervisor.optionsr� ra rb r r r r r r r rS rV rW rw ) r r� r� r` r� r� r, r- r r. r r r �4test_ctor_stdout_logfile_none_and_stdout_syslog_true� s zKPOutputDispatcherTests.test_ctor_stdout_logfile_none_and_stdout_syslog_truec C s� ddl m}m} ddlm}m} ddlm} |� }|j|_ t |dd|d�|d�d �}t |�}| j |�} | j | j|� | j | jd � | j | jd� | j t| jj�d � | j | jjd j|� | jj� d S ) Nr )r� r� )� FileHandlerr` )r� r+ z /bin/process1z/tmp/foor� )rF r� r rM )r� r� r� rs r� r` r� r� ra rb r r r r r r r rS rV rW rw rx ) r r� r� r� r` r� r, r- r r. r r r �4test_ctor_stdout_logfile_str_and_stdout_syslog_false s zKPOutputDispatcherTests.test_ctor_stdout_logfile_str_and_stdout_syslog_falsec s� ddl m}m} ddlm� m}m� ddlm} |� }|j |_ t |dd|d�|d�d �}t |�}| j |�}| j|j|� | j|jd � | j|jd� | jt|jj�d � | jt� fd d �|jjD ��� | jt�fdd �|jjD ��� |jj� d S )Nr )r� r� )r� r` r� )r� r+ z /bin/process1z/tmp/foor� )rF r� r r� c 3 s | ]}t |� �V qd S )N)� isinstance)r� �h)r� r r � <genexpr>! s z]POutputDispatcherTests.test_ctor_stdout_logfile_str_and_stdout_syslog_true.<locals>.<genexpr>c 3 s | ]}t |� �V qd S )N)r� )r� r� )r� r r r� # s )r� r� r� rs r� r` r� r� r� ra rb r r r r r r r rS rV rW r; �anyrx ) r r� r� r` r� r, r- r r. r )r� r� r �3test_ctor_stdout_logfile_str_and_stdout_syslog_true s& zJPOutputDispatcherTests.test_ctor_stdout_logfile_str_and_stdout_syslog_truec C sb t � }t|dd�}t|�}| j|�}t|�}| jd|k� | j|jd�d� | j|jd�|� d S )Nr+ z /bin/process1r& z"supervisor.tests.base.DummyProcessrM z (stdout)>���) r r r r �reprr; �assertNotEqual�find�endswith)r r, r- r r. �dreprr r r � test_repr' s z POutputDispatcherTests.test_reprc C sT t � }t|dd�}t|�}| j|�}|j� | j|jd� |j� | j|jd� d S )Nr+ z /bin/process1T)r r r r rx r r/ )r r, r- r r. r r r � test_close3 s z!POutputDispatcherTests.test_closeN)r )%r" r# r$ r r r r r r0 r1 r3 r9 r< rE rT r[ r] rd rg rh rk rl rm rn r} r� r� r� r� r� r� r� r� r� r� r� r� r r r r r% ) sD ,F 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!S )"�PInputDispatcherTestsc C s ddl m} |S )Nr )�PInputDispatcher)r r� )r r� r r r r ? s z%PInputDispatcherTests._getTargetClassc C s d}| j � ||d�S )N�stdinr )r )r r r r r r r C s z PInputDispatcherTests._makeOnec C s2 t d �}| j|�}d|_d|_| j|j� d� d S )Nr_ FT)r r � input_bufferr/ r r )r r r. r r r �test_writable_open_nodataG s z/PInputDispatcherTests.test_writable_open_nodatac C s2 t d �}| j|�}d|_d|_| j|j� d� d S )Nr� F)r r r� r/ r r )r r r. r r r �test_writable_open_withdataN s z1PInputDispatcherTests.test_writable_open_withdatac C s2 t d �}| j|�}d|_d|_| j|j� d� d S )Nr_ TF)r r r� r/ r r )r r r. r r r �test_writable_closed_nodataU s z1PInputDispatcherTests.test_writable_closed_nodatac C s2 t d �}| j|�}d|_d|_| j|j� d� d S )Nr� TF)r r r� r/ r r )r r r. r r r � test_writable_closed_withdata\ s z3PInputDispatcherTests.test_writable_closed_withdatac C s& t d �}| j|�}| j|j� d� d S )NF)r r r r )r r r. r r r r c s z#PInputDispatcherTests.test_readablec C sP t � }t|dd�}t|�}| j|�}d|_| j|j� d � | j|jd d� d S )Nr+ z /bin/process1�halloooor )r r r r r� r r2 �written)r r, r- r r. r r r r3 h s z-PInputDispatcherTests.test_handle_write_eventc C sZ t � }t|dd�}t|�}| j|�}| j|jd� |j� | j|jd� | j|ji � d S )Nr= z/testr: )r r r r r r� r2 r� )r r, r- r r. r r r � test_handle_write_event_nodataq s z4PInputDispatcherTests.test_handle_write_event_nodatac C s� t � }t|dd�}t|�}| j|�}d|_dd l}t|jtj |j��|_ |j � | j |jd� | j |jjd jd�� | j |jjd jd�� d S )Nr= z/testr� r r: z fd 0 closed, stopped monitoringz(stdin)>)r r r r r� �errnorz ZEPIPErt �strerror�write_exceptionr2 r r; rA rB rC r� )r r, r- r r. r� r r r �$test_handle_write_event_epipe_raised{ s z:PInputDispatcherTests.test_handle_write_event_epipe_raisedc C sZ t � }t|dd�}t|�}| j|�}d|_dd l}t|jtj |j��|_ | j t|j � d S )Nr= z/testr� r ) r r r r r� r� rz ZEBADFrt r� r� r r2 )r r, r- r r. r� r r r �'test_handle_write_event_uncaught_raised� s z=PInputDispatcherTests.test_handle_write_event_uncaught_raisedc C sd t � }t|dd�}t|�}| j|�}d|_dd |_|j� | jt|j�d� | j|j d d� d S )Nr= z/testrM r_ �2 �1 r ) r r r r Z write_acceptr� r2 r rS r� )r r, r- r r. r r r �%test_handle_write_event_over_os_limit� s z;PInputDispatcherTests.test_handle_write_event_over_os_limitc C s$ t d �}| j|�}| jt|j� d S )N)r r r r r7 )r r r. r r r r9 � s z,PInputDispatcherTests.test_handle_read_eventc C sh t � }t|dd�}t|�}| j|�}y td��W n |j� Y nX |jjd }| j|j d�|� d S )Nr= z/testr> r z,uncaptured python exception, closing channel) r r r r r? r@ rA rB r; rC )r r, r- r r. rD r r r rE � s z'PInputDispatcherTests.test_handle_errorc C sb t � }t|dd�}t|�}| j|�}t|�}| jd|k� | j|jd�d� | j|jd�|� d S )Nr+ z /bin/process1r� z"supervisor.tests.base.DummyProcessrM z(stdin)>r� ) r r r r r� r; r� r� r� )r r, r- r r. r� r r r r� � s z PInputDispatcherTests.test_reprc C sT t � }t|dd�}t|�}| j|�}|j� | j|jd� |j� | j|jd� d S )Nr+ z /bin/process1T)r r r r rx r r/ )r r, r- r r. r r r r� � s z PInputDispatcherTests.test_closeN)r" r# r$ r r r� r� r� r� r r3 r� r� r� r� r9 rE 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� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dES )F� PEventListenerDispatcherTestsc C s ddl m} |� d S )Nr )r )r r )r r r r r r � s z#PEventListenerDispatcherTests.setUpc C s ddl m} |� d S )Nr )r )r r )r r r r r r � s z&PEventListenerDispatcherTests.tearDownc C s ddl m} |S )Nr )�PEventListenerDispatcher)r r� )r r� r r r r � s z-PEventListenerDispatcherTests._getTargetClassc C s d}| j � ||d�S )Nr r )r )r r r r r r r � s z&PEventListenerDispatcherTests._makeOnec C s8 t � }t|dd�}t|�}| j|�}| j|j� d� d S )Nr+ z /bin/process1F)r r r r r r )r r, r- r r. r r r r � s z+PEventListenerDispatcherTests.test_writablec C s> t � }t|dd�}t|�}| j|�}d|_| j|j� d� d S )Nr+ z /bin/process1FT)r r r r r/ r r )r r, r- r r. r r r r0 � s z0PEventListenerDispatcherTests.test_readable_openc C s> t � }t|dd�}t|�}| j|�}d|_| j|j� d� d S )Nr+ z /bin/process1TF)r r r r r/ r r )r r, r- r r. r r r r1 � s z2PEventListenerDispatcherTests.test_readable_closedc C s6 t � }t|dd�}t|�}| j|�}| jt|j� d S )Nr+ z /bin/process1)r r r r r r r2 )r r, r- r r. r r r r3 � s z5PEventListenerDispatcherTests.test_handle_write_eventc C s� t � }t|dddd�}t|�}ddlm} |j|_| j|�}|j|_ | j |j � d � | j |j|j � | j |j d� | j t|jj�d� | j |jjd |j� d S ) Nr+ z /bin/process1z/tmp/foo)rF r )�EventListenerStatesr: rM )r r r r r� � ACKNOWLEDGED�listener_stater ZREADY_FOR_EVENTS_TOKENr6 r r7 �READY� state_bufferrS rZ rB )r r, r- r r� r. r r r �9test_handle_read_event_calls_handle_listener_state_change� s zWPEventListenerDispatcherTests.test_handle_read_event_calls_handle_listener_state_changec C sj t � }d|_t|dd�}t|�}| j|�}| j|j� d � | j|jd� ddlm } | j|j j |j � d S )Nr� r+ z /bin/process1r: r )r� ) r r6 r r r r r7 r� r r� r r� r� )r r, r- r r. r� r r r � test_handle_read_event_nodata s z;PEventListenerDispatcherTests.test_handle_read_event_nodatac C sL t � }d|_t|dd�}t|�}| j|�}| j|j� d � | j|jd � d S )Ns" supercalifragilisticexpialidociousr+ z /bin/process1)r r6 r r r r r7 rZ )r r, r- r r. r r r �%test_handle_read_event_logging_nologs s zCPEventListenerDispatcherTests.test_handle_read_event_logging_nologsc C sj t � }d|_t|dddd�}t|�}| j|�}| j|j� d � | jt|jj �d� | j|jj d d� d S )Ns" supercalifragilisticexpialidociousr+ z /bin/process1z/tmp/foo)rF rM r ) r r6 r r r r r7 rS rZ rB )r r, r- r r. r r r �'test_handle_read_event_logging_childlog s zEPEventListenerDispatcherTests.test_handle_read_event_logging_childlogc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j g � | j |j|j� d S )Nr+ z /bin/process1r )r� s whateverr: ) r r r r r� r �UNKNOWNr� r� r � handle_listener_state_changerA rB )r r, r- r r� r. r r r �.test_handle_listener_state_change_from_unknown* s zLPEventListenerDispatcherTests.test_handle_listener_state_change_from_unknownc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j d d� | j |j|j � d S )Nr+ z /bin/process1r )r� s READY r: z process1: ACKNOWLEDGED -> READY)r r r r r� r r� r� r� r r� rA rB r� )r r, r- r r� r. r r r �7test_handle_listener_state_change_acknowledged_to_ready7 s zUPEventListenerDispatcherTests.test_handle_listener_state_change_acknowledged_to_readyc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j d d� | j |j j d d � | j |j|j � d S ) Nr+ z /bin/process1r )r� s READY garbage r: z process1: ACKNOWLEDGED -> READYrM zprocess1: READY -> UNKNOWN)r r r r r� r r� r� r� r r� rA rB r� )r r, r- r r� r. r r r �6test_handle_listener_state_change_acknowledged_gobblesE s zTPEventListenerDispatcherTests.test_handle_listener_state_change_acknowledged_gobblesc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j g � | j |j|j� d S )Nr+ z /bin/process1r )r� s RE) r r r r r� r r� r� r� r r� rA rB )r r, r- r r� r. r r r �>test_handle_listener_state_change_acknowledged_to_insufficientU s z\PEventListenerDispatcherTests.test_handle_listener_state_change_acknowledged_to_insufficientc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j d d� | j |j j d d � | j |j|j � d S ) Nr+ z /bin/process1r )r� s bogus data yor: z!process1: ACKNOWLEDGED -> UNKNOWNrM z�process1: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol)r r r r r� r r� r� r� r r� rA rB r� )r r, r- r r� r. r r r �9test_handle_listener_state_change_acknowledged_to_unknownc s zWPEventListenerDispatcherTests.test_handle_listener_state_change_acknowledged_to_unknownc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j j d d� | j |j j d d � | j |j|j � d S ) Nr+ z /bin/process1r )r� s bogus data yor: zprocess1: READY -> UNKNOWNrM z�process1: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol)r r r r r� r r� r� r� r r� rA rB r� )r r, r- r r� r. r r r �2test_handle_listener_state_change_ready_to_unknownu s zPPEventListenerDispatcherTests.test_handle_listener_state_change_ready_to_unknownc C sp t � }t|dd�}t|�}ddlm} | j|�}|j|_d|_| j |j � d � | j |jd� | j |j|j� d S )Nr+ z /bin/process1r )r� s bogus data yo) r r r r r� r �BUSYr� r� r r� )r r, r- r r� r. r r r �6test_handle_listener_state_change_busy_to_insufficient� s zTPEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_insufficientc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_G dd� d�}|� |_|� |j_ ddlm } ||jj _ d|_ | j |j� d � | j |j d � | j |jjd d � | j |jjd d � | j |j|j� d S ) Nr+ z /bin/process1r )r� c @ s e Zd ZdS )ziPEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_acknowledged_procd.<locals>.DummyN)r" r# r$ r r r r �Dummy� s r� )�default_handlers RESULT 2 OKabcs abcz process1: event was processedrM z process1: BUSY -> ACKNOWLEDGED)r r r r r� r r� r� �groupr- r� �result_handlerr� r r� rA rB r� )r r, r- r r� r. r� r� r r r �<test_handle_listener_state_change_busy_to_acknowledged_procd� s( zZPEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_acknowledged_procdc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_G dd� d�}|� |_|� |j_ ddlm } ||jj _ d|_ | j |j� d � | j |j d � | j |jjd d � | j |jjd d � | j |j|j� d S ) Nr+ z /bin/process1r )r� c @ s e Zd ZdS )zlPEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_acknowledged_rejected.<locals>.DummyN)r" r# r$ r r r r r� � s r� )r� s RESULT 4 FAILabcs abcz process1: event was rejectedrM z process1: BUSY -> ACKNOWLEDGED)r r r r r� r r� r� r� r- r� r� r� r r� rA rB r� )r r, r- r r� r. r� r� r r r �?test_handle_listener_state_change_busy_to_acknowledged_rejected� s( z]PEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_acknowledged_rejectedc s ddl m} ddl m} g � � fdd�}|||� t� }t|dd�}t|�}ddlm} | j|�}|j |_ t � } | |_ d |_ | j|j� d � | j|j d � | j|jjd d � | j|jjd d � | j|jjd d� | j|j |j� | j� d j|� | j� d j | � d S )Nr )�EventRejectedEvent)rQ c s � j | � d S )N)rI )rJ )r' r r rL � s z]PEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_unknown.<locals>.doitr+ z /bin/process1)r� s bogus data r: z'process1: bad result line: 'bogus data'rM zprocess1: BUSY -> UNKNOWNr� z�process1: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol)r r� rQ r r r r r� r r� r� r rJ r� r r� rA rB r� r ) r r� rQ rL r, r- r r� r. Z current_eventr )r' r �1test_handle_listener_state_change_busy_to_unknown� s4 zOPEventListenerDispatcherTests.test_handle_listener_state_change_busy_to_unknownc C s� t � }t|dd�}t|�}ddlm} | j|�}|j|_G dd� d�}|� |_|� |j_ ddlm } ||jj _ d|_ | j |j� d � | j |j d � | j |jjd d � | j |jjd d � | j |jjd d� | j |jjd d� | j |j|j� d S )Nr+ z /bin/process1r )r� c @ s e Zd ZdS )zTPEventListenerDispatcherTests.test_handle_listener_state_busy_gobbles.<locals>.DummyN)r" r# r$ r r r r r� � s r� )r� s RESULT 2 OKbogus data r: z process1: event was processedrM z process1: BUSY -> ACKNOWLEDGEDr� z!process1: ACKNOWLEDGED -> UNKNOWN� z�process1: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol)r r r r r� r r� r� r� r- r� r� r� r r� rA rB r� )r r, r- r r� r. r� r� r r r �'test_handle_listener_state_busy_gobbles� s0 zEPEventListenerDispatcherTests.test_handle_listener_state_busy_gobblesc s� ddl m} t� }t|dd�}t|�}g � � fdd�}ddlm} ||j|� ddlm } | j |�}d d � } G d d � d �} | � |_ | � |j _ | |j j _ |j|_|jd � | jt� �d� | j|j|j� | j|jjd d� | j|jjd d� d S )Nr )rQ r+ z /bin/process1c s � j | � d S )N)rI )rJ )rK r r rL s zEPEventListenerDispatcherTests.test_handle_result_accept.<locals>.doit)r' )r� c S s d S )Nr )rJ rD r r r �handle s zGPEventListenerDispatcherTests.test_handle_result_accept.<locals>.handlec @ s e Zd ZdS )zFPEventListenerDispatcherTests.test_handle_result_accept.<locals>.DummyN)r" r# r$ r r r r r� s r� r> z process1: event was processedrM z process1: BUSY -> ACKNOWLEDGED)r rQ r r r r) r' r� r r� r r� r- r� r� r� � handle_resultr rS r� rA rB ) r rQ r, r- r rL r' r� r. r� r� r )rK r �test_handle_result_accept� s0 z7PEventListenerDispatcherTests.test_handle_result_acceptc s ddl m} t� }t|dd�}t|�}g � � fdd�}ddlm} ||j|� ddlm } | j |�}d d � } G d d � d �} | � |_ | � |j _ | |j j _ |j|_|jd � | jt� �d� | j� d j|j� | j|j|j� | j|jjd d� | j|jjd d� d S )Nr )rQ r+ z /bin/process1c s � j | � d S )N)rI )rJ )rK r r rL s zJPEventListenerDispatcherTests.test_handle_result_rejectevent.<locals>.doit)r' )r� c S s ddl m} ||��d S )Nr )� RejectEvent)r r� )rJ rD r� r r r �rejected& s zNPEventListenerDispatcherTests.test_handle_result_rejectevent.<locals>.rejectedc @ s e Zd ZdS )zKPEventListenerDispatcherTests.test_handle_result_rejectevent.<locals>.DummyN)r" r# r$ r r r r r� ) s r� r> rM z process1: event was rejectedz process1: BUSY -> ACKNOWLEDGED)r rQ r r r r) r' r� r r� r r� r- r� r� r� r� r rS rw r� rA rB ) r rQ r, r- r rL r' r� r. r� r� r )rK r � test_handle_result_rejectevent s2 z<PEventListenerDispatcherTests.test_handle_result_rejecteventc s$ ddl m} t� }t|dd�}t|�}g � � fdd�}ddlm} ||j|� ddlm } | j |�}d d � } G d d � d �} | � |_ | � |j _ | |j j _ | |j _ |j|_|jd � | jt� �d� | j� d j|j� | j|j|j� | j|jjd d� | j|jjd d� | j|jjd d� d S )Nr )rQ r+ z /bin/process1c s � j | � d S )N)rI )rJ )rK r r rL ? s zHPEventListenerDispatcherTests.test_handle_result_exception.<locals>.doit)r' )r� c S s t �d S )N)r? )rJ rD r r r � exceptionE s zMPEventListenerDispatcherTests.test_handle_result_exception.<locals>.exceptionc @ s e Zd ZdS )zIPEventListenerDispatcherTests.test_handle_result_exception.<locals>.DummyN)r" r# r$ r r r r r� G s r� r> rM z process1: event caused an errorzprocess1: BUSY -> UNKNOWNr� z�process1: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol)r rQ r r r r) r' r� r r� r r� r- r� r� r� r� r rS rw r� rA rB ) r rQ r, r- r rL r' r� r. r� r� r )rK r � test_handle_result_exception9 s8 z:PEventListenerDispatcherTests.test_handle_result_exceptionc C sh t � }t|dd�}t|�}| j|�}y td��W n |j� Y nX |jjd }| j|j d�|� d S )Nr= z/testr> r z,uncaptured python exception, closing channel) r r r r r? r@ rA rB r; rC )r r, r- r r. rD r r r rE \ s z/PEventListenerDispatcherTests.test_handle_errorc C s` t � }t|dddd�}t|�}| j|�}|j� | j|jjd jd� | j|jjd j d� d S )Nr+ z /bin/process1z/tmp/foo)rF r T) r r r r rU r rZ rW rX rY )r r, r- r r. r r r r[ i s z-PEventListenerDispatcherTests.test_removelogsc C sJ t � }t|dddd�}t|�}| j|�}|j� | j|jjd jd� d S )Nr+ z /bin/process1z/tmp/foo)rF r T) r r r r r\ r rZ rW rX )r r, r- r r. r r r r] s s z-PEventListenerDispatcherTests.test_reopenlogsc C s� t � }d|_t|dddd�}t|�}| j|�}d}d}||_|j� | jt|j j �d� | j|j j d |� d |_||_|j� | jt|j j �d � | j|j j d |� d S ) NTr+ z /bin/process1z/tmp/foo)rF s4 [34mHello world... this is longer than a token![0ms+ Hello world... this is longer than a token!rM r Fr� ) r r� r r r r6 r7 r rS rZ rB )r r, r- r r. r� r� r r r r� | s" z-PEventListenerDispatcherTests.test_strip_ansic C s` t � }t|dd�}t|�}| j|�}| j|j|� | j|jd� | j|jd� | j|jd � d S )Nr+ z /bin/process1r r ) r r r r r r r r rZ )r r, r- r r. r r r �test_ctor_nologfiles� s z2PEventListenerDispatcherTests.test_ctor_nologfilesc C sf t � }t|dddd�}t|�}| j|�}| j|j|� | j|jd� | j|jd� | j|jj t � d S )Nr+ z /bin/process1z/tmp/foo)rF r r ) r r r r r r r r rZ rw r )r r, r- r r. r r r r� � s z4PEventListenerDispatcherTests.test_ctor_logfile_onlyc C sb t � }t|dd�}t|�}| j|�}t|�}| jd|k� | j|jd�d� | j|jd�|� d S )Nr+ z /bin/process1r� z"supervisor.tests.base.DummyProcessrM z (stdout)>r� ) r r r r r� r; r� r� r� )r r, r- r r. r� r r r r� � s z'PEventListenerDispatcherTests.test_reprc C sT t � }t|dd�}t|�}| j|�}|j� | j|jd� |j� | j|jd� d S )Nr+ z /bin/process1T)r r r r rx r r/ )r r, r- r r. r r r r� � s z(PEventListenerDispatcherTests.test_closeN)%r" r# r$ r r r r r r0 r1 r3 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rE r[ r] r� r� r� r� r� r r r r r� � sD # r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �stripEscapeTestsc C s ddl m} ||�S )Nr )� stripEscapes)r r� )r �sr� r r r �_callFUT� s zstripEscapeTests._callFUTc C s | j | jd�d� d S )Nr: )r r� )r r r r �test_zero_length_string� s z(stripEscapeTests.test_zero_length_stringc C s d}d}| j | j|�|� d S )Ns4 [34mHello world... this is longer than a token![0ms+ Hello world... this is longer than a token!)r r� )r r� r� r r r � test_ansi� s zstripEscapeTests.test_ansic C s d}| j | j|�|� d S )Ns+ Hello world... this is longer than a token!)r r� )r r� r r r � test_noansi� s z stripEscapeTests.test_noansiN)r" r# 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� rt r� Zsupervisor.compatr Zsupervisor.tests.baser r r r r ZTestCaser r% r� r� r� r� r" �mainr r r r �<module> s0 y
Back to File Manager