val/irctest
1
0
Fork 0
mirror of https://github.com/progval/irctest.git synced 2025-04-05 14:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Better handling of connections closed by clients.

Browse Source
This commit is contained in:
Valentin Lorentz
2021-02-20 10:43:00 +01:00
parent c4a9592156
commit fe694487c7
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
irctest/cases.py
View File

@ -113,7 +113,11 @@ class BaseClientTestCase(_IrcTestCase):
self._setUpServer() self._setUpServer()
def tearDown(self): def tearDown(self):
if self.conn: if self.conn:
self.conn.sendall(b'QUIT :end of test.') try:
self.conn.sendall(b'QUIT :end of test.')
except BrokenPipeError:
# client already disconnected
pass
self.controller.kill() self.controller.kill()
if self.conn: if self.conn:
self.conn_file.close() self.conn_file.close()

2
irctest/client_tests/test_tls.py
View File

@ -139,5 +139,5 @@ class TlsTestCase(cases.BaseClientTestCase):
tls_config=tls_config, tls_config=tls_config,
) )
self.acceptClient(tls_cert=BAD_CERT, tls_key=BAD_KEY) self.acceptClient(tls_cert=BAD_CERT, tls_key=BAD_KEY)
with self.assertRaises(ConnectionClosed): with self.assertRaises((ConnectionClosed, ConnectionResetError)):
m = self.getMessage() m = self.getMessage()