1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
#
# This file is part of Deluge and is licensed under GNU General Public License 3.0, or later, with
# the additional special exception to link portions of this program with the OpenSSL library.
# See LICENSE for more details.
#
import os.path
import pytest
from twisted.internet import defer
from twisted.internet.error import CannotListenError
import deluge.component as component
from . import common
@pytest.mark.usefixtures('config_dir')
class DaemonBase:
def common_set_up(self):
self.listen_port = 58900
self.core = None
return component.start()
def terminate_core(self, *args):
if args[0] is not None:
if hasattr(args[0], 'getTraceback'):
print('terminate_core: Errback Exception: %s' % args[0].getTraceback())
if not self.core.killed:
d = self.core.kill()
return d
@defer.inlineCallbacks
def start_core(
self,
arg,
custom_script='',
logfile='',
print_stdout=True,
print_stderr=True,
timeout=5,
port_range=10,
extra_callbacks=None,
):
if logfile == '':
logfile = 'daemon_%s.log' % self.id()
# We are running py.test
if hasattr(pytest, 'config'):
if self.basetemp:
if not os.path.exists(self.basetemp):
os.makedirs(self.basetemp)
logfile = os.path.join(self.basetemp, logfile)
for dummy in range(port_range):
try:
d, self.core = common.start_core(
listen_port=self.listen_port,
logfile=logfile,
timeout=timeout,
timeout_msg='Timeout!',
custom_script=custom_script,
print_stdout=print_stdout,
print_stderr=print_stderr,
extra_callbacks=extra_callbacks,
config_directory=self.config_dir,
)
yield d
except CannotListenError as ex:
exception_error = ex
self.listen_port += 1
except (KeyboardInterrupt, SystemExit):
raise
else:
return
raise exception_error
|