Пример #1
0
    def test_refresh_token(self, mock_request):
        mock_request.return_value = 'new_token'

        web = WebObject()
        web.refresh_token()

        url = 'https://www.yikyak.com/api/auth/token/refresh'
        mock_request.assert_called_with('POST', url)

        self.assertEqual(web.session.headers['x-access-token'], 'new_token')
Пример #2
0
    def test__request(self):
        """Assert ._request uses session object to make requests"""
        web = WebObject()
        mock_response = mock.Mock()
        mock_response.json.return_value = {}
        web.session = mock.Mock()
        web.session.request.return_value = mock_response

        web._request('GET', 'http://yikyak.com')
        web.session.request.assert_called_with('GET', 'http://yikyak.com')
Пример #3
0
    def test_request_invalid_json(self):
        """
        Assert that _request() will still work if a JSONDecodeError occurs

        Exception should result in an empty dict being returned
        """
        # Mock response object
        mock_response = mock.Mock()
        mock_response.json.side_effect = json.decoder.JSONDecodeError('', '', 0)

        web = WebObject()
        web.session = mock.Mock()
        web.session.request.return_value = mock_response

        response = web._request('', '')
        self.assertEqual(response, {})