def test_prepare_group_subservices(self): '''Test prepare group with multiple internal dependencies.''' group = ServiceGroup('GROUP') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') subserv_a = Service('SUB1') subserv_b = Service('SUB2') subserv_c = Service('SUB3') subserv_a.add_action(ac_suc1) subserv_b.add_action(ac_suc2) subserv_c.add_action(ac_suc3) group.add_inter_dep(target=subserv_a) group.add_inter_dep(target=subserv_b) group.add_inter_dep(base=subserv_a, target=subserv_c) group.add_inter_dep(base=subserv_b, target=subserv_c) group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(subserv_a.status, DONE) self.assertEqual(subserv_b.status, DONE) self.assertEqual(subserv_c.status, DONE)
def test_prepare_group_subservices(self): '''Test prepare group with multiple internal dependencies.''' group = ServiceGroup('GROUP') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') subserv_a = Service('SUB1') subserv_b = Service('SUB2') subserv_c = Service('SUB3') subserv_a.add_action(ac_suc1) subserv_b.add_action(ac_suc2) subserv_c.add_action(ac_suc3) group.add_inter_dep(target=subserv_a) group.add_inter_dep(target=subserv_b) group.add_inter_dep(base=subserv_a, target=subserv_c) group.add_inter_dep(base=subserv_b, target=subserv_c) group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(subserv_a.status, DONE) self.assertEqual(subserv_b.status, DONE) self.assertEqual(subserv_c.status, DONE)
def test_has_subservice(self): '''Test whether a service is an internal dependency of a group''' group = ServiceGroup('group') serv = Service('intern_service') self.assertFalse(group.has_subservice(serv.name)) group.add_inter_dep(target=serv) self.assertTrue(group.has_subservice(serv.name))
def test_has_subservice(self): '''Test whether a service is an internal dependency of a group''' group = ServiceGroup('group') serv = Service('intern_service') self.assertFalse(group.has_subservice(serv.name)) group.add_inter_dep(target=serv) self.assertTrue(group.has_subservice(serv.name))
def setUp(self): ''' Set up the graph of services within the service manager Graph _ start group --> service / `- stop ''' CLICommon.setUp(self) # ServiceGroup group = ServiceGroup('ServiceGroup') # Service self.service = service = Service('service') service.desc = 'I am the service' # Actions start_action = Action('start', command='/bin/true') stop_action = Action('stop', command='/bin/false') self.timeout_action = Action('timeout', command='sleep 1', timeout=0.1) start_action.inherits_from(service) stop_action.inherits_from(service) service.add_action(start_action) service.add_action(stop_action) service.add_action(self.timeout_action) # Build graph group.add_inter_dep(target=service) # Register services within the manager self.manager.register_services(group, service)
def test_update_target(self): '''Test update of the target of a group of services''' grp = ServiceGroup('G') srva = Service('A') grp.add_inter_dep(target=srva) grp.update_target('fortoy[5-10]') self.assertTrue(grp.target == NodeSet('fortoy[5-10]')) self.assertTrue(srva.target == NodeSet('fortoy[5-10]'))
def test_missing_group(self): """A group with only MISSING services should be MISSING""" grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('stop', command='/bin/true')) grp.add_inter_dep(target=svc1) grp.run('start') self.assertEqual(grp.status, MISSING)
def test_update_target(self): '''Test update of the target of a group of services''' grp = ServiceGroup('G') srva = Service('A') grp.add_inter_dep(target=srva) grp.update_target('fortoy[5-10]') self.assertTrue(grp.target == NodeSet('fortoy[5-10]')) self.assertTrue(srva.target == NodeSet('fortoy[5-10]'))
def test_missing_group(self): """A group with only MISSING services should be MISSING""" grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('stop', command='/bin/true')) grp.add_inter_dep(target=svc1) grp.run('start') self.assertEqual(grp.status, MISSING)
def test_prepare_group_subservice(self): '''Test prepare group with an internal dependency.''' group = ServiceGroup('GROUP') subserv = Service('SUB1') subserv.add_action(Action('start', command='/bin/true')) group.add_inter_dep(target=subserv) group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(subserv.status, DONE)
def setUp(self): ''' Set up the graph of services within the service manager Graph __ S2 __ I1 S1 / -- G1 -- (src) / ^ -- (sink) `-- S3 --/ `-- I2 Each node has an action start and an action stop ''' CLICommon.setUp(self) svc1 = Service('S1') svc1.desc = 'I am the service S1' self.svc2 = svc2 = Service('S2') svc2.desc = 'I am the service S2' svc3 = Service('S3') svc3.desc = 'I am the service S3' group1 = ServiceGroup('G1') inter1 = Service('I1') inter1.desc = 'I am the service I1' inter2 = Service('I2') inter2.desc = 'I am the service I2' # Actions S1 start_svc1 = Action('start', HOSTNAME + ', BADNODE', '/bin/true') start_svc1.delay = 1 stop_svc1 = Action('stop', HOSTNAME + ',BADNODE', '/bin/true') stop_svc1.delay = 1 svc1.add_actions(start_svc1, stop_svc1) # Actions S2 svc2.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/true')) svc2.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/true')) # Actions S3 svc3.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/false')) svc3.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/false')) # Actions I1 inter1.add_action(Action('start', HOSTNAME, 'echo ok')) inter1.add_action(Action('stop', HOSTNAME, 'echo ok')) # Actions I2 inter2.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/true')) inter2.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/true')) # Build graph svc1.add_dep(target=svc2) svc1.add_dep(target=svc3) svc3.add_dep(target=group1) inter2.add_dep(inter1) group1.add_inter_dep(target=inter1) group1.add_inter_dep(target=inter2) # Register services within the manager self.manager.add_service(svc1) self.manager.add_service(svc2) self.manager.add_service(svc3) self.manager.add_service(group1)
def test_prepare_group_subservice(self): '''Test prepare group with an internal dependency.''' group = ServiceGroup('GROUP') subserv = Service('SUB1') subserv.add_action(Action('start', command='/bin/true')) group.add_inter_dep(target=subserv) group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(subserv.status, DONE)
def test_graph_in_graph(self): """Test the DOT graph output with a servicegroup within a servicegroup""" grp = ServiceGroup('Group') subgrp = ServiceGroup('subGroup') grp.add_inter_dep(subgrp) self.assertEqual( grp.graph(), 'subgraph "cluster_Group" {\nlabel="Group";\nstyle=rounded;\nnode [style=filled];\n"Group.__hook" [style=invis];\nsubgraph "cluster_Group.subGroup" {\nlabel="Group.subGroup";\nstyle=rounded;\nnode [style=filled];\n"Group.subGroup.__hook" [style=invis];\n}\n}\n' )
def test_skip(self): """Test skip method for ServiceGroup""" grp = ServiceGroup('group') srv = Service('skipped') srv.add_action(Action('start', target=NodeSet('foo'), command='/bin/true')) grp.add_inter_dep(target=srv) self.assertFalse(grp.to_skip('start')) grp.skip() self.assertTrue(grp.to_skip('start'))
def test_skipped_group(self): """A group with only SKIPPED services should be SKIPPED""" grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('start', target="@NOTHING", command=':')) svc2 = Service('svc2') svc2.add_action(Action('start', target="@NOTHING", command=':')) grp.add_inter_dep(target=svc1) grp.add_inter_dep(target=svc2) grp.run('start') self.assertEqual(grp.status, SKIPPED)
def test_skipped_group(self): """A group with only SKIPPED services should be SKIPPED""" grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('start', target="@NOTHING", command=':')) svc2 = Service('svc2') svc2.add_action(Action('start', target="@NOTHING", command=':')) grp.add_inter_dep(target=svc1) grp.add_inter_dep(target=svc2) grp.run('start') self.assertEqual(grp.status, SKIPPED)
def test_eval_deps_status_error(self): '''Test the method eval_deps_status DEP_ERROR''' group = ServiceGroup('group') e1 = Service('E1') e2 = Service('E2') e1.status = DEP_ERROR group.add_dep(target=e1) group.add_dep(target=e2) group.add_inter_dep(target=Service('I1')) self.assertEqual(group.eval_deps_status(), DEP_ERROR) self.assertEqual(group.eval_deps_status(), DEP_ERROR)
def setUp(self): ''' Set up the graph of services within the service manager Graph __ S2 __ I1 S1 / -- G1 -- (src) / ^ -- (sink) `-- S3 --/ `-- I2 Each node has an action start and an action stop ''' CLICommon.setUp(self) svc1 = Service('S1') svc1.desc = 'I am the service S1' self.svc2 = svc2 = Service('S2') svc2.desc = 'I am the service S2' svc3 = Service('S3') svc3.desc = 'I am the service S3' group1 = ServiceGroup('G1') inter1 = Service('I1') inter1.desc = 'I am the service I1' inter2 = Service('I2') inter2.desc = 'I am the service I2' # Actions S1 start_svc1 = Action('start', HOSTNAME + ', BADNODE', '/bin/true') start_svc1.delay = 1 stop_svc1 = Action('stop', HOSTNAME + ',BADNODE', '/bin/true') stop_svc1.delay = 1 svc1.add_actions(start_svc1, stop_svc1) # Actions S2 svc2.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/true')) svc2.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/true')) # Actions S3 svc3.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/false')) svc3.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/false')) # Actions I1 inter1.add_action(Action('start', HOSTNAME, 'echo ok')) inter1.add_action(Action('stop', HOSTNAME, 'echo ok')) # Actions I2 inter2.add_action(Action('start', HOSTNAME + ',BADNODE', '/bin/true')) inter2.add_action(Action('stop', HOSTNAME + ',BADNODE', '/bin/true')) # Build graph svc1.add_dep(target=svc2) svc1.add_dep(target=svc3) svc3.add_dep(target=group1) inter2.add_dep(inter1) group1.add_inter_dep(target=inter1) group1.add_inter_dep(target=inter2) # Register services within the manager self.manager.register_services(svc1, svc2, svc3, group1)
def test_eval_deps_status_error(self): '''Test the method eval_deps_status DEP_ERROR''' group = ServiceGroup('group') e1 = Service('E1') e2 = Service('E2') e1.status = DEP_ERROR group.add_dep(target=e1) group.add_dep(target=e2) group.add_inter_dep(target=Service('I1')) self.assertEqual(group.eval_deps_status(), DEP_ERROR) self.assertEqual(group.eval_deps_status(), DEP_ERROR)
def test_eval_deps_status_done(self): '''Test the method eval_deps_status NO_STATUS''' group = ServiceGroup('group') e1 = Service('E1') e2 = Service('E2') group.add_dep(target=e1) group.add_dep(target=e2) group.add_inter_dep(target=Service('I1')) self.assertEqual(group.eval_deps_status(), NO_STATUS) e1.status = DONE e2.status = DONE self.assertEqual(group.eval_deps_status(), NO_STATUS)
def test_eval_deps_status_done(self): '''Test the method eval_deps_status NO_STATUS''' group = ServiceGroup('group') e1 = Service('E1') e2 = Service('E2') group.add_dep(target=e1) group.add_dep(target=e2) group.add_inter_dep(target=Service('I1')) self.assertEqual(group.eval_deps_status(), NO_STATUS) e1.status = DONE e2.status = DONE self.assertEqual(group.eval_deps_status(), NO_STATUS)
def test_eval_deps_status_ws(self): '''Test the method eval_deps_status WAITING_STATUS''' group = ServiceGroup('group') ext1 = Service('E1') ext2 = Service('E2') ext1.status = DONE ext2.status = WARNING group.add_dep(target=ext1) group.add_dep(target=ext2) int1 = Service('E1') group.add_inter_dep(target=int1) int1.status = WAITING_STATUS self.assertEqual(group.eval_deps_status(), WAITING_STATUS)
def test_eval_deps_status_ws(self): '''Test the method eval_deps_status WAITING_STATUS''' group = ServiceGroup('group') ext1 = Service('E1') ext2 = Service('E2') ext1.status = DONE ext2.status = WARNING group.add_dep(target=ext1) group.add_dep(target=ext2) int1 = Service('E1') group.add_inter_dep(target=int1) int1.status = WAITING_STATUS self.assertEqual(group.eval_deps_status(), WAITING_STATUS)
def test_group_with_weak_dep_error(self): """A group with a weak dep error runs fine (add_inter_dep()).""" dep1 = Service('dep1') dep1.add_action(Action('stop', command='/bin/false')) grp = ServiceGroup('group') svc = Service('svc') svc.add_action(Action('stop', command='/bin/true')) grp.add_inter_dep(svc) grp.add_dep(dep1, sgth=REQUIRE_WEAK) grp.run('stop') self.assertEqual(grp.status, DONE)
def test_group_with_weak_dep_error(self): """A group with a weak dep error runs fine (add_inter_dep()).""" dep1 = Service('dep1') dep1.add_action(Action('stop', command='/bin/false')) grp = ServiceGroup('group') svc = Service('svc') svc.add_action(Action('stop', command='/bin/true')) grp.add_inter_dep(svc) grp.add_dep(dep1, sgth=REQUIRE_WEAK) grp.run('stop') self.assertEqual(grp.status, DONE)
def test_run_stop_on_group(self): '''Test stop algorithm on a group''' group = ServiceGroup('G1') i1 = Service('I1') i1.add_action(Action('stop', command='/bin/true')) group.add_inter_dep(target=i1) s1 = Service('S1') s1.add_action(Action('stop', command='/bin/true')) s1.add_dep(target=group) s1.algo_reversed = True group.algo_reversed = True group.run('stop') self.assertEqual(s1.status, DONE) self.assertEqual(i1.status, DONE) self.assertEqual(group.status, DONE)
def test_run_stop_on_group(self): '''Test stop algorithm on a group''' group = ServiceGroup('G1') i1 = Service('I1') i1.add_action(Action('stop', command='/bin/true')) group.add_inter_dep(target=i1) s1 = Service('S1') s1.add_action(Action('stop', command='/bin/true')) s1.add_dep(target=group) s1.algo_reversed = True group.algo_reversed = True group.run('stop') self.assertEqual(s1.status, DONE) self.assertEqual(i1.status, DONE) self.assertEqual(group.status, DONE)
def test_filter(self): """ServiceGroup supports filter dependency""" dep = Service('dep') dep.add_action(Action('stop', command='false', target=HOSTNAME)) grp = ServiceGroup('group') svc = Service('svc') svc.add_action(Action('stop', command='true', target=HOSTNAME)) grp.add_inter_dep(svc) grp.add_dep(dep, sgth=FILTER) grp.run('stop') self.assertEqual(dep.status, ERROR) self.assertEqual(svc.status, SKIPPED) self.assertEqual(grp.status, SKIPPED)
def test_skipped_group_dep_error(self): """A full SKIPPED service group with DEP_ERROR should be SKIPPED""" svc = Service('error') svc.add_action(Action('start', command='/bin/false')) grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('start', target="@NOTHING", command=':')) svc2 = Service('svc2') svc2.add_action(Action('start', target="@NOTHING", command=':')) grp.add_inter_dep(target=svc1) grp.add_inter_dep(target=svc2) grp.add_dep(svc, sgth=REQUIRE) grp.run('start') self.assertEqual(svc.status, ERROR) self.assertEqual(grp.status, SKIPPED)
def test_skipped_group_dep_error(self): """A full SKIPPED service group with DEP_ERROR should be SKIPPED""" svc = Service('error') svc.add_action(Action('start', command='/bin/false')) grp = ServiceGroup('group') svc1 = Service('svc1') svc1.add_action(Action('start', target="@NOTHING", command=':')) svc2 = Service('svc2') svc2.add_action(Action('start', target="@NOTHING", command=':')) grp.add_inter_dep(target=svc1) grp.add_inter_dep(target=svc2) grp.add_dep(svc, sgth=REQUIRE) grp.run('start') self.assertEqual(svc.status, ERROR) self.assertEqual(grp.status, SKIPPED)
def test_reset_service_group(self): '''Test the ability to reset values of a service group''' group = ServiceGroup('GROUP') ser1 = Service('I1') action = Action(name='start', delay=3) action.maxretry = 5 action.tries = 3 action.status = DONE ser1.add_action(action) ser1.status = ERROR group.add_inter_dep(target=ser1) group.status = DEP_ERROR group.reset() self.assertEqual(group.status, NO_STATUS) self.assertEqual(ser1.status, NO_STATUS) self.assertEqual(action.status, NO_STATUS) self.assertEqual(action.tries, 0)
def test_reset_service_group(self): '''Test the ability to reset values of a service group''' group = ServiceGroup('GROUP') ser1 = Service('I1') action = Action(name='start', delay=3) action.maxretry = 5 action.tries = 3 action.status = DONE ser1.add_action(action) ser1.status = ERROR group.add_inter_dep(target=ser1) group.status = DEP_ERROR group.reset() self.assertEqual(group.status, NO_STATUS) self.assertEqual(ser1.status, NO_STATUS) self.assertEqual(action.status, NO_STATUS) self.assertEqual(action.tries, 0)
def test_search_node_graph(self): """Test search node in a graph trough a ServiceGroup""" group = ServiceGroup('GROUP') ser1 = Service('I1') ser2 = Service('I2') eser1 = Service('E1') eser2 = Service('E2') eser3 = Service('E3') group.add_inter_dep(target=ser1) group.add_inter_dep(target=ser2) group.add_dep(target=eser1, parent=False) group.add_dep(target=eser2) group.add_dep(target=eser3) self.assertTrue(group.search('I1')) self.assertTrue(group.search('E2')) self.assertTrue(eser1.search('I1')) self.assertTrue(eser1.search('E3')) self.assertFalse(group.search('E0'))
def test_run_partial_deps(self): '''Test start algorithm as soon as the calling point is done.''' serv = Service('NOT_CALLED') serv_a = ServiceGroup('CALLING_GROUP') serv_b = Service('SERV_1') serv_c = Service('SERV_2') act_suc1 = Action('start', command='/bin/true') act_suc2 = Action('start', command='/bin/true') serv_b.add_action(act_suc1) serv_c.add_action(act_suc2) serv.add_dep(serv_a) serv_a.add_dep(target=serv_b) serv_a.add_inter_dep(target=serv_c) serv_a.run('start') self.assertEqual(serv.status, NO_STATUS) self.assertEqual(serv_a.status, DONE) self.assertEqual(serv_b.status, DONE) self.assertEqual(serv_c.status, DONE)
def test_run_partial_deps(self): '''Test start algorithm as soon as the calling point is done.''' serv = Service('NOT_CALLED') serv_a = ServiceGroup('CALLING_GROUP') serv_b = Service('SERV_1') serv_c = Service('SERV_2') act_suc1 = Action('start', command='/bin/true') act_suc2 = Action('start', command='/bin/true') serv_b.add_action(act_suc1) serv_c.add_action(act_suc2) serv.add_dep(serv_a) serv_a.add_dep(target=serv_b) serv_a.add_inter_dep(target=serv_c) serv_a.run('start') self.assertEqual(serv.status, NO_STATUS) self.assertEqual(serv_a.status, DONE) self.assertEqual(serv_b.status, DONE) self.assertEqual(serv_c.status, DONE)
def test_search_node_graph(self): """Test search node in a graph trough a ServiceGroup""" group = ServiceGroup('GROUP') ser1 = Service('I1') ser2 = Service('I2') eser1 = Service('E1') eser2 = Service('E2') eser3 = Service('E3') group.add_inter_dep(target=ser1) group.add_inter_dep(target=ser2) group.add_dep(target=eser1, parent=False) group.add_dep(target=eser2) group.add_dep(target=eser3) self.assertTrue(group.search('I1')) self.assertTrue(group.search('E2')) self.assertTrue(eser1.search('I1')) self.assertTrue(eser1.search('E3')) self.assertFalse(group.search('E0'))
def test_inheritance(self): '''Test inheritance between on a group''' ser = Service('parent') ser.target = '127.0.0.1' ser.timeout = 15 group = ServiceGroup('group') subser1 = Service('subser1') subser1.target = HOSTNAME subser2 = Service('subser2') subser2.timeout = None group.add_inter_dep(target=subser1) group.add_inter_dep(target=subser2) group.inherits_from(ser) self.assertEqual(group.target, NodeSet('127.0.0.1')) self.assertEqual(group.timeout, 15) self.assertEqual(subser1.target, NodeSet(HOSTNAME)) self.assertEqual(subser1.timeout, 15) self.assertEqual(subser2.target, NodeSet('127.0.0.1')) self.assertEqual(subser2.timeout, None)
def test_inheritance(self): """Test inheritance between services""" ser = Service('parent') ser.target = '127.0.0.1' ser.timeout = 15 group = ServiceGroup('group') subser1 = Service('subser1') subser1.target = HOSTNAME subser2 = Service('subser2') subser2.timeout = 3 group.add_inter_dep(target=subser1) group.add_inter_dep(target=subser2) group.inherits_from(ser) self.assertEqual(group.target, NodeSet('127.0.0.1')) self.assertEqual(group.timeout, 15) self.assertEqual(subser1.target, NodeSet(HOSTNAME)) self.assertEqual(subser1.timeout, 15) self.assertEqual(subser2.target, NodeSet('127.0.0.1')) self.assertEqual(subser2.timeout, 3)
def test_remove_inter_dep(self): '''Test ability to remove a dependency in a subgraph''' group = ServiceGroup('GROUP') s1 = Service('alpha') s2 = Service('beta') s3 = Service('lambda') group.add_inter_dep(target=s1) group.add_inter_dep(target=s2) group.add_inter_dep(target=s3) group.add_inter_dep(base=s1, target=s3) group.add_inter_dep(base=s2, target=s3) group.remove_inter_dep('lambda') self.assertTrue(s1.has_parent_dep('sink')) self.assertTrue(s2.has_parent_dep('sink')) self.assertTrue(s1.has_child_dep('source')) self.assertTrue(s2.has_child_dep('source')) self.assertFalse(s1.has_parent_dep('lambda')) self.assertFalse(s2.has_parent_dep('lambda')) group.remove_inter_dep('alpha') self.assertFalse(group._source.has_parent_dep('alpha')) self.assertTrue(group._source.has_parent_dep('beta')) self.assertFalse(group._sink.has_child_dep('alpha')) self.assertTrue(group._sink.has_child_dep('beta')) group.remove_inter_dep('beta') self.assertFalse(group._source.parents) self.assertFalse(group._sink.children)
def test_remove_inter_dep(self): '''Test ability to remove a dependency in a subgraph''' group = ServiceGroup('GROUP') s1 = Service('alpha') s2 = Service('beta') s3 = Service('lambda') group.add_inter_dep(target=s1) group.add_inter_dep(target=s2) group.add_inter_dep(target=s3) group.add_inter_dep(base=s1, target=s3) group.add_inter_dep(base=s2, target=s3) group.remove_inter_dep('lambda') self.assertTrue(s1.has_parent_dep('sink')) self.assertTrue(s2.has_parent_dep('sink')) self.assertTrue(s1.has_child_dep('source')) self.assertTrue(s2.has_child_dep('source')) self.assertFalse(s1.has_parent_dep('lambda')) self.assertFalse(s2.has_parent_dep('lambda')) group.remove_inter_dep('alpha') self.assertFalse(group._source.has_parent_dep('alpha')) self.assertTrue(group._source.has_parent_dep('beta')) self.assertFalse(group._sink.has_child_dep('alpha')) self.assertTrue(group._sink.has_child_dep('beta')) group.remove_inter_dep('beta') self.assertFalse(group._source.parents) self.assertFalse(group._sink.children)
def test_prepare_group_with_errors_one(self): '''Test prepare a group terminated by WARNING''' # Group group = ServiceGroup('GROUP') # Internal inter_serv1 = Service('INT_SERV1') inter_serv2 = Service('INT_SERV2') inter_serv3 = Service('INT_SERV3') # External ext_serv1 = Service('EXT_SERV1') ext_serv2 = Service('EXT_SERV2') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') ac_err1 = Action('start', command='/bin/false') ac_err2 = Action('start', command='/bin/false') # Add actions inter_serv1.add_action(ac_suc1) inter_serv2.add_action(ac_suc2) inter_serv3.add_action(ac_err1) ext_serv1.add_action(ac_suc3) ext_serv2.add_action(ac_err2) # Add dependencies group.add_inter_dep(target=inter_serv1) group.add_inter_dep(target=inter_serv2) group.add_inter_dep(base=inter_serv2, target=inter_serv3, sgth=REQUIRE_WEAK) group.add_dep(ext_serv1) group.add_dep(target=ext_serv2, sgth=REQUIRE_WEAK) # Prepare group group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(ext_serv1.status, DONE) self.assertEqual(ext_serv2.status, ERROR) self.assertEqual(inter_serv1.status, DONE) self.assertEqual(inter_serv2.status, DONE) self.assertEqual(inter_serv3.status, ERROR)
def test_prepare_group_with_errors_one(self): '''Test prepare a group terminated by WARNING''' # Group group = ServiceGroup('GROUP') # Internal inter_serv1 = Service('INT_SERV1') inter_serv2 = Service('INT_SERV2') inter_serv3 = Service('INT_SERV3') # External ext_serv1 = Service('EXT_SERV1') ext_serv2 = Service('EXT_SERV2') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') ac_err1 = Action('start', command='/bin/false') ac_err2 = Action('start', command='/bin/false') # Add actions inter_serv1.add_action(ac_suc1) inter_serv2.add_action(ac_suc2) inter_serv3.add_action(ac_err1) ext_serv1.add_action(ac_suc3) ext_serv2.add_action(ac_err2) # Add dependencies group.add_inter_dep(target=inter_serv1) group.add_inter_dep(target=inter_serv2) group.add_inter_dep(base=inter_serv2, target=inter_serv3, sgth=REQUIRE_WEAK) group.add_dep(ext_serv1) group.add_dep(target=ext_serv2, sgth=REQUIRE_WEAK) # Prepare group group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(ext_serv1.status, DONE) self.assertEqual(ext_serv2.status, ERROR) self.assertEqual(inter_serv1.status, DONE) self.assertEqual(inter_serv2.status, DONE) self.assertEqual(inter_serv3.status, ERROR)
def test_prepare_group_internal_external_deps(self): '''Test prepare a group with internal and external dependencies''' # Group group = ServiceGroup('GROUP') # Internal inter_serv1 = Service('INT_SERV1') inter_serv2 = Service('INT_SERV2') inter_serv3 = Service('INT_SERV3') # External ext_serv1 = Service('EXT_SERV1') ext_serv2 = Service('EXT_SERV2') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') ac_suc4 = Action('start', command='/bin/true') ac_suc5 = Action('start', command='/bin/true') # Add actions inter_serv1.add_action(ac_suc1) inter_serv2.add_action(ac_suc2) inter_serv3.add_action(ac_suc3) ext_serv1.add_action(ac_suc4) ext_serv2.add_action(ac_suc5) # Add dependencies group.add_inter_dep(target=inter_serv1) group.add_inter_dep(target=inter_serv2) group.add_inter_dep(base=inter_serv2, target=inter_serv3) group.add_dep(ext_serv1) group.add_dep(ext_serv2) # Prepare group group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(ext_serv1.status, DONE) self.assertEqual(ext_serv2.status, DONE) self.assertEqual(inter_serv1.status, DONE) self.assertEqual(inter_serv2.status, DONE) self.assertEqual(inter_serv3.status, DONE)
def test_add_dep_service_group(self): '''Test ability to add dependencies to a ServiceGroup''' ser_group = ServiceGroup('GROUP') s1 = Service('alpha') s1.add_action(Action('start', HOSTNAME, '/bin/true')) s2 = Service('beta') s2.add_action(Action('action', HOSTNAME, '/bin/true')) s3 = Service('lambda') ser_group.add_inter_dep(target=s1) ser_group.add_inter_dep(target=s2) ser_group.add_dep(target=s3) self.assertTrue(ser_group.has_action('start')) self.assertTrue(ser_group.has_action('action')) self.assertTrue(s1.name in ser_group._source.parents) self.assertTrue(s1.name in ser_group._sink.children) self.assertTrue(s2.name in ser_group._source.parents) self.assertTrue(s2.name in ser_group._sink.children) self.assertFalse(s3.name in ser_group.children) self.assertTrue(s3.name in ser_group.parents) s4 = Service('theta') s4.add_action(Action('fire', HOSTNAME,'/bin/true')) ser_group.add_dep(target=s4, parent=False) self.assertTrue(s4.name in ser_group.children) self.assertTrue(s4.has_parent_dep(ser_group.name))
def test_add_inter_dep_service_group_second(self): '''Test ability to add dependencies to the subgraph N2''' group = ServiceGroup('GROUP') s1 = Service('alpha') s2 = Service('beta') s3 = Service('lambda') group.add_inter_dep(target=s1) group.add_inter_dep(base=s1, target=s2) group.add_inter_dep(target=s3) group.add_inter_dep(base=s1, target=s3) self.assertTrue(s1.has_parent_dep('beta')) self.assertTrue(s1.has_parent_dep('lambda')) self.assertTrue(s1.has_child_dep('source')) self.assertTrue(s2.has_child_dep('alpha')) self.assertTrue(s3.has_child_dep('alpha')) self.assertTrue(s2.has_parent_dep('sink')) self.assertTrue(s3.has_parent_dep('sink'))
def test_add_dep_service_group(self): '''Test ability to add dependencies to a ServiceGroup''' ser_group = ServiceGroup('GROUP') s1 = Service('alpha') s1.add_action(Action('start', HOSTNAME, '/bin/true')) s2 = Service('beta') s2.add_action(Action('action', HOSTNAME, '/bin/true')) s3 = Service('lambda') ser_group.add_inter_dep(target=s1) ser_group.add_inter_dep(target=s2) ser_group.add_dep(target=s3) self.assertTrue(ser_group.has_action('start')) self.assertTrue(ser_group.has_action('action')) self.assertTrue(s1.name in ser_group._source.parents) self.assertTrue(s1.name in ser_group._sink.children) self.assertTrue(s2.name in ser_group._source.parents) self.assertTrue(s2.name in ser_group._sink.children) self.assertFalse(s3.name in ser_group.children) self.assertTrue(s3.name in ser_group.parents) s4 = Service('theta') s4.add_action(Action('fire', HOSTNAME, '/bin/true')) ser_group.add_dep(target=s4, parent=False) self.assertTrue(s4.name in ser_group.children) self.assertTrue(s4.has_parent_dep(ser_group.name))
def test_add_inter_dep_service_group_second(self): '''Test ability to add dependencies to the subgraph N2''' group = ServiceGroup('GROUP') s1 = Service('alpha') s2 = Service('beta') s3 = Service('lambda') group.add_inter_dep(target=s1) group.add_inter_dep(base=s1, target=s2) group.add_inter_dep(target=s3) group.add_inter_dep(base=s1, target=s3) self.assertTrue(s1.has_parent_dep('beta')) self.assertTrue(s1.has_parent_dep('lambda')) self.assertTrue(s1.has_child_dep('source')) self.assertTrue(s2.has_child_dep('alpha')) self.assertTrue(s3.has_child_dep('alpha')) self.assertTrue(s2.has_parent_dep('sink')) self.assertTrue(s3.has_parent_dep('sink'))
def test_prepare_group_internal_external_deps(self): '''Test prepare a group with internal and external dependencies''' # Group group = ServiceGroup('GROUP') # Internal inter_serv1 = Service('INT_SERV1') inter_serv2 = Service('INT_SERV2') inter_serv3 = Service('INT_SERV3') # External ext_serv1 = Service('EXT_SERV1') ext_serv2 = Service('EXT_SERV2') ac_suc1 = Action('start', command='/bin/true') ac_suc2 = Action('start', command='/bin/true') ac_suc3 = Action('start', command='/bin/true') ac_suc4 = Action('start', command='/bin/true') ac_suc5 = Action('start', command='/bin/true') # Add actions inter_serv1.add_action(ac_suc1) inter_serv2.add_action(ac_suc2) inter_serv3.add_action(ac_suc3) ext_serv1.add_action(ac_suc4) ext_serv2.add_action(ac_suc5) # Add dependencies group.add_inter_dep(target=inter_serv1) group.add_inter_dep(target=inter_serv2) group.add_inter_dep(base=inter_serv2, target=inter_serv3) group.add_dep(ext_serv1) group.add_dep(ext_serv2) # Prepare group group.run('start') self.assertEqual(group.status, DONE) self.assertEqual(ext_serv1.status, DONE) self.assertEqual(ext_serv2.status, DONE) self.assertEqual(inter_serv1.status, DONE) self.assertEqual(inter_serv2.status, DONE) self.assertEqual(inter_serv3.status, DONE)
def test_core_behaviour_reverse(self): '''Test ability of the core to solve a large graph in reverse mode''' # Define Group 1 grp1 = ServiceGroup('GRP1') grp1.algo_reversed = True grp1_i1 = Service('I1') grp1_i1.algo_reversed = True grp1_i2 = Service('I2') grp1_i2.algo_reversed = True grp1_i3 = Service('I3') grp1_i3.algo_reversed = True grp1_i1.add_action(Action('stop', command='/bin/true')) grp1_i2.add_action(Action('stop', command='/bin/true')) grp1_i3.add_action(Action('stop', command='/bin/true')) grp1.add_inter_dep(target=grp1_i1) grp1.add_inter_dep(base=grp1_i1, target=grp1_i2) grp1.add_inter_dep(target=grp1_i3) # Define Group 2 grp2 = ServiceGroup('GRP2') grp2.algo_reversed = True grp2_i1 = Service('I1') grp2_i1.algo_reversed = True grp2_i2 = Service('I2') grp2_i2.algo_reversed = True grp2_i1.add_action(Action('stop', command='/bin/true')) grp2_i2.add_action(Action('stop', command='/bin/true')) grp2.add_inter_dep(target=grp2_i1) grp2.add_inter_dep(base=grp2_i1, target=grp2_i2) # Define Group init svc1 = Service('S1') svc1.algo_reversed = True svc1.add_action(Action('stop', command='/bin/true')) svc2 = Service('S2') svc2.algo_reversed = True svc2.add_action(Action('stop', command='/bin/true')) svc3 = Service('S3') svc3.algo_reversed = True svc3.add_action(Action('stop', command='/bin/true')) group_init = ServiceGroup('GROUP_INIT') group_init.algo_reversed = True group_init.add_inter_dep(target=svc1) group_init.add_inter_dep(base=svc1, target=svc2, sgth=REQUIRE_WEAK) group_init.add_inter_dep(base=svc1, target=grp1) group_init.add_inter_dep(base=svc2, target=svc3) group_init.add_inter_dep(base=grp1, target=svc3) group_init.add_inter_dep(base=svc3, target=grp2) # Solve the graph group_init.run('stop') # Assertions self.assertEqual(grp2.status, DONE) self.assertEqual(svc3.status, DONE) self.assertEqual(svc2.status, DONE) self.assertEqual(grp1.status, DONE) self.assertEqual(svc1.status, DONE) self.assertEqual(group_init.status, DONE)
def test_graph_in_graph(self): """Test the DOT graph output with a servicegroup within a servicegroup""" grp = ServiceGroup('Group') subgrp = ServiceGroup('subGroup') grp.add_inter_dep(subgrp) self.assertEqual(grp.graph(), 'subgraph "cluster_Group" {\nlabel="Group";\nstyle=rounded;\nnode [style=filled];\n"Group.__hook" [style=invis];\nsubgraph "cluster_Group.subGroup" {\nlabel="Group.subGroup";\nstyle=rounded;\nnode [style=filled];\n"Group.subGroup.__hook" [style=invis];\n}\n}\n')
def test_core_behaviour_reverse(self): '''Test ability of the core to solve a large graph in reverse mode''' # Define Group 1 grp1 = ServiceGroup('GRP1') grp1.algo_reversed = True grp1_i1 = Service('I1') grp1_i1.algo_reversed = True grp1_i2 = Service('I2') grp1_i2.algo_reversed = True grp1_i3 = Service('I3') grp1_i3.algo_reversed = True grp1_i1.add_action(Action('stop', command='/bin/true')) grp1_i2.add_action(Action('stop', command='/bin/true')) grp1_i3.add_action(Action('stop', command='/bin/true')) grp1.add_inter_dep(target=grp1_i1) grp1.add_inter_dep(base=grp1_i1, target=grp1_i2) grp1.add_inter_dep(target=grp1_i3) # Define Group 2 grp2 = ServiceGroup('GRP2') grp2.algo_reversed = True grp2_i1 = Service('I1') grp2_i1.algo_reversed = True grp2_i2 = Service('I2') grp2_i2.algo_reversed = True grp2_i1.add_action(Action('stop', command='/bin/true')) grp2_i2.add_action(Action('stop', command='/bin/true')) grp2.add_inter_dep(target=grp2_i1) grp2.add_inter_dep(base=grp2_i1, target=grp2_i2) # Define Group init svc1 = Service('S1') svc1.algo_reversed = True svc1.add_action(Action('stop', command='/bin/true')) svc2 = Service('S2') svc2.algo_reversed = True svc2.add_action(Action('stop', command='/bin/true')) svc3 = Service('S3') svc3.algo_reversed = True svc3.add_action(Action('stop', command='/bin/true')) group_init = ServiceGroup('GROUP_INIT') group_init.algo_reversed = True group_init.add_inter_dep(target=svc1) group_init.add_inter_dep(base=svc1, target=svc2, sgth=REQUIRE_WEAK) group_init.add_inter_dep(base=svc1, target=grp1) group_init.add_inter_dep(base=svc2, target=svc3) group_init.add_inter_dep(base=grp1, target=svc3) group_init.add_inter_dep(base=svc3, target=grp2) # Solve the graph group_init.run('stop') # Assertions self.assertEqual(grp2.status, DONE) self.assertEqual(svc3.status, DONE) self.assertEqual(svc2.status, DONE) self.assertEqual(grp1.status, DONE) self.assertEqual(svc1.status, DONE) self.assertEqual(group_init.status, DONE)