def test_role_filter_with_valid_runlist(self): """Should return a role list when a valid run list is given""" role_list = filters.get_role_list(self.run_list) expected_roles = ['dbserver', 'webserver', 'worker', 'loadbalancer'] self.assertEqual(len(role_list), len(expected_roles)) for role in role_list: self.assertTrue(role in expected_roles) expected_roles.remove(role) self.assertEqual(len(expected_roles), 0)
def test_role_filter_with_runlist_and_exclude_node_prefix(self): """Should exclude roles with prefix when EXCLUDE_ROLE_PREFIX is set""" role_to_filter = REPO['EXCLUDE_ROLE_PREFIX'] + "_filterthisrole" run_list_with_excluded = self.run_list + [role_to_filter] role_list = filters.get_role_list(run_list_with_excluded) expected_roles = ['dbserver', 'webserver', 'worker', 'loadbalancer'] self.assertEqual(len(role_list), len(expected_roles)) for role in role_list: self.assertTrue(role in expected_roles) expected_roles.remove(role) self.assertEqual(len(expected_roles), 0)
def test_role_filter_with_wrong_runlist(self): """Should return an empty role list when an invalid run list is given """ role_list = filters.get_role_list(None) self.assertEqual(role_list, [])