ua1zbe
/
aprs-python
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix StopIteration breaking a test on py37+

This commit is contained in:
Rossen Georgiev 2019-10-08 20:02:19 +01:00
parent 2f0f74a06d
commit 512d97122c
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_IS.py
View File

@ -47,6 +47,9 @@ class TC_IS(unittest.TestCase):
f.write("something")
f.close()
class BreakBlocking(Exception):
pass
self.m.ReplayAll()
self.ais.sock = mox.MockAnything()
# part 1 - conn drop before setblocking
@ -74,7 +77,7 @@ class TC_IS(unittest.TestCase):
self.ais.sock.fileno().AndReturn(fdr)
self.ais.sock.recv(mox.IgnoreArg()).AndReturn(b"b\r\n"*3)
self.ais.sock.fileno().AndReturn(fdr)
self.ais.sock.recv(mox.IgnoreArg()).AndRaise(StopIteration)
self.ais.sock.recv(mox.IgnoreArg()).AndRaise(BreakBlocking)
mox.Replay(self.ais.sock)
next_method = '__next__' if sys.version_info[0] >= 3 else 'next'
@ -92,8 +95,9 @@ class TC_IS(unittest.TestCase):
for line in self.ais._socket_readlines():
self.assertEqual(line, b'a')
# part 5
for line in self.ais._socket_readlines(blocking=True):
self.assertEqual(line, b'b')
with self.assertRaises(BreakBlocking):
for line in self.ais._socket_readlines(blocking=True):
self.assertEqual(line, b'b')
mox.Verify(self.ais.sock)