def test_update_security_group_log_limit_success(self):
changed_security_group_info = dict(
copy.deepcopy(self.security_group_info),
log_limit = self.mockModule.params['log_limit'],
)
mock_describe_security_group = mock.MagicMock(
return_value=(
self.result['present'],
changed_security_group_info,
))
with mock.patch('niftycloud_fw.update_security_group_attribute', mock_describe_security_group):
(result, info) = niftycloud_fw.update_security_group_log_limit(
self.mockModule,
self.result['present'],
self.security_group_info
)
self.assertEqual(result, dict(
created = False,
changed_attributes = dict(
log_limit = self.mockModule.params['log_limit'],
),
state = 'present',
))
self.assertEqual(info, changed_security_group_info)
def test_update_security_group_log_limit_failed(self):
with mock.patch('niftycloud_fw.update_security_group_attribute', self.mockDescribeSecurityGroup):
with self.assertRaises(Exception) as cm:
(result, info) = niftycloud_fw.update_security_group_log_limit(
self.mockModule,
self.result['present'],
self.security_group_info
)
self.assertEqual(cm.exception.message, 'failed')
def test_update_security_group_log_limit_absent(self): (result, info) = niftycloud_fw.update_security_group_log_limit( self.mockModule, self.result['absent'], None ) self.assertEqual(result, self.result['absent']) self.assertIsNone(info)
def test_update_security_group_log_limit_skip(self): changed_security_group_info = dict( copy.deepcopy(self.security_group_info), log_limit = self.mockModule.params['log_limit'], ) (result, info) = niftycloud_fw.update_security_group_log_limit( self.mockModule, self.result['present'], changed_security_group_info ) self.assertEqual(result, self.result['present']) self.assertEqual(info, changed_security_group_info)
def test_update_security_group_log_limit_none(self):
security_group_info = dict(
copy.deepcopy(self.security_group_info),
log_limit = self.mockModule.params['description'],
)
mock_module = mock.MagicMock(
params = dict(
copy.deepcopy(self.mockModule.params),
log_limit = None,
)
)
(result, info) = niftycloud_fw.update_security_group_log_limit(
mock_module,
self.result['present'],
security_group_info
)
self.assertEqual(result, self.result['present'])
self.assertEqual(info, security_group_info)