def test_prepare_empty_group_external_deps(self):
'''Test prepare an empty group with a single external dependency.'''
group = ServiceGroup('GROUP')
ext_serv = Service('EXT_SERV')
ac_suc = Action('start', command='/bin/true')
ext_serv.add_action(ac_suc)
group.add_dep(ext_serv)
group.run('start')
self.assertEqual(ext_serv.status, DONE)
self.assertEqual(group.status, MISSING)
def test_prepare_empty_group_external_deps(self):
'''Test prepare an empty group with a single external dependency.'''
group = ServiceGroup('GROUP')
ext_serv = Service('EXT_SERV')
ac_suc = Action('start', command='/bin/true')
ext_serv.add_action(ac_suc)
group.add_dep(ext_serv)
group.run('start')
self.assertEqual(ext_serv.status, DONE)
self.assertEqual(group.status, MISSING)
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_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_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_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_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_group_with_weak_dep_error(self):
"""A group with a weak dep error runs fine."""
dep1 = Service('dep1')
dep1.add_action(Action('stop', command='/bin/false'))
grp = ServiceGroup('group')
grp.fromdict({
'services': {
'svc1': {
'actions': {
'stop': { 'cmd': "/bin/true" },
}
}
}
})
grp.add_dep(dep1, sgth=REQUIRE_WEAK)
grp.run('stop')
self.assertEqual(grp.status, DONE)
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_group_with_weak_dep_error(self):
"""A group with a weak dep error runs fine."""
dep1 = Service('dep1')
dep1.add_action(Action('stop', command='/bin/false'))
grp = ServiceGroup('group')
grp.fromdict({
'services': {
'svc1': {
'actions': {
'stop': {
'cmd': "/bin/true"
},
}
}
}
})
grp.add_dep(dep1, sgth=REQUIRE_WEAK)
grp.run('stop')
self.assertEqual(grp.status, DONE)
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_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_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_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)
