def test_extract_oauth_key_from_auth_header_returns_key(self):
token = factory.make_string(18)
self.assertEqual(
token,
extract_oauth_key_from_auth_header(
factory.make_oauth_header(oauth_token=token)),
)
def test_get_queried_node_looks_up_oauth_key_by_default(self):
node = factory.make_node()
token = NodeKey.objects.get_token_for_node(node)
request = self.fake_request(
HTTP_AUTHORIZATION=factory.make_oauth_header(
oauth_token=token.key))
self.assertEqual(node, get_queried_node(request))
def test_extract_oauth_key_from_auth_header_no_whitespace_rtns_key(self):
token = factory.make_string(18)
auth_header = factory.make_oauth_header(oauth_token=token)
auth_header = auth_header.replace(", ", ",")
self.assertEqual(
token, extract_oauth_key_from_auth_header(auth_header)
)
def test_get_node_for_request_finds_node(self):
node = factory.make_node()
token = NodeKey.objects.get_token_for_node(node)
request = self.fake_request(
HTTP_AUTHORIZATION=factory.make_oauth_header(
oauth_token=token.key))
self.assertEqual(node, get_node_for_request(request))
def test_get_node_for_request_finds_node(self):
node = factory.make_node()
token = NodeKey.objects.get_token_for_node(node)
request = self.fake_request(
HTTP_AUTHORIZATION=factory.make_oauth_header(
oauth_token=token.key))
self.assertEqual(node, get_node_for_request(request))
def test_get_queried_node_looks_up_oauth_key_by_default(self):
node = factory.make_node()
token = NodeKey.objects.get_token_for_node(node)
request = self.fake_request(
HTTP_AUTHORIZATION=factory.make_oauth_header(
oauth_token=token.key))
self.assertEqual(node, get_queried_node(request))
def test_extract_oauth_key_returns_key(self):
token = factory.getRandomString(18)
self.assertEqual(
token,
extract_oauth_key(
self.make_fake_request(
factory.make_oauth_header(oauth_token=token))))
def test_get_oauth_token_finds_token(self):
user = factory.make_user()
consumer, token = user.get_profile().create_authorisation_token()
self.assertEqual(
token,
get_oauth_token(
self.make_fake_request(
factory.make_oauth_header(oauth_token=token.key))))
def test_get_oauth_token_finds_token(self):
user = factory.make_user()
consumer, token = user.get_profile().create_authorisation_token()
self.assertEqual(
token,
get_oauth_token(
self.make_fake_request(
factory.make_oauth_header(oauth_token=token.key))))
def make_request(env=None, oauth_env=None, missing_oauth_param=None):
# Return a minimal WSGIRequest.
if oauth_env is None:
oauth_env = {}
base_env = {
"REQUEST_METHOD":
"GET",
"wsgi.input":
wsgi._InputStream(io.BytesIO()),
"SERVER_NAME":
"server",
"SERVER_PORT":
80,
"HTTP_AUTHORIZATION":
factory.make_oauth_header(missing_param=missing_oauth_param,
**oauth_env),
}
if env is not None:
base_env.update(env)
request = WSGIRequest(base_env)
return request
def test_extract_oauth_key_from_auth_header_returns_key(self):
token = factory.getRandomString(18)
self.assertEqual(
token,
extract_oauth_key_from_auth_header(
factory.make_oauth_header(oauth_token=token)))
def test_get_oauth_token_raises_Unauthorized_for_unknown_token(self):
fake_token = factory.getRandomString(18)
header = factory.make_oauth_header(oauth_token=fake_token)
self.assertRaises(
Unauthorized,
get_oauth_token, self.make_fake_request(header))
def test_get_oauth_token_raises_Unauthorized_for_unknown_token(self):
fake_token = factory.make_string(18)
header = factory.make_oauth_header(oauth_token=fake_token)
self.assertRaises(Unauthorized, get_oauth_token,
make_fake_request(header))
