Edit File: test_poller.cpython-36.pyc
3 �k�`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 � r�ejZneZG d d � d e�Ze � r�ejZneZG d d � d e�ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �Zdd� Zedk�r&ej dd� 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.6/test_poller.py�_makeOne s zBasePollerTests._makeOnec C s | j d �}| jt|jd � d S )N)r � assertRaises�NotImplementedError�register_readable)r �instr r r �test_register_readable s z&BasePollerTests.test_register_readablec C s | j d �}| jt|jd � d S )N)r r r �register_writable)r r r r r �test_register_writable s z&BasePollerTests.test_register_writablec C s | j d �}| jt|jd � d S )N)r r r �unregister_readable)r r r r r �test_unregister_readable s z(BasePollerTests.test_unregister_readablec C s | j d �}| jt|jd � d S )N)r r r �unregister_writable)r r r r r �test_unregister_writable! s z(BasePollerTests.test_unregister_writablec C s | j d �}| jt|jd � d S )N)r r r �poll)r r r r r � test_poll% s zBasePollerTests.test_pollc C s | j d �}| j|j� d � d S )N)r � assertEqual�before_daemonize)r r r r r �test_before_daemonize) s z%BasePollerTests.test_before_daemonizec C s | j d �}| j|j� d � d S )N)r r �after_daemonize)r r r r r �test_after_daemonize- s 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 s t |�S )N)r )r r r r r r 3 s zSelectPollerTests._makeOnec C s: | j t� �}|jd� |jd� | jt|j�ddg� d S )N� � )r r r r �sorted� readables)r �pollerr r r r 6 s z(SelectPollerTests.test_register_readablec C s: | j t� �}|jd� |jd� | jt|j�ddg� d S )Nr' r( )r r r r r) � writables)r r+ r r r r < s z(SelectPollerTests.test_register_writablec C s� | j t� �}|jd� |jd� |jd� |jd� |jd� |jd� |jd� | jt|j�dg� | jt|j�ddg� d S )Nr' r( � � �d ) r r r r r r �listr* r, )r r+ r r r r B s z*SelectPollerTests.test_unregister_readablec C s� | j t� �}|jd� |jd� |jd� |jd� |jd� |jd� |jd� | jt|j�ddg� | jt|j�dg� d S )Nr' r( r- r/ ) r r r r r r r0 r* r, )r r+ r r r r N s z*SelectPollerTests.test_unregister_writablec C sr t dgdgd�d�}| jt� �}||_|jd� |jd� |jd� |jd�\}}| j|dg� | j|dg� d S )Nr' r- )r* r, )�resultr( � )� DummySelectr r �_selectr r r r )r r4 r+ r* r, r r r �)test_poll_returns_readables_and_writablesZ s z;SelectPollerTests.test_poll_returns_readables_and_writablesc C sN t tjd�}t� }| j|�}||_|jd� |jd� | j|j j d d� d S )N)�errorr' r2 r zEINTR encountered in poll) r3 �errno�EINTRr r r4 r r r �logger�data)r r4 r r+ r r r �test_poll_ignores_eintrf s z)SelectPollerTests.test_poll_ignores_eintrc C sr t tjd�}t� }| j|�}||_|jd� |jd� | j|j j d d� | jt |j �g � | jt |j �g � d S )N)r6 r' r2 r zEBADF encountered in poll)r3 r7 �EBADFr r r4 r r r r9 r: r0 r* r, )r r4 r r+ r r r �test_poll_ignores_ebadfo s z)SelectPollerTests.test_poll_ignores_ebadfc C sB t tjd�}t� }| j|�}||_|jd� | jtj |j d� d S )N)r6 r' r2 ) r3 r7 ZEPERMr r r4 r r �selectr6 r )r r4 r r+ r r r � test_poll_uncaught_exceptionz s z.SelectPollerTests.test_poll_uncaught_exceptionN) r# r$ r% r r r r r r5 r; r= r? 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 s t |�S )N)r )r r r r r r � s zKQueuePollerTests._makeOnec C s^ t � }| jt� �}||_|jd� | jt|j�dg� | jt|j �d� | j |j d d� d S )Nr' r2 r ) � DummyKQueuer r �_kqueuer r r0 r* �len�registered_kevents�assertReadEventAdded)r �kqueuer+ r r r r � s z(KQueuePollerTests.test_register_readablec C s^ t � }| jt� �}||_|jd� | jt|j�dg� | jt|j �d� | j |j d d� d S )Nr( r2 r ) rA r r rB r r r0 r, rC rD �assertWriteEventAdded)r rF r+ r r r r � s z(KQueuePollerTests.test_register_writablec C s� t � }| jt� �}||_|jd� |jd� |jd� |jd� |jd� | jt|j �dg� | jt|j �g � | j |j d d� | j |j d d� | j|j d d� | j|j d d� d S )Nr( r- r/ r r2 � � )rA r r rB r r r r r0 r, r* rG rD rE �assertReadEventDeleted)r rF r+ r r r r � s z*KQueuePollerTests.test_unregister_readablec C s� t � }| jt� �}||_|jd� |jd� |jd� |jd� |jd� | jt|j �g � | jt|j �dg� | j |j d d� | j |j d d� | j|j d d� | j|j d d� d S )Nr( r- r/ r r2 rH rI )rA r r rB r r r r r0 r, r* rG rD rE �assertWriteEventDeleted)r rF r+ r r r r � s z*KQueuePollerTests.test_unregister_writablec C s� t dtjfdtjfdtjfgd�}| jt� �}||_|jd� |jd� |jd� |j d�\}}| j |ddg� | j |dg� d S )Nr' r( r- )r1 i� ) rA r> �KQ_FILTER_READ�KQ_FILTER_WRITEr r rB r r r r )r rF r+ r* r, r r r r5 � s z;KQueuePollerTests.test_poll_returns_readables_and_writablesc C sN t tjd�}t� }| j|�}||_|jd� |jd� | j|j j d d� d S )N)�raise_errno_pollr' i� r zEINTR encountered in poll) rA r7 r8 r r rB r r r r9 r: )r rF r r+ r r r r; � s z)KQueuePollerTests.test_poll_ignores_eintrc C sb t tjd�}t� }| j|�}||_|jd� |jd� | j|j j d d� | j|j j d d� d S )N)�raise_errno_registerr' r( r z6EBADF encountered in kqueue. Invalid file descriptor 6r2 z6EBADF encountered in kqueue. Invalid file descriptor 7) rA r7 r< r r rB r r r r9 r: )r rB r r+ r r r �1test_register_readable_and_writable_ignores_ebadf� s zCKQueuePollerTests.test_register_readable_and_writable_ignores_ebadfc C s6 t tjd�}t� }| j|�}||_| jt|jd� d S )N)rO � ) rA r7 ZENOMEMr r rB r �OSErrorr )r rB r r+ r r r � test_register_uncaught_exception� s z2KQueuePollerTests.test_register_uncaught_exceptionc C s@ t tjd�}t� }| j|�}||_|jd� | jt|j d� d S )N)rN r' i� ) rA r7 ZEINVALr r rB r r rR r )r rF r r+ r r r r? � s z.KQueuePollerTests.test_poll_uncaught_exceptionc C s@ t � }t� }| j|�}||_|j� |jj� | j|jd � d S )N)r r r rB r �close�assert_called_once_withr )r � mock_kqueuer r+ r r r �#test_before_daemonize_closes_kqueue� s z5KQueuePollerTests.test_before_daemonize_closes_kqueuec C sh t � }| j|�}dg|_dg|_t� |_t� |_|j� | jt |j t j �� |jj d� |jj d� d S )Nr2 rI )r r r* r, r r r r! Z assertTrue� isinstancerB r> rF Zassert_called_with)r r r+ r r r �$test_after_daemonize_restores_kqueue� s z6KQueuePollerTests.test_after_daemonize_restores_kqueuec C s@ t � }t� }| j|�}||_|j� |jj� | j|jd � d S )N)r r r rB rT rU r )r rV r r+ r r r �test_close_closes_kqueue s z*KQueuePollerTests.test_close_closes_kqueuec C s | j ||tjtj� d S )N)� assertKeventr> rL � KQ_EV_ADD)r �kevent�fdr r r rE s z&KQueuePollerTests.assertReadEventAddedc C s | j ||tjtj� d S )N)r[ r> rM r\ )r r] r^ r r r rG s z'KQueuePollerTests.assertWriteEventAddedc C s | j ||tjtj� d S )N)r[ r> rL � KQ_EV_DELETE)r r] r^ r r r rJ s z(KQueuePollerTests.assertReadEventDeletedc C s | j ||tjtj� d S )N)r[ r> rM r_ )r r] r^ r r r rK s z)KQueuePollerTests.assertWriteEventDeletedc C s. | j |j|� | j |j|� | j |j|� d S )N)r �ident�filter�flags)r r] r` ra rb r r r r[ s z KQueuePollerTests.assertKeventN)r# r$ r% r r r r r r5 r; rP rS r? rW rY rZ rE rG rJ rK r[ r r r r r@ � s$ r@ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�PollerPollTestsc C s t |�S )N)r )r r r r r r % s zPollerPollTests._makeOnec C sB t � }| jt� �}||_|jd� |jd� | j|jddg� d S )Nr' r( )�DummySelectPollr r �_pollerr r �registered_as_readable)r � select_pollr+ r r r r ( s z&PollerPollTests.test_register_readablec C s6 t � }| jt� �}||_|jd� | j|jdg� d S )Nr' )rd r r re r r �registered_as_writable)r rg r+ r r r r 0 s z&PollerPollTests.test_register_writablec C s� t dtjfdtjfdtjfdtjfgd�}| jt� �}||_|j d� |j d� |j d� |j d� |j d�\}}| j |dddg� | j |dg� d S )Nr' r( r- r. )r1 i� ) rd r> �POLLIN�POLLPRI�POLLOUT�POLLHUPr r re r r r r )r rg r+ r* r, r r r r5 7 s z9PollerPollTests.test_poll_returns_readables_and_writablesc C sN t tjd�}t� }| j|�}||_|jd� |jd� | j|j j d d� d S )N)r6 r. i� r zEINTR encountered in poll) rd r7 r8 r r re r r r r9 r: )r rg r r+ r r r r; F s z'PollerPollTests.test_poll_ignores_eintrc C sB t tjd�}t� }| j|�}||_|jd� | jtj |j d� d S )N)r6 r. i� ) rd r7 r< r r re r r r> r6 r )r rg r r+ r r r r? O s z,PollerPollTests.test_poll_uncaught_exceptionc C sp t dtjfdtjfgd�}| jt� �}||_|jd� |jd� |jd�\}}| j |dg� | j |j dg� d S )Nr' r( )r1 i� ) rd r> ZPOLLNVALrj r r re r r r � unregistered)r rg r+ r* r, 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 r5 r; r? rn r r r r rc # s rc c @ s" e Zd ZdZddd�Zdd� ZdS )r3 z0 Fake implementation of select.select() Nc C s. |pi }|j dg �| _|j dg �| _|| _d S )Nr* r, )�getr* r, r6 )r r1 r6 r r r �__init__f s zDummySelect.__init__c C s | j rtj | j ��| j| jg fS )N)r6 r> r* r, )r �r�w�x�timeoutr r r r> l s zDummySelect.select)NN)r# r$ r% �__doc__rp r> r r r r r3 b s r3 c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d � ZdS ) rd z. Fake implementation of select.poll() Nc C s& |pg | _ || _g | _g | _g | _d S )N)r1 r6 rf rh rm )r r1 r6 r r r rp u s zDummySelectPoll.__init__c C sN |t jt jB t jB kr$| jj|� n&|t jkr<| jj|� ntdj |���d S )Nz+Registered a fd on unknown eventmask: '{0}') r> ri rj rl rf �appendrk rh � ValueError�format)r r^ Z eventmaskr r r �register| s zDummySelectPoll.registerc C s | j j|� d S )N)rm rv )r r^ r r r � unregister� s zDummySelectPoll.unregisterc C s | j rtj | j ��| jS )N)r6 r> r1 )r rt r r r r � s zDummySelectPoll.poll)NN)r# r$ r% ru rp ry rz r r r r r rd q s rd c @ sD e Zd ZdZddd�Zddd�Zdd� Zd d � Zd d � Zd d� Z dS )rA z0 Fake implementation of select.kqueue() Nc C s& |pg | _ || _|| _g | _g | _d S )N)r1 � errno_poll�errno_registerrD Zregistered_flags)r r1 rN rO r r r rp � s zDummyKQueue.__init__c C sL |d kr&| j |� | j| j� | j� S | j|� | j| j� | jj|� d S )N)�assert_max_events_on_poll� raise_errorr{ � build_result� assert_max_events_on_registerr| rD �extend)r Zkevents� max_eventsrt r r r �control� s zDummyKQueue.controlc C s |sd S t � }||_|�d S )N)rR r7 )r �errZexr r r r~ � s zDummyKQueue.raise_errorc C s dd� | j D �S )Nc S s g | ]\}}t ||��qS r )� FakeKEvent)�.0r` ra r r r � <listcomp>� s z,DummyKQueue.build_result.<locals>.<listcomp>)r1 )r r r r r � s zDummyKQueue.build_resultc C s |t jkstdt j ��d S )Nz8`max_events` parameter of `kqueue.control() should be %d)r r� �AssertionError)r r� r r r r} � s z%DummyKQueue.assert_max_events_on_pollc C s |dkst d��d S )Nr zD`max_events` parameter of `kqueue.control()` should be 0 on register)r� )r r� r r r r� � s z)DummyKQueue.assert_max_events_on_register)NNN)N) r# r$ r% ru rp r� r~ r r} r� r r r r rA � s rA c @ s e Zd Zdd� ZdS )r� c C s || _ || _d S )N)r` ra )r r` ra r r r rp � s zFakeKEvent.__init__N)r# r$ r% rp 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� r7 r> Zsupervisor.tests.baser r r r r r r r �objectZ SkipTestCaseZTestCaser r& ZKQueuePollerTestsBaser@ ZPollerPollTestsBaserc r3 rd rA r� r� r# �mainr r r r �<module> s4 !Q ? '
Back to File Manager