class SitHelperTest(unittest.TestCase):
    def setUp(self):
        self.sit_helper = SITHelper('tests/helpers/configs')
        self.sit_helper_empty = SITHelper('tests/helpers/empty_configs')

    def test_get_custom_user_data(self):
        self.assertEquals(self.sit_helper.get_custom_user_data(), "'test'\n")

    def test_get_roles(self):
        self.assertEquals(self.sit_helper.get_roles(), ['lb', 'php'])
        self.assertEquals(self.sit_helper.get_roles(), ['lb', 'php'])
        self.assertEquals(self.sit_helper_empty.get_roles(), False)

    def test_get_states_for_role(self):
        self.assertEquals(self.sit_helper.get_states_for_role('php'),
                          ['apache', 'server'])
        self.assertEquals(self.sit_helper.get_states_for_role('lb'),
                          ['lb', 'server'])
        self.assertEquals(
            self.sit_helper.get_states_for_role('saltmaster'),
            "Failed to find state list for role: saltmaster. error: 'saltmaster'"
        )

    def test_get_configs(self):
        self.assertEquals(
            self.sit_helper.get_configs('roles'), {
                'lb': {
                    'priority': 1,
                    'subroles': ['lb', 'server']
                },
                'php': {
                    'priority': 2,
                    'subroles': ['apache', 'server']
                }
            })
class SitHelperTest(unittest.TestCase):

    def setUp(self):
        self.sit_helper = SITHelper('tests/helpers/configs')
        self.sit_helper_empty = SITHelper('tests/helpers/empty_configs')

    def test_get_custom_user_data(self):
        self.assertEquals(self.sit_helper.get_custom_user_data(), "'test'\n")

    def test_get_roles(self):
        self.assertEquals(self.sit_helper.get_roles(), ['php', 'lb'])
        self.assertEquals(self.sit_helper.get_roles(), ['php', 'lb'])
        self.assertEquals(self.sit_helper_empty.get_roles(), False)

    def test_get_states_for_role(self):
        self.assertEquals(self.sit_helper.get_states_for_role('php'), ['apache', 'server'])
        self.assertEquals(self.sit_helper.get_states_for_role('lb'), ['lb', 'server'])
        self.assertEquals(self.sit_helper.get_states_for_role('saltmaster'),
                          "Failed to find state list for role: saltmaster. error: 'saltmaster'")

    def test_get_configs(self):
        self.assertEquals(self.sit_helper.get_configs('roles'), {'php': ['apache', 'server'], 'lb': ['lb', 'server']})
 def __init__(self, configs_directory='tests/sit/configs'):
     sit_helper = SITHelper(configs_directory)
     CONFIGS = sit_helper.get_configs('troposphere')
     self.LAUNCH_CONFIGURATION_NAME = CONFIGS['launch_configuration_name']
     self.AUTOSCALING_GROUP_NAME = CONFIGS['autoscaling_group_name']
     self.CUSTOM_USER_DATA = sit_helper.get_custom_user_data()
예제 #4
0
 def __init__(self, configs_directory="tests/sit/configs"):
     sit_helper = SITHelper(configs_directory)
     CONFIGS = sit_helper.get_configs("troposphere")
     self.LAUNCH_CONFIGURATION_NAME = CONFIGS["launch_configuration_name"]
     self.AUTOSCALING_GROUP_NAME = CONFIGS["autoscaling_group_name"]
     self.CUSTOM_USER_DATA = sit_helper.get_custom_user_data()