Python SITLoader.stack_created_successfully Examples

Example #1

class SITLoaderTest(unittest.TestCase):
    def setUp(self):
        self.session = Session(region_name='us-west-1')
        self.configs_directory = 'tests/sit/configs'
        self.sit_loader = SITLoader(session=self.session,
                                    configs_directory=self.configs_directory)
        self.mock_cf_helper_true_responses()

    def mock_cf_helper_true_responses(self):
        cf_helper = CFHelper(configs_directory=self.configs_directory,
                             session=self.session)
        cf_helper.get_stack_info = MagicMock(return_value=False)
        cf_helper.validate_template = MagicMock(return_value=True)
        cf_helper.create_or_update_stack = MagicMock(return_value=True)
        cf_helper.create_stack = MagicMock(return_value=True)
        cf_helper.stack_was_created_successfully = MagicMock(return_value=True)
        self.sit_loader.cf_helper = cf_helper

    def test_successful_run(self):
        self.sit_loader.run()

    @raises(SystemExit)
    def test_failed_run(self):
        self.sit_loader.cf_helper.stack_was_created_successfully = MagicMock(
            return_value=False)
        self.assertRaises(self.sit_loader.stack_created_successfully(),
                          Exception)

Example #2

File: sit_loader_test.py Project: banderon1/salt-integration-testing

class SITLoaderTest(unittest.TestCase):

    def setUp(self):
        self.session = Session(region_name='us-west-1')
        self.configs_directory = 'tests/sit/configs'
        self.sit_loader = SITLoader(session=self.session, configs_directory=self.configs_directory)
        self.mock_cf_helper_true_responses()

    def mock_cf_helper_true_responses(self):
        cf_helper = CFHelper(configs_directory=self.configs_directory, session=self.session)
        cf_helper.get_stack_info = MagicMock(return_value=False)
        cf_helper.validate_template = MagicMock(return_value=True)
        cf_helper.create_or_update_stack = MagicMock(return_value=True)
        cf_helper.create_stack = MagicMock(return_value=True)
        cf_helper.stack_was_created_successfully = MagicMock(return_value=True)
        self.sit_loader.cf_helper = cf_helper

    def test_successful_run(self):
        self.sit_loader.run()

    @raises(SystemExit)
    def test_failed_run(self):
        self.sit_loader.cf_helper.stack_was_created_successfully = MagicMock(return_value=False)
        self.assertRaises(self.sit_loader.stack_created_successfully(), Exception)