1 ##############################################################################
2 # Copyright (c) 2017 ZTE Corporation and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
10 from os.path import isfile, join
13 from alarm import Alarm
15 from consumer import get_consumer
16 from image import Image
17 from instance import Instance
18 from inspector import get_inspector
19 import logger as doctor_log
21 from network import Network
22 from monitor import get_monitor
25 LOG = doctor_log.Logger('doctor').getLogger()
28 class DoctorTest(object):
30 def __init__(self, conf):
32 self.image = Image(self.conf, LOG)
33 self.user = User(self.conf, LOG)
34 self.network = Network(self.conf, LOG)
35 self.instance = Instance(self.conf, LOG)
36 self.alarm = Alarm(self.conf, LOG)
37 self.inspector = get_inspector(self.conf, LOG)
38 self.monitor = get_monitor(self.conf,
39 self.inspector.get_inspector_url(),
41 self.consumer = get_consumer(self.conf, LOG)
44 # prepare the cloud env
46 # preparing VM image...
49 # creating test user...
51 self.user.update_quota()
55 self.instance.create()
56 self.instance.wait_for_vm_launch()
61 # starting doctor sample components...
62 self.inspector.start()
69 LOG.info('doctor test starting.......')
73 # injecting host failure...
75 # verify the test results
77 except Exception as e:
78 LOG.error('doctor test failed, Exception=%s' % e)
85 self.instance.delete()
96 doctor_root_dir = os.path.dirname(sys.path[0])
97 config_file_dir = '{0}/{1}'.format(doctor_root_dir, 'etc/')
98 config_files = [join(config_file_dir, f) for f in os.listdir(config_file_dir)
99 if isfile(join(config_file_dir, f))]
101 conf = config.prepare_conf(args=sys.argv[1:],
102 config_files=config_files)
104 doctor = DoctorTest(conf)
108 if __name__ == '__main__':