def test_verify_no_content_type():
resp = FakeResponse('text/html')
del resp.headers['content-type']
assert util.verify_header(resp, 'txt') == 'txt'
def test_verify_header():
class FakeResponse():
def __init__(self, header):
self.headers = {"content-type": header}
self.text = "TEST_RESPONSE"
json_header = "application/json"
jwt_header = "application/jwt"
default_header = util.DEFAULT_POST_CONTENT_TYPE
plain_text_header = "text/plain"
undefined_header = "undefined"
assert util.verify_header(FakeResponse(json_header), "json") == "json"
assert util.verify_header(FakeResponse(jwt_header), "json") == "jwt"
assert util.verify_header(FakeResponse(jwt_header), "jwt") == "jwt"
assert util.verify_header(FakeResponse(default_header),
"urlencoded") == "urlencoded"
assert util.verify_header(FakeResponse(plain_text_header),
"urlencoded") == "urlencoded"
with pytest.raises(WrongContentType):
util.verify_header(FakeResponse(json_header), "urlencoded")
util.verify_header(FakeResponse(jwt_header), "urlencoded")
util.verify_header(FakeResponse(default_header), "json")
util.verify_header(FakeResponse(plain_text_header), "jwt")
util.verify_header(FakeResponse(undefined_header), "json")
with pytest.raises(ValueError):
util.verify_header(FakeResponse(json_header), "undefined")
def test_verify_header():
json_header = "application/json"
jwt_header = "application/jwt"
default_header = util.DEFAULT_POST_CONTENT_TYPE
plain_text_header = "text/plain"
undefined_header = "undefined"
assert util.verify_header(FakeResponse(json_header), "json") == "json"
assert util.verify_header(FakeResponse(jwt_header), "json") == "jwt"
assert util.verify_header(FakeResponse(jwt_header), "jwt") == "jwt"
assert util.verify_header(FakeResponse(default_header),
"urlencoded") == "urlencoded"
assert util.verify_header(FakeResponse(plain_text_header),
"urlencoded") == "urlencoded"
assert util.verify_header(FakeResponse('text/html'), 'txt')
assert util.verify_header(FakeResponse('text/plain'), 'txt')
assert util.verify_header(FakeResponse(json_header), "") == "json"
assert util.verify_header(FakeResponse(jwt_header), "") == "jwt"
assert util.verify_header(FakeResponse(jwt_header), "") == "jwt"
assert util.verify_header(FakeResponse(default_header), "") == "urlencoded"
assert util.verify_header(FakeResponse(plain_text_header), "") == "txt"
assert util.verify_header(FakeResponse('text/html'), '') == 'txt'
with pytest.raises(WrongContentType):
util.verify_header(FakeResponse(json_header), "urlencoded")
util.verify_header(FakeResponse(jwt_header), "urlencoded")
util.verify_header(FakeResponse(default_header), "json")
util.verify_header(FakeResponse(plain_text_header), "jwt")
util.verify_header(FakeResponse(undefined_header), "json")
with pytest.raises(ValueError):
util.verify_header(FakeResponse(json_header), "undefined")