def test_resp_headers(self):
oauth2.set_response(self.request, self.response, 200, {
"foo": "bar",
"banana": "ananab"
}, "")
self.assertEqual(self.response.headers["foo"], "bar")
self.assertEqual(self.response.headers["banana"], "ananab")
def test_resp_body_json_accept_json(self):
self.request.headers["Accept"] = "application/json"
oauth2.set_response(
self.request, self.response, 200,
{"Content-Type": "application/json; charset=utf-8"},
'{"foo": "bar", "bar": "foo"}')
self.assertIn("application/json", self.response["Content-Type"])
self.assertEqual(self.response.body, '{"foo": "bar", "bar": "foo"}')
def test_resp_body_string(self):
oauth2.set_response(
self.request, self.response, 200,
{"Content-Type": "text/html"},
"foobar"
)
self.assertEqual(self.response["Content-Type"], "text/html")
self.assertEqual(self.response.body, "foobar")
def test_resp_body_json_accept_all_multi_item(self):
self.request.headers["Accept"] = "*/*"
oauth2.set_response(
self.request, self.response, 200,
{"Content-Type": "application/json; charset=utf-8"},
'{"foo": "bar", "bar": 42}')
self.assertIn("application/x-www-form-urlencoded",
self.response["Content-Type"])
self.assertEqual(self.response.body, 'foo=bar&bar=42')
def test_resp_body_json_accept_all_encoded_item(self):
self.request.headers["Accept"] = "*/*"
oauth2.set_response(
self.request, self.response, 200,
{"Content-Type": "application/json; charset=utf-8"},
'{"foo": "bar", "bar": "http://foo?bar#fragment"}'
)
self.assertIn("application/x-www-form-urlencoded", self.response["Content-Type"])
self.assertEqual(self.response.body, 'foo=bar&bar=http%3A//foo%3Fbar%23fragment')
def test_wrong_body(self):
with self.assertRaises(
TypeError,
msg="must raise an error cuz body is not a valid string"):
oauth2.set_response(self.request, self.response, 200, {},
{"x": 42})
def test_no_body(self):
oauth2.set_response(self.request, self.response, 200, {}, None)
self.assertEqual(self.response.body, '')
def test_wrong_headers(self):
with self.assertRaises(
TypeError,
msg="must raise an error cuz headers is not a valid dict"):
oauth2.set_response(self.request, self.response, 200, "string",
"body")
def test_wrong_status(self):
with self.assertRaises(
ValueError,
msg="must raise an error cuz not a valid HTTP status_code"):
oauth2.set_response(self.request, self.response, {}, {}, "body")
