upstream/certbot
1
0
Fork 0
mirror of https://github.com/certbot/certbot.git synced 2026-06-03 13:59:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Avoid race conditions in acme.standalone_test.

Browse source
This commit is contained in:
Jakub Warmuz 2015-10-18 12:22:54 +00:00
parent c0b6305f1c
commit 4090085b17
No known key found for this signature in database
GPG key ID: 2A7BAD3A489B52EA
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
acme/acme/standalone_test.py
View file

@ -35,14 +35,19 @@ class ACMEServerMixinTest(unittest.TestCase):
def setUp(self):
from acme.standalone import ACMEServerMixin
class _MockHandler(socketserver.BaseRequestHandler):
def handle(self):
self.request.sendall(b"DONE")
class _MockServer(socketserver.TCPServer, ACMEServerMixin):
def __init__(self, *args, **kwargs):
socketserver.TCPServer.__init__(self, *args, **kwargs)
ACMEServerMixin.__init__(self)
self.server = _MockServer(("", 0), socketserver.BaseRequestHandler)
self.server = _MockServer(("", 0), _MockHandler)
def _busy_wait(self): # pragma: no cover
# This function is used to avoid race coditions in tests, but
# This function is used to avoid race conditions in tests, but
# not all of the functionality is always used, hence "no
# cover"
while True:
@ -53,6 +58,7 @@ class ACMEServerMixinTest(unittest.TestCase):
except socket.error:
pass
else:
sock.recv(4) # wait until handle_request is actually called
break
finally:
sock.close()