Make TestAPIShutdown much like TestProcessShutdown

(this version should pass on Fedora)
This commit is contained in:
Dmitri Bogomolov 2020-11-04 17:35:17 +02:00
parent 66e8dbf621
commit 698123afec
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13

View File

@ -9,12 +9,14 @@ from .common import skip_python3
skip_python3()
import psutil
try: # nosec
from xmlrpclib import ServerProxy, ProtocolError
except ImportError:
from xmlrpc.client import ServerProxy, ProtocolError
from .test_process import TestProcessProto, TestProcessShutdown
from .test_process import TestProcessProto
class TestAPIProto(TestProcessProto):
@ -34,18 +36,16 @@ class TestAPIProto(TestProcessProto):
time.sleep(1)
class TestAPIShutdown(TestAPIProto, TestProcessShutdown):
class TestAPIShutdown(TestAPIProto):
"""Separate test case for API command 'shutdown'"""
def test_shutdown(self):
"""Shutdown the pybitmessage"""
self.assertEqual(self.api.shutdown(), 'done')
for _ in range(5):
if not self.process.is_running():
break
time.sleep(2)
else:
try:
self.process.wait(20)
except psutil.TimeoutExpired:
self.fail(
'%s has not stopped in 10 sec' % ' '.join(self._process_cmd))
'%s has not stopped in 20 sec' % ' '.join(self._process_cmd))
# TODO: uncovered API commands