CasperSecurity
o
VƧ\5A � @ s, d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z d dlm
Z
m Z d dlm Z e
ZG dd� dej�ZG dd � d ej�Ze � rIejZneZG d
d � d e�Ze
� rZejZneZG d d
� d
e�ZG dd� de
�ZG dd� de
�ZG dd� de
�ZG dd� de
�Zdd� Zedkr�ej dd� dS dS )� N)�Mock)� SelectPoller�
PollPoller� KQueuePoller)�implements_poll�implements_kqueue)� DummyOptionsc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�BasePollerTestsc C s ddl m} ||�S )Nr )�
BasePoller)�supervisor.pollerr
)�self�optionsr
� r �>/usr/lib/python3/dist-packages/supervisor/tests/test_poller.py�_makeOne s zBasePollerTests._makeOnec C � | � d �}| �t|jd � d S �N)r � assertRaises�NotImplementedError�register_readable�r �instr r r �test_register_readable �
z&BasePollerTests.test_register_readablec C r r )r r r �register_writabler r r r �test_register_writable r z&BasePollerTests.test_register_writablec C r r )r r r �unregister_readabler r r r �test_unregister_readable r z(BasePollerTests.test_unregister_readablec C r r )r r r �unregister_writabler r r r �test_unregister_writable! r z(BasePollerTests.test_unregister_writablec C r r )r r r �pollr r r r � test_poll% r zBasePollerTests.test_pollc C � | � d �}| �|�� d � d S r )r � assertEqual�before_daemonizer r r r �test_before_daemonize) r z%BasePollerTests.test_before_daemonizec C r" r )r r# �after_daemonizer r r r �test_after_daemonize- r z$BasePollerTests.test_after_daemonizeN) �__name__�
__module__� __qualname__r r r r r r! r% r' r r r r r s r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�SelectPollerTestsc C � t |�S r )r �r r
r r r r 3 � zSelectPollerTests._makeOnec C �: | � t� �}|�d� |�d� | �t|j�ddg� d S �N� � )r r r r# �sorted� readables�r �pollerr r r r 6 �
z(SelectPollerTests.test_register_readablec C r/ r0 )r r r r# r3 � writablesr5 r r r r < r7 z(SelectPollerTests.test_register_writablec C s� | � t� �}|�d� |�d� |�d� |�d� |�d� |�d� |�d� | �t|j�dg� | �t|j�ddg� d S )Nr1 r2 � � �d ) r r r r r r# �listr4 r8 r5 r r r r B s
z*SelectPollerTests.test_unregister_readablec C s� | � t� �}|�d� |�d� |�d� |�d� |�d� |�d� |�d� | �t|j�ddg� | �t|j�dg� d S )Nr1 r2 r9 r; ) r r r r r r# r< r4 r8 r5 r r r r N s
z*SelectPollerTests.test_unregister_writablec C sr t dgdgd�d�}| �t� �}||_|�d� |�d� |�d� |�d�\}}| �|dg� | �|dg� d S )Nr1 r9 )r4 r8 ��resultr2 � )� DummySelectr r �_selectr r r r# )r rA r6 r4 r8 r r r �)test_poll_returns_readables_and_writablesZ s
�
z;SelectPollerTests.test_poll_returns_readables_and_writablesc C �N t tjd�}t� }| �|�}||_|�d� |�d� | �|j j
d d� d S )N��errorr1 r? r �EINTR encountered in poll) r@ �errno�EINTRr r rA r r r# �logger�data�r rA r
r6 r r r �test_poll_ignores_eintrf �
z)SelectPollerTests.test_poll_ignores_eintrc C sr t tjd�}t� }| �|�}||_|�d� |�d� | �|j j
d d� | �t |j �g � | �t |j
�g � d S )NrD r1 r? r zEBADF encountered in poll)r@ rG �EBADFr r rA r r r# rI rJ r< r4 r8 rK r r r �test_poll_ignores_ebadfo s
z)SelectPollerTests.test_poll_ignores_ebadfc C �B t tjd�}t� }| �|�}||_|�d� | �tj |j
d� d S )NrD r1 r? ) r@ rG �EPERMr r rA r r �selectrE r rK r r r � test_poll_uncaught_exceptionz �
z.SelectPollerTests.test_poll_uncaught_exceptionN) r( r) r* r r r r r rB rL rO rS r r r r r+ 1 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%S )&�KQueuePollerTestsc C r, r )r r- r r r r � r. zKQueuePollerTests._makeOnec C �^ t � }| �t� �}||_|�d� | �t|j�dg� | �t|j �d� | �
|j d d� d S )Nr1 r? r ) � DummyKQueuer r �_kqueuer r# r< r4 �len�registered_kevents�assertReadEventAdded�r �kqueuer6 r r r r � �
z(KQueuePollerTests.test_register_readablec C rV )Nr2 r? r ) rW r r rX r r# r< r8 rY rZ �assertWriteEventAddedr\ r r r r � r^ z(KQueuePollerTests.test_register_writablec C s� t � }| �t� �}||_|�d� |�d� |�d� |�d� |�d� | �t|j �dg� | �t|j
�g � | � |j d d� | �
|j d d� | �|j d d� | �|j d d� d S �Nr2 r9 r; r r? � � )rW r r rX r r r r# r< r8 r4 r_ rZ r[ �assertReadEventDeletedr\ r r r r � s
z*KQueuePollerTests.test_unregister_readablec C s� t � }| �t� �}||_|�d� |�d� |�d� |�d� |�d� | �t|j �g � | �t|j
�dg� | � |j d d� | �
|j d d� | �|j d d� | �|j d d� d S r` )rW r r rX r r r r# r< r8 r4 r_ rZ r[ �assertWriteEventDeletedr\ r r r r � s
z*KQueuePollerTests.test_unregister_writablec C s� t dtjfdtjfdtjfgd�}| �t� �}||_|�d� |�d� |�d� |� d�\}}| �
|ddg� | �
|dg� d S )Nr1 r2 r9 r= �� ) rW rR �KQ_FILTER_READ�KQ_FILTER_WRITEr r rX r r r r# )r r] r6 r4 r8 r r r rB � s
�
z;KQueuePollerTests.test_poll_returns_readables_and_writablesc C rC )N��raise_errno_pollr1 re r rF ) rW rG rH r r rX r r r# rI rJ �r r] r
r6 r r r rL � rM z)KQueuePollerTests.test_poll_ignores_eintrc C sb t tjd�}t� }| �|�}||_|�d� |�d� | �|j j
d d� | �|j j
d d� d S )N��raise_errno_registerr1 r2 r z6EBADF encountered in kqueue. Invalid file descriptor 6r? z6EBADF encountered in kqueue. Invalid file descriptor 7) rW rG rN r r rX r r r# rI rJ �r rX r
r6 r r r �1test_register_readable_and_writable_ignores_ebadf� s
��zCKQueuePollerTests.test_register_readable_and_writable_ignores_ebadfc C s6 t tjd�}t� }| �|�}||_| �t|jd� d S )Nrk � ) rW rG �ENOMEMr r rX r �OSErrorr rm r r r � test_register_uncaught_exception� s
z2KQueuePollerTests.test_register_uncaught_exceptionc C s@ t tjd�}t� }| �|�}||_|�d� | �t|j d� d S )Nrh r1 re )
rW rG �EINVALr r rX r r rq r rj r r r rS � s
z.KQueuePollerTests.test_poll_uncaught_exceptionc C s@ t � }t� }| �|�}||_|�� |j�� | �|jd � d S r )r r r rX r$ �close�assert_called_once_withr# �r � mock_kqueuer
r6 r r r �#test_before_daemonize_closes_kqueue� �
z5KQueuePollerTests.test_before_daemonize_closes_kqueuec C sh t � }| �|�}dg|_dg|_t� |_t� |_|�� | �t |j
t j �� |j�
d� |j�
d� d S )Nr? rb )r r r4 r8 r r r r& �
assertTrue�
isinstancerX rR r] �assert_called_with)r r
r6 r r r �$test_after_daemonize_restores_kqueue� s
z6KQueuePollerTests.test_after_daemonize_restores_kqueuec C s@ t � }t� }| �|�}||_|�� |j�� | �|jd � d S r )r r r rX rt ru r# rv r r r �test_close_closes_kqueue ry z*KQueuePollerTests.test_close_closes_kqueuec C � | � ||tjtj� d S r )� assertKeventrR rf � KQ_EV_ADD�r �kevent�fdr r r r[
� z&KQueuePollerTests.assertReadEventAddedc C r r )r� rR rg r� r� r r r r_ r� z'KQueuePollerTests.assertWriteEventAddedc C r r )r� rR rf � KQ_EV_DELETEr� r r r rc r� z(KQueuePollerTests.assertReadEventDeletedc C r r )r� rR rg r� r� r r r rd r� z)KQueuePollerTests.assertWriteEventDeletedc C s. | � |j|� | � |j|� | � |j|� d S r )r# �ident�filter�flags)r r� r� r� r� r r r r� s z KQueuePollerTests.assertKeventN)r( r) r* r r r r r rB rL rn rr rS rx r} r~ r[ r_ rc rd r� r r r r rU � s&
rU c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�PollerPollTestsc C r, r )r r- r r r r % r. zPollerPollTests._makeOnec C sB t � }| �t� �}||_|�d� |�d� | �|jddg� d S r0 )�DummySelectPollr r �_pollerr r# �registered_as_readable�r � select_pollr6 r r r r ( s
z&PollerPollTests.test_register_readablec C s6 t � }| �t� �}||_|�d� | �|jdg� d S )Nr1 )r� r r r� r r# �registered_as_writabler� r r r r 0 s
z&PollerPollTests.test_register_writablec C s� t dtjfdtjfdtjfdtjfgd�}| �t� �}||_|� d� |� d� |�
d� |� d� |� d�\}}| � |g d�� | � |dg� d S )Nr1 r2 r9 r: r= re )r1 r2 r: )
r� rR �POLLIN�POLLPRI�POLLOUT�POLLHUPr r r� r r r r# �r r� r6 r4 r8 r r r rB 7 s
�
z9PollerPollTests.test_poll_returns_readables_and_writablesc C rC )NrD r: re r rF ) r� rG rH r r r� r r r# rI rJ �r r� r
r6 r r r rL F rM z'PollerPollTests.test_poll_ignores_eintrc C rP )NrD r: re ) r� rG rN r r r� r r rR rE r r� r r r rS O rT z,PollerPollTests.test_poll_uncaught_exceptionc C sp t dtjfdtjfgd�}| �t� �}||_|�d� |�d� |�d�\}}| � |dg� | � |j
dg� d S )Nr1 r2 r= re ) r� rR �POLLNVALr� r r r� r r r# � unregisteredr� r r r �+test_poll_ignores_and_unregisters_closed_fdW s
�
z;PollerPollTests.test_poll_ignores_and_unregisters_closed_fdN)
r( r) r* r r r rB rL rS r� r r r r r� # s r� c @ s" e Zd ZdZddd�Zdd� ZdS )r@ z0
Fake implementation of select.select()
Nc C s. |pi }|� dg �| _|� dg �| _|| _d S )Nr4 r8 )�getr4 r8 rE �r r> rE r r r �__init__f s
zDummySelect.__init__c C s | j r t� | j ��| j| jg fS r )rE rR r4 r8 )r �r�w�x�timeoutr r r rR l s zDummySelect.select�NN)r( r) r* �__doc__r� rR r r r r r@ b s
r@ c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� ZdS ) r� z.
Fake implementation of select.poll()
Nc C s& |pg | _ || _g | _g | _g | _d S r )r> rE r� r� r� r� r r r r� u �
zDummySelectPoll.__init__c C sN |t jt jB t jB kr| j�|� d S |t jkr | j�|� d S td� |���)Nz+Registered a fd on unknown eventmask: '{0}')
rR r� r� r� r� �appendr� r� �
ValueError�format)r r� � eventmaskr r r �register| s
zDummySelectPoll.registerc C s | j �|� d S r )r� r� )r r� r r r �
unregister� � zDummySelectPoll.unregisterc C s | j r t� | j ��| jS r )rE rR r> )r r� r r r r � s zDummySelectPoll.pollr� )r( r) r* r� r� r� r� r r r r r r� q s
r� c @ sD e Zd ZdZddd�Zddd�Zdd� Zd d
� Zd d � Zd
d� Z dS )rW z0
Fake implementation of select.kqueue()
Nc C s& |pg | _ || _|| _g | _g | _d S r )r> �
errno_poll�errno_registerrZ �registered_flags)r r> ri rl r r r r� � r� zDummyKQueue.__init__c C sL |d u r| � |� | �| j� | �� S | �|� | �| j� | j�|� d S r )�assert_max_events_on_poll� raise_errorr� � build_result� assert_max_events_on_registerr� rZ �extend)r �kevents�
max_eventsr� r r r �control� s
zDummyKQueue.controlc C s |sd S t � }||_|�r )rq rG )r �err�exr r r r� � s zDummyKQueue.raise_errorc C s dd� | j D �S )Nc S s g | ] \}}t ||��qS r )�
FakeKEvent)�.0r� r� r r r �
<listcomp>� s z,DummyKQueue.build_result.<locals>.<listcomp>r= )r r r r r� � r� zDummyKQueue.build_resultc C s |t jks J dt j ��d S )Nz8`max_events` parameter of `kqueue.control() should be %d)r r� �r r� r r r r� � s
��z%DummyKQueue.assert_max_events_on_pollc C s |dksJ d��d S )Nr zD`max_events` parameter of `kqueue.control()` should be 0 on registerr r� r r r r� � s
�z)DummyKQueue.assert_max_events_on_register)NNNr )
r( r) r* r� r� r� r� r� r� r� r r r r rW � s
rW c @ s e Zd Zdd� ZdS )r� c C s || _ || _d S r )r� r� )r r� r� r r r r� � s
zFakeKEvent.__init__N)r( r) r* r� r r r r r� � s r� c C s t �tjt �S r )�unittest�
findTestCases�sys�modulesr( r r r r �
test_suite� r� r� �__main__)� defaultTest) r� r� rG rR �supervisor.tests.baser r r r r r r r �object� SkipTestCase�TestCaser r+ �KQueuePollerTestsBaserU �PollerPollTestsBaser� r@ r� rW r� r� r( �mainr r r r �<module> s8 !Q ? '�