def test_api_get_fail(self, mock_login, mock_get, mock_headers): """ Test ._api_get() 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_get.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_get( url=url, data={'k': 'v'} ) mock_get.assert_called_once_with( alert.session, url=url, headers={}, data={'k': 'v'} )
def test_get_api_headers(self, mock_login, mock_get_token): """ Test .get_api_headers() """ mock_login.return_value = self.account_id mock_get_token.return_value = self.csrf_token alert = SpecialAlertBase(self.username, self.password) self.assertDictEqual( alert._get_api_headers(**{'xxx': 'yyy'}), { 'Accept': 'application/json, text/plain, */*', 'Accept-Language': 'en-US,en;q=0.5', 'Cache-Control': 'no-cache', 'Connection': 'keep-alive', 'Content-Type': 'application/json;charset=utf-8', 'Host': 'logentries.com', 'Pragma': 'no-cache', 'Referer': 'https://logentries.com/app/{account_id}'.format(account_id=self.account_id), 'X-CSRFToken': self.csrf_token, 'xxx': 'yyy' } ) mock_get_token.assert_called_once_with()
def test_api_get(self, mock_login, mock_get, mock_headers): """ Test ._api_get() """ 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_get( url=url, data={'k': 'v'} ) mock_get.assert_called_once_with( alert.session, url=url, headers={}, data={'k': 'v'} )