Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-20 16:33:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

asyncio: New error handling API. Issue #20681.

Browse source
This commit is contained in:
Yury Selivanov 2014-02-18 18:02:19 -05:00
parent 6acc5e1330
commit 569efa2e4b
15 changed files with 491 additions and 99 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Lib/asyncio/test_utils.py
View file

@ -4,6 +4,7 @@
import contextlib
import io
import os
import re
import socket
import socketserver
import sys
@ -301,7 +302,7 @@ def close(self):
raise AssertionError("Time generator is not finished")
def add_reader(self, fd, callback, *args):
self.readers[fd] = events.Handle(callback, args)
self.readers[fd] = events.Handle(callback, args, self)
def remove_reader(self, fd):
self.remove_reader_count[fd] += 1
@ -320,7 +321,7 @@ def assert_reader(self, fd, callback, *args):
handle._args, args)
def add_writer(self, fd, callback, *args):
self.writers[fd] = events.Handle(callback, args)
self.writers[fd] = events.Handle(callback, args, self)
def remove_writer(self, fd):
self.remove_writer_count[fd] += 1
@ -362,3 +363,16 @@ def _write_to_self(self):
def MockCallback(**kwargs):
return unittest.mock.Mock(spec=['__call__'], **kwargs)
class MockPattern(str):
"""A regex based str with a fuzzy __eq__.
Use this helper with 'mock.assert_called_with', or anywhere
where a regexp comparison between strings is needed.
For instance:
mock_call.assert_called_with(MockPattern('spam.*ham'))
"""
def __eq__(self, other):
return bool(re.search(str(self), other, re.S))