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

Add test for systemd socket activation

Browse Source
This commit is contained in:
Kovid Goyal 2015-05-27 09:21:05 +05:30
parent 700e1996c8
commit 5f4e8230a9
1 changed files with 36 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,18 +6,20 @@ 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 import httplib, ssl, os, subprocess, time, socket
from unittest import skipIf from unittest import skipIf, skipUnless
from calibre.constants import islinux
try: try:
from calibre.utils.certgen import create_server_cert from calibre.utils.certgen import create_server_cert
except ImportError: except ImportError:
create_server_cert = None create_server_cert = None
from calibre.srv.tests.base import BaseTest, TestServer from calibre.srv.tests.base import BaseTest, TestServer
from calibre.ptempfile import TemporaryDirectory from calibre.ptempfile import TemporaryDirectory
HOST = 'localhost.test' SYSTEMD = '/usr/lib/systemd/systemd-activate'
class LoopTest(BaseTest): class LoopTest(BaseTest):
@ -39,3 +41,34 @@ class LoopTest(BaseTest):
cert = conn.sock.getpeercert() cert = conn.sock.getpeercert()
subject = dict(x[0] for x in cert['subject']) subject = dict(x[0] for x in cert['subject'])
self.ae(subject['commonName'], address) self.ae(subject['commonName'], address)
@skipUnless(islinux and os.access(SYSTEMD, os.X_OK), 'systemd-activate not available')
def test_systemd(self):
'Test systemd socket activation'
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0)
s.bind(('localhost', 0))
port = s.getsockname()[1]
f = os.path.abspath(__file__).rstrip('c')
s.close()
p = subprocess.Popen([SYSTEMD, '-l', '127.0.0.1:%d' % port, 'calibre-debug', f], stdout=open(os.devnull, 'wb'), stderr=subprocess.STDOUT)
try:
conn = httplib.HTTPConnection('localhost', port, strict=True, timeout=1)
conn.request('GET', '/test', 'body')
r = conn.getresponse()
self.ae(r.status, httplib.OK)
self.ae(r.read(), b'testbody')
conn.request('GET', '/quit')
st = time.time()
while p.poll() is None and time.time() - st < 2:
time.sleep(0.01)
finally:
if p.poll() is None:
p.kill()
if __name__ == '__main__':
def handler(data):
if data.path and data.path[0] == 'quit':
raise SystemExit(0)
return (data.path[0] + data.read()) if data.path else ''
with TestServer(handler, allow_socket_preallocation=True) as server:
server.join()