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')