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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
import sys
import subprocess
import time
class BoothRunner:
default_config_file = '/etc/booth/booth.conf'
default_lock_file = '/var/run/booth.pid'
def __init__(self, boothd_path, mode, args):
self.boothd_path = boothd_path
self.args = (mode, )
self.final_args = tuple(args) # will be appended to self.args
self.mode = mode
self.config_file = None
self.lock_file = None
def set_config_file_arg(self):
self.args += ('-c', self.config_file)
def set_config_file(self, config_file):
self.config_file = config_file
self.set_config_file_arg()
def set_lock_file(self, lock_file):
self.lock_file = lock_file
self.args += ('-l', self.lock_file)
def set_debug(self):
self.args += ('-D', )
def set_foreground(self):
self.args += ('-S', )
def all_args(self):
return (self.boothd_path, ) + self.args + self.final_args
def show_output(self, stdout, stderr):
if stdout:
print("STDOUT:")
print("------")
print(stdout.rstrip('\n'))
if stderr:
print("STDERR: (N.B. crm_ticket failures indicate daemon started correctly)")
print("------")
print(stderr.rstrip('\n'))
print("-" * 70)
def subproc_completed_within(self, p, timeout):
start = time.time()
wait = 0.1
while True:
if p.poll() is not None:
return True
elapsed = time.time() - start
if elapsed + wait > timeout:
wait = timeout - elapsed
print("Waiting on %d for %.1fs ..." % (p.pid, wait))
time.sleep(wait)
elapsed = time.time() - start
if elapsed >= timeout:
return False
wait *= 2
def lock_file_used(self):
return self.lock_file or self.default_lock_file
def config_file_used(self):
return self.config_file or self.default_config_file
def config_text_used(self):
config_file = self.config_file_used()
try:
c = open(config_file)
except:
return None
text = "".join(c.readlines())
c.close()
text = text.replace('\t', '<TAB>')
text = text.replace('\n', '|\n')
return text
def show_args(self):
print("\n")
print("-" * 70)
print("Running", ' '.join(self.all_args()))
msg = "with config from %s" % self.config_file_used()
config_text = self.config_text_used()
if config_text is not None:
msg += ": [%s]" % config_text
print(msg)
def run(self, expected_exitcode = None):
p = subprocess.Popen(self.all_args(), stdout=subprocess.PIPE, stderr=subprocess.PIPE)
if not p:
raise RuntimeError("failed to start subprocess")
print("Started subprocess pid %d" % p.pid)
# Wait for end of process for short time when daemonize expected
# and for longer time when exit is expected - to avoid false
# negatives for overloaded machines
timeout = 2
if expected_exitcode is not None:
timeout = 30
completed = self.subproc_completed_within(p, timeout)
if completed:
(stdout, stderr) = p.communicate()
if sys.version_info[0] >= 3:
# only expect ASCII/UTF-8 encodings for the obtained input bytes
stdout, stderr = str(stdout, 'UTF-8'), str(stderr, 'UTF-8')
self.show_output(stdout, stderr)
return (p.pid, p.returncode, stdout, stderr)
return (p.pid, None, None, None)
|
