def test_do_resource_delete(self):
self._prepare_master_node()
expected_value = {
'kubernetes': {
'body': {
'payload_param': 'payload_value'
},
'first': 'second'
}
}
class _Result(object):
def to_dict(self):
return expected_value
class _DeleteResource(object):
def __call__(self, api_mapping, id, options):
if api_mapping == 'fake_api_mapping':
if id == 'fake_id':
if options['first'] == 'second':
return _Result()
client = MagicMock()
client.delete_resource = _DeleteResource()
result = tasks._do_resource_delete(client=client,
api_mapping='fake_api_mapping',
id='fake_id')
self.assertEqual(result, expected_value)
def test_external_do_resource_delete(self):
self._prepare_master_node(external=True)
expected_value = {
'kubernetes': {
'body': {
'payload_param': 'payload_value'
},
'first': 'second'
}
}
fake_resource_def = MagicMock()
setattr(fake_resource_def, 'metadata', {'name': 'name'})
class _Result(object):
def to_dict(self):
return expected_value
class _ReadResource(object):
def __call__(self, api_mapping, resource_definition, options):
if api_mapping == 'fake_api_mapping':
if resource_definition == 'fake_id':
if options['first'] == 'second':
return _Result()
client = MagicMock()
client.read_resource = _ReadResource()
result = tasks._do_resource_delete(
client=client,
api_mapping='fake_api_mapping',
resource_definition=fake_resource_def,
resource_id='fake_id')
self.assertEqual(result, expected_value)