def test_fetch_request_token_via_header():
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
assert 'oauth_consumer_key="id"' in auth_header
assert 'oauth_signature=' in auth_header
app = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client('id', 'secret', app=app) as client:
response = client.fetch_request_token(oauth_url)
assert response == request_token
def test_fetch_request_token_via_header(self):
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
self.assertIn('oauth_consumer_key="id"', auth_header)
self.assertIn('oauth_signature=', auth_header)
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client('id', 'secret', dispatch=mock_response) as client:
response = client.fetch_request_token(self.oauth_url)
self.assertEqual(response, request_token)
def test_get_via_header(self):
mock_response = MockDispatch(b'hello')
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
dispatch=mock_response,
) as client:
response = client.get('https://example.com/')
self.assertEqual(b'hello', response.content)
request = response.request
auth_header = request.headers.get('authorization')
self.assertIn('oauth_token="foo"', auth_header)
self.assertIn('oauth_consumer_key="id"', auth_header)
self.assertIn('oauth_signature=', auth_header)
def test_get_via_header():
mock_response = MockDispatch(b'hello')
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
app=mock_response,
) as client:
response = client.get('https://example.com/')
assert response.content == b'hello'
request = response.request
auth_header = request.headers.get('authorization')
assert 'oauth_token="foo"' in auth_header
assert 'oauth_consumer_key="id"' in auth_header
assert 'oauth_signature=' in auth_header
def test_get_via_body(self):
mock_response = MockDispatch(b'hello')
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
signature_type=SIGNATURE_TYPE_BODY,
dispatch=mock_response,
) as client:
response = client.post('https://example.com/')
self.assertEqual(b'hello', response.content)
request = response.request
auth_header = request.headers.get('authorization')
self.assertIsNone(auth_header)
self.assertIn(b'oauth_token=foo', request.content)
self.assertIn(b'oauth_consumer_key=id', request.content)
self.assertIn(b'oauth_signature=', request.content)
def test_fetch_request_token_via_body(self):
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
self.assertIsNone(auth_header)
self.assertIn(b'oauth_consumer_key=id', request.content)
self.assertIn(b'&oauth_signature=', request.content)
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
signature_type=SIGNATURE_TYPE_BODY,
dispatch=mock_response,
) as client:
response = client.fetch_request_token(self.oauth_url)
self.assertEqual(response, request_token)
def test_get_via_query():
mock_response = MockDispatch(b'hello')
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
signature_type=SIGNATURE_TYPE_QUERY,
app=mock_response,
) as client:
response = client.get('https://example.com/')
assert response.content == b'hello'
request = response.request
auth_header = request.headers.get('authorization')
assert auth_header is None
url = str(request.url)
assert 'oauth_token=foo' in url
assert 'oauth_consumer_key=id' in url
assert 'oauth_signature=' in url
def test_fetch_request_token_via_query():
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
assert auth_header is None
url = str(request.url)
assert 'oauth_consumer_key=id' in url
assert '&oauth_signature=' in url
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
signature_type=SIGNATURE_TYPE_QUERY,
app=mock_response,
) as client:
response = client.fetch_request_token(oauth_url)
assert response == request_token
def test_fetch_request_token_via_body():
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
assert auth_header is None
content = request.form
assert content.get('oauth_consumer_key') == 'id'
assert 'oauth_signature' in content
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
signature_type=SIGNATURE_TYPE_BODY,
app=mock_response,
) as client:
response = client.fetch_request_token(oauth_url)
assert response == request_token
def test_fetch_access_token(self):
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
self.assertIn('oauth_verifier="d"', auth_header)
self.assertIn('oauth_token="foo"', auth_header)
self.assertIn('oauth_consumer_key="id"', auth_header)
self.assertIn('oauth_signature=', auth_header)
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
dispatch=mock_response,
) as client:
self.assertRaises(OAuthError, client.fetch_access_token,
self.oauth_url)
response = client.fetch_access_token(self.oauth_url, verifier='d')
self.assertEqual(response, request_token)
def test_get_via_body():
def assert_func(request):
content = request.form
assert content.get('oauth_token') == 'foo'
assert content.get('oauth_consumer_key') == 'id'
assert 'oauth_signature' in content
mock_response = MockDispatch(b'hello', assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
signature_type=SIGNATURE_TYPE_BODY,
app=mock_response,
) as client:
response = client.post('https://example.com/')
assert response.content == b'hello'
request = response.request
auth_header = request.headers.get('authorization')
assert auth_header is None
def test_fetch_access_token():
request_token = {'oauth_token': '1', 'oauth_token_secret': '2'}
def assert_func(request):
auth_header = request.headers.get('authorization')
assert 'oauth_verifier="d"' in auth_header
assert 'oauth_token="foo"' in auth_header
assert 'oauth_consumer_key="id"' in auth_header
assert 'oauth_signature=' in auth_header
mock_response = MockDispatch(request_token, assert_func=assert_func)
with OAuth1Client(
'id',
'secret',
token='foo',
token_secret='bar',
app=mock_response,
) as client:
with pytest.raises(OAuthError):
client.fetch_access_token(oauth_url)
response = client.fetch_access_token(oauth_url, verifier='d')
assert response == request_token