def test_api_delete_fail(self, mock_login, mock_delete, mock_headers): """ Test ._api_delete() raises an exception """ mock_login.return_value = self.account_id mock_headers.return_value = {} mock_response = Mock(spec=requests.Response) mock_response.status_code = 500 mock_response.text = "Server error" mock_response.ok = False mock_delete.return_value = mock_response url = 'https://logentries.com/app/{account_id}/rest/tag'.format(account_id=self.account_id) alert = SpecialAlertBase(self.username, self.password) with self.assertRaises(ServerException): alert._api_delete( url=url, data={'k': 'v'} ) mock_delete.assert_called_once_with( alert.session, url=url, headers={}, data={'k': 'v'} )
def test_api_delete(self, mock_login, mock_delete, mock_headers): """ Test ._api_delete() """ mock_login.return_value = self.account_id mock_headers.return_value = {} url = 'https://logentries.com/app/{account_id}/rest/tag'.format(account_id=self.account_id) alert = SpecialAlertBase(self.username, self.password) alert._api_delete( url=url, data={'k': 'v'} ) mock_delete.assert_called_once_with( alert.session, url=url, headers={}, data={'k': 'v'} )