Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-08 02:34:06 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add test for shutdown with hung worker

Browse Source
This commit is contained in:
Kovid Goyal 2015-05-29 18:45:37 +05:30
parent a485b61fe1
commit 00df6549c0
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/calibre/srv/tests/loop.py
View File

@ -6,7 +6,7 @@ from __future__ import (unicode_literals, division, absolute_import,
__license__ = 'GPL v3' __license__ = 'GPL v3'
__copyright__ = '2015, Kovid Goyal <kovid at kovidgoyal.net>' __copyright__ = '2015, Kovid Goyal <kovid at kovidgoyal.net>'
import httplib, ssl, os, socket import httplib, ssl, os, socket, time
from unittest import skipIf from unittest import skipIf
try: try:
@ -28,6 +28,18 @@ class LoopTest(BaseTest):
server.loop.stop() server.loop.stop()
server.join() server.join()
self.ae(0, sum(int(w.is_alive()) for w in server.loop.pool.workers)) self.ae(0, sum(int(w.is_alive()) for w in server.loop.pool.workers))
# Test shutdown with hung worker
with TestServer(lambda data:time.sleep(1000), worker_count=3, shutdown_timeout=0.01, timeout=0.01) as server:
pool = server.loop.pool
self.ae(3, sum(int(w.is_alive()) for w in pool.workers))
conn = server.connect()
conn.request('GET', '/')
with self.assertRaises(socket.timeout):
conn.getresponse()
self.ae(pool.busy, 1)
server.loop.stop()
server.join()
self.ae(1, sum(int(w.is_alive()) for w in pool.workers))
def test_ring_buffer(self): def test_ring_buffer(self):
class FakeSocket(object): class FakeSocket(object):