forked from zenoss/ZenPacks.zenoss.MySqlMonitor
-
Notifications
You must be signed in to change notification settings - Fork 0
/
functional_test.py
90 lines (64 loc) · 2.16 KB
/
functional_test.py
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
import os
import subprocess
from pprint import pformat
from add_device import add_device, model_device, python_monitor_device
ZENPACK_NAME = 'ZenPacks.zenoss.MySqlMonitor'
DUMP_FILE = '/home/zenoss/all.sql'
def main():
if not unittests_ok():
print 'Unittests failed'
return False
try:
dump_zodb()
zenpack_remove()
zenpack_install()
zenoss('restart')
device = add_device()
model_device(device)
monitoring_results = python_monitor_device(device)
assert '- is that the correct name?' not in monitoring_results
return True
except Exception:
raise
finally:
zenoss('stop')
kill_proc()
restore_zodb()
zenoss('start')
return True
def here():
''' return directory which contains this file '''
return os.path.abspath(os.path.dirname(__file__))
def unittests_ok():
bprint(''' Run unittests and check if ok ''')
return subprocess.check_call(['make', 'test']) == 0
def dump_zodb():
bprint(''' Dump zodb database to file ''')
os.system('mysqldump -u root --all-databases > %s' % DUMP_FILE)
def kill_proc():
bprint('''Kill all mysql processes!''')
o = subprocess.check_output(['mysql', '--batch', '-u', 'root', '-e',
'SELECT id FROM information_schema.processlist\
WHERE Command = "Sleep"'])
for process_id in o.split()[1:]:
cmd = 'mysqladmin -u root KILL %s' % process_id
os.system(cmd)
def restore_zodb():
bprint('''Restoring zodb from file''')
os.system('mysql -u root -e "drop database zodb;\
drop database zodb_session"')
os.system('mysql -u root < %s' % DUMP_FILE)
def zenpack_remove():
bprint('removing zenpack')
assert os.system('zenpack --remove %s' % ZENPACK_NAME) == 0
def zenpack_install():
bprint('install zenpack')
assert os.system('zenpack --link --install %s' % here()) == 0
def zenoss(cmd):
bprint('zenoss %s' % cmd)
os.system('zenoss %s' % cmd)
def bprint(*args):
print '\033[94m%s\033[0m' % pformat(*args)
if __name__ == '__main__':
if main():
print 'All ok'