diff options
| author | Victor Stinner <vstinner@redhat.com> | 2018-06-13 12:05:15 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-06-13 12:05:15 +0200 |
| commit | 142e3c08a40c75b5788474b0defe7d5c0671f675 (patch) | |
| tree | 949c50cbff8fed695ef04e05d28485f133ca9580 /Lib/asyncio/test_utils.py | |
| parent | 961332dfd3dbff4c63ed4d9f8b5de937aa63475b (diff) | |
| download | cpython-git-142e3c08a40c75b5788474b0defe7d5c0671f675.tar.gz | |
[3.6] bpo-32356: idempotent pause_/resume_reading (GH-4914) (GH-7629)
Backport note: don't add new is_reading() method from master to 3.6.
(cherry picked from commit d757aaf9dd767d13205bf9917e520ebf43e7f6e5)
Diffstat (limited to 'Lib/asyncio/test_utils.py')
| -rw-r--r-- | Lib/asyncio/test_utils.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/Lib/asyncio/test_utils.py b/Lib/asyncio/test_utils.py index 8b8c22a747..f41720428c 100644 --- a/Lib/asyncio/test_utils.py +++ b/Lib/asyncio/test_utils.py @@ -335,12 +335,19 @@ class TestLoop(base_events.BaseEventLoop): return False def assert_reader(self, fd, callback, *args): - assert fd in self.readers, 'fd {} is not registered'.format(fd) + if fd not in self.readers: + raise AssertionError(f'fd {fd} is not registered') handle = self.readers[fd] - assert handle._callback == callback, '{!r} != {!r}'.format( - handle._callback, callback) - assert handle._args == args, '{!r} != {!r}'.format( - handle._args, args) + if handle._callback != callback: + raise AssertionError( + f'unexpected callback: {handle._callback} != {callback}') + if handle._args != args: + raise AssertionError( + f'unexpected callback args: {handle._args} != {args}') + + def assert_no_reader(self, fd): + if fd in self.readers: + raise AssertionError(f'fd {fd} is registered') def _add_writer(self, fd, callback, *args): self.writers[fd] = events.Handle(callback, args, self) |
