/usr/local/lib/python3.6/site-packages/asynctest/__pycache__
Edit: /usr/local/lib/python3.6/site-packages/asynctest/__pycache__/selector.cpython-36.pyc (11224B)
3
8%Eg- @ sz d Z ddlZy ddlZW n ek
r6 ddljZY nX ddlZy ddlZW n ek
rd dZY nX ddlmZ ddlm Z G dd de
Z dd Z d
d Z
G d d
d
ejZG dd deZerG dd deZdd Zdd Zdd ZG dd dejZdd Zeed r d d Zn eejd r4d d Znd!d Zd"d# Zd$d% Z d&d' Z d(e j d)< e e e j _ e e e j _!dS )*ay
Module ``selector``
-------------------
Mock of :mod:`selectors` and compatible objects performing asynchronous IO.
This module provides classes to mock objects performing IO (files, sockets,
etc). These mocks are compatible with :class:`~asynctest.TestSelector`, which
can simulate the behavior of a selector on the mock objects, or forward actual
work to a real selector.
N )mock)_fail_onc s, e Zd ZdZdZ fddZdd Z ZS )FileDescriptora
A subclass of int which allows to identify the virtual file-descriptor of a
:class:`~asynctest.FileMock`.
If :class:`~asynctest.FileDescriptor()` without argument, its value will be
the value of :data:`~FileDescriptor.next_fd`.
When an object is created, :data:`~FileDescriptor.next_fd` is set to the
highest value for a :class:`~asynctest.FileDescriptor` object + 1.
r c sL | r | r t j| tj}nt j| f||}ttjd |d t_|S )Nr )super__new__r next_fdmax)clsargskwargss) __class__ s
r c C s2 yt | tpt | j tS tk
r, dS X dS )zh
Return ``True`` if the ``obj`` or ``obj.fileno()`` is
a :class:`asynctest.FileDescriptor`.
FN)r r r AttributeError)objr r r
isfilemockQ s
r" c s( e Zd ZdZ fddZdd Z ZS )FileMockz
Mock a file-like object.
A FileMock is an intelligent mock which can work with TestSelector to
simulate IO events during tests.
.. method:: fileno()
Return a :class:`~asynctest.FileDescriptor` object.
c s t j|| t | j_d S )N)r __init__r r return_value)r r r )r r r r$ i s zFileMock.__init__c O s t jf |S )N)r Mock)r r r r r r _get_child_mockn s zFileMock._get_child_mock)r r r r r$ r' r r r )r r r# ^ s
r# c s0 e Zd ZdZdejddddf fdd Z ZS )
SocketMockz?
Mock a socket.
See :class:`~asynctest.FileMock`.
Nc s( t jtjf||||||d| d S )N) side_effectr% wrapsnamespec_setparent)r r$ socket)r r) r% r* r+ r, r- r )r r r r$ y s zSocketMock.__init__)r r r r r DEFAULTr$ r r r )r r r( s s r( c @ s( e Zd ZdZdejddddfddZdS )
SSLSocketMockz
Mock a socket wrapped by the :mod:`ssl` module.
See :class:`~asynctest.FileMock`.
.. versionadded:: 0.5
Nc K s( t j| tjf||||||d| d S )N)r) r% r* r+ r, r- )r# r$ ssl SSLSocket)r r) r% r* r+ r, r- r r r r r$ s zSSLSocketMock.__init__)r r r r r r/ r$ r r r r r0 s r0 c C s4 |j }|j| }||jkr0|j|j| |fg d S )N) _selector_fileobj_lookup
_fd_to_key_process_events)r loopeventselectorr r r r _set_event_ready s
r: c C s |j t| |tj dS )a
Schedule callbacks registered on ``loop`` as if the selector notified that
data is ready to be read on ``fileobj``.
:param fileobj: file object or :class:`~asynctest.FileMock` on which the
event is mocked.
:param loop: :class:`asyncio.SelectorEventLoop` watching for events on
``fileobj``.
::
mock = asynctest.SocketMock()
mock.recv.return_value = b"Data"
def read_ready(sock):
print("received:", sock.recv(1024))
loop.add_reader(mock, read_ready, mock)
set_read_ready(mock, loop)
loop.run_forever() # prints received: b"Data"
.. versionadded:: 0.4
N)call_soon_threadsafer: selectors
EVENT_READ)r r7 r r r set_read_ready s r> c C s |j t| |tj dS )a^
Schedule callbacks registered on ``loop`` as if the selector notified that
data can be written to ``fileobj``.
:param fileobj: file object or :class:`~asynctest.FileMock` on which th
event is mocked.
:param loop: :class:`asyncio.SelectorEventLoop` watching for events on
``fileobj``.
.. versionadded:: 0.4
N)r; r: r< EVENT_WRITE)r r7 r r r set_write_ready s r@ c sl e Zd ZdZd fdd Z fddZd fdd Z fd d
Zd fd d Zdd
dZ fddZ
Z S ) TestSelectora
A selector which supports IOMock objects.
It can wrap an actual implementation of a selector, so the selector will
work both with mocks and real file-like objects.
A common use case is to patch the selector loop::
loop._selector = asynctest.TestSelector(loop._selector)
:param selector: optional, if provided, this selector will be used to work
with real file-like objects.
Nc s t j || _d S )N)r r$ r3 )r r9 )r r r r$ s
zTestSelector.__init__c s t |rt|S t j|S )N)r" r r r4 )r r )r r r r4 s z TestSelector._fileobj_lookupc sH t |s| jdkr$t j|||}n | jj|||}|rD|| j|j< |S )a
Register a file object or a :class:`~asynctest.FileMock`.
If a real selector object has been supplied to the
:class:`~asynctest.TestSelector` object and ``fileobj`` is not
a :class:`~asynctest.FileMock` or a :class:`~asynctest.FileDescriptor`
returned by :meth:`FileMock.fileno()`, the object will be registered to
the real selector.
See :meth:`selectors.BaseSelector.register`.
N)r" r3 r registerr5 r )r r eventsdatakey)r r r rB s zTestSelector.registerc sJ t |s| jdkr t j|}n&| jj|}|rF|j| jkrF| j|j= |S )z
Unregister a file object or a :class:`~asynctest.FileMock`.
See :meth:`selectors.BaseSelector.unregister`.
N)r" r3 r
unregisterr r5 )r r rE )r r r rF s
zTestSelector.unregisterc sd t |s| jdkr$t j|||}n<| j|}|| jkr@| j|= | jj|||}|r`|| j|j< |S )z
Shortcut when calling :meth:`TestSelector.unregister` then
:meth:`TestSelector.register` to update the registration of a an object
to the selector.
See :meth:`selectors.BaseSelector.modify`.
N)r" r3 r modifyr4 r5 r )r r rC rD rE r )r r r rG s
zTestSelector.modifyc C s | j dkrg S | j j|S )z
Perform the selection.
This method is a no-op if no actual selector has been supplied.
See :meth:`selectors.BaseSelector.select`.
N)r3 select)r timeoutr r r rH s
zTestSelector.selectc s" | j dk r| j j t j dS )z
Close the selector.
Close the actual selector if supplied, unregister all mocks.
See :meth:`selectors.BaseSelector.close`.
N)r3 closer )r )r r r rJ - s
zTestSelector.close)N)N)N)N) r r r r r$ r4 rB rF rG rH rJ r r r )r r rA s
rA c C sX g }x | j j D ]}|j| qW | jd k rPx | jj j D ]}|j| q>W t|S )N)get_mapvaluesappendr3 set)r9 Zwatched_eventsr8 r r r get_registered_events; s
rO format_helpersc C s t jj| j| jd S )N)asynciorP _format_callback _callback_args)handler r r rR J s
rR _format_args_and_kwargsc C s t jj| j| jd S )N)rQ rC rR rS rT )rU r r r rR O s c C s t jj| j| jS )N)rQ rC rR rS rT )rU r r r rR T s c C s\ g }| j tj@ r.|jdj| jt| jd | j tj@ rX|jdj| jt| jd |S )Nzadd_reader({}, {})r zadd_writer({}, {})r ) rC r< r= rM r r rR rD r? )r8 callbacksr r r
_format_eventX s rX c C s t | jj| _d S )N)rO r7 r3 _active_selector_callbacks)caser r r -fail_on_before_test_active_selector_callbacksf s r[ c C sZ | j }t| jj}dg}x tt|| D ]}|j| q(W t|dkrV| jdj | d S )Nz6some events watched during the tests were not removed:r z
- )
rY rO r7 r3 maprX extendlenZfailjoin)rZ Zignored_eventsZ
active_eventsoutputcr r r !fail_on_active_selector_callbacksk s rb Factive_selector_callbacks)"r rQ r< ImportErrorZasyncio.selectorsr. r1 r r intr r r" r& r# r( r0 r: r> r@ _BaseSelectorImplrA rO hasattrrR rC rX r[ rb ZDEFAULTS staticmethodrc Z%before_test_active_selector_callbacksr r r r
sF
!q