diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2021-12-20 10:23:05 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-12-20 10:23:05 (GMT) |
commit | 6ca78affc8023bc5023189d64d8050857662042a (patch) | |
tree | e6cae1875586d1c722005311492b4126707d9a2e /Lib/test/test_asyncio/test_sendfile.py | |
parent | 7c5c3f7254d78babcaf7a2ec187fd6ec53b8403c (diff) | |
download | cpython-6ca78affc8023bc5023189d64d8050857662042a.zip cpython-6ca78affc8023bc5023189d64d8050857662042a.tar.gz cpython-6ca78affc8023bc5023189d64d8050857662042a.tar.bz2 |
bpo-23819: Get rid of assert statements in test_asyncio (GH-30212)
To keep checks even if run tests with optimized Python.
Either use special assertion methods like assertEqual() or
raise an AssertionError explicitly.
Diffstat (limited to 'Lib/test/test_asyncio/test_sendfile.py')
-rw-r--r-- | Lib/test/test_asyncio/test_sendfile.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/Lib/test/test_asyncio/test_sendfile.py b/Lib/test/test_asyncio/test_sendfile.py index 01c6986..0a5466a 100644 --- a/Lib/test/test_asyncio/test_sendfile.py +++ b/Lib/test/test_asyncio/test_sendfile.py @@ -36,25 +36,29 @@ class MySendfileProto(asyncio.Protocol): self.data = bytearray() self.close_after = close_after + def _assert_state(self, *expected): + if self.state not in expected: + raise AssertionError(f'state: {self.state!r}, expected: {expected!r}') + def connection_made(self, transport): self.transport = transport - assert self.state == 'INITIAL', self.state + self._assert_state('INITIAL') self.state = 'CONNECTED' if self.connected: self.connected.set_result(None) def eof_received(self): - assert self.state == 'CONNECTED', self.state + self._assert_state('CONNECTED') self.state = 'EOF' def connection_lost(self, exc): - assert self.state in ('CONNECTED', 'EOF'), self.state + self._assert_state('CONNECTED', 'EOF') self.state = 'CLOSED' if self.done: self.done.set_result(None) def data_received(self, data): - assert self.state == 'CONNECTED', self.state + self._assert_state('CONNECTED') self.nbytes += len(data) self.data.extend(data) super().data_received(data) |