def test_delete_global_reserved_address(self, mock_service):
# Mock Discovery API
mock_service = mock.MagicMock()
# Import service.addresses().delete() response JSON data
with open('tests/fixtures/gce.operation.response.json') as json_file:
delete_ip = json.load(json_file)
# Import get-addresses response JSON data
with open('tests/fixtures/address-reserved-global.json') as json_file:
address = json.load(json_file)
# Map imported JSON payload to service.addresses().delete() return value
mock_service.globalAddresses.return_value.delete.return_value.execute.return_value = delete_ip
# Make call with mocked return values
response = delete_addresses(mock_service, self.project, address)
# Assertion (to test delete
self.assertEquals(response['operationType'], 'delete')
def test_delete_regional_inuse_router_address(self, mock_service):
# Mock Discovery API
mock_service = mock.MagicMock()
# Import service.routers().delete() response JSON data
with open('tests/fixtures/gce.operation.response.json') as json_file:
gce_operation = json.load(json_file)
# Map imported JSON payload to service.routers().delete() return value
mock_service.routers.return_value.delete.return_value.execute.return_value = gce_operation
mock_service.regionOperations.return_value.get.return_value.execute.return_value = gce_operation
# Import get-addresses response JSON data
with open('tests/fixtures/address-inuse-regional-router.json'
) as json_file:
address = json.load(json_file)
# Map imported JSON payload to service.addresses().delete() return value
mock_service.addresses.return_value.delete.return_value.execute.return_value = gce_operation
# Make call with mocked return values
response = delete_addresses(mock_service, self.project, address)
# Assertion (to test delete
self.assertEquals(response['operationType'], 'delete')