Implement a service connection handler counter, use it in Pathod test suite

Lots of failures, but that's a good thing.
This commit is contained in:
Aldo Cortesi 2016-06-03 11:47:07 +12:00
parent c6d15904f3
commit c5076f5e01
3 changed files with 44 additions and 7 deletions

View File

@ -833,6 +833,25 @@ class BaseHandler(_Connection):
return b""
class Counter:
def __init__(self):
self._count = 0
self._lock = threading.Lock()
@property
def count(self):
with self._lock:
return self._count
def __enter__(self):
with self._lock:
self._count += 1
def __exit__(self, *args):
with self._lock:
self._count -= 1
class TCPServer(object):
request_queue_size = 20
@ -845,8 +864,10 @@ class TCPServer(object):
self.socket.bind(self.address())
self.address = Address.wrap(self.socket.getsockname())
self.socket.listen(self.request_queue_size)
self.counter = Counter()
def connection_thread(self, connection, client_address):
with self.counter:
client_address = Address(client_address)
try:
self.handle_client_connection(connection, client_address)

View File

@ -7,6 +7,10 @@ from six.moves import queue
from . import pathod
class TimeoutError(Exception):
pass
class Daemon:
IFACE = "127.0.0.1"
@ -40,6 +44,17 @@ class Daemon:
def text_log(self):
return self.logfp.getvalue()
def wait_for_silence(self, timeout=5):
start = time.time()
while 1:
if time.time() - start >= timeout:
raise TimeoutError(
"%s service threads still alive" %
self.thread.server.counter.count
)
if self.thread.server.counter.count == 0:
return
def expect_log(self, n, timeout=5):
l = []
start = time.time()

View File

@ -64,6 +64,7 @@ class DaemonTests(object):
shutil.rmtree(cls.confdir)
def teardown(self):
self.d.wait_for_silence()
if not (self.noweb or self.noapi):
self.d.clear_log()