mirror of
https://github.com/dnlbauer/django-signposting.git
synced 2026-09-10 22:15:30 +00:00
77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
from re import A
|
|
from django.http import HttpResponse
|
|
from django_signposting.middleware import SignpostingMiddleware
|
|
from signposting import LinkRel, Signpost
|
|
|
|
def test_middleware_no_signposting():
|
|
response = HttpResponse()
|
|
response.status_code = 200
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
assert "Link" not in response.headers
|
|
|
|
|
|
def test_middleware_signposting():
|
|
response = HttpResponse()
|
|
response.status_code = 200
|
|
response._signposts = [
|
|
Signpost(LinkRel.author, "http://example.com")
|
|
]
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
assert response.headers["Link"] == '<http://example.com> ; rel="author"'
|
|
|
|
|
|
def test_middleware_multiple_signposts():
|
|
response = HttpResponse()
|
|
response.status_code = 200
|
|
response._signposts = [
|
|
Signpost(LinkRel.author, "http://example.com"),
|
|
Signpost(LinkRel.author, "http://example2.com"),
|
|
Signpost(LinkRel.cite_as, "http://example3.com"),
|
|
]
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
links = [x.strip() for x in response.headers["Link"].split(",")]
|
|
assert '<http://example.com> ; rel="author"' in links
|
|
assert '<http://example2.com> ; rel="author"' in links
|
|
assert '<http://example3.com> ; rel="cite-as"' in links
|
|
|
|
|
|
def test_middleware_signpost_with_content_type():
|
|
response = HttpResponse()
|
|
response.status_code = 200
|
|
response._signposts = [
|
|
Signpost(LinkRel.item, "http://example.com", "text/json")
|
|
]
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
assert response.headers["Link"] == '<http://example.com> ; rel="item" ; type="text/json"'
|
|
|
|
|
|
def test_middleware_ignore_error_responses():
|
|
response = HttpResponse()
|
|
response.status_code = 400
|
|
response._signposts = [
|
|
Signpost(LinkRel.author, "http://example.com")
|
|
]
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
assert "Link" not in response.headers
|
|
|
|
|
|
def test_middleware_type_link():
|
|
response = HttpResponse()
|
|
response.status_code = 200
|
|
response._signposts = [
|
|
Signpost(LinkRel.type, "http://schema.org/Dataset")
|
|
]
|
|
|
|
middleware = SignpostingMiddleware(lambda request: response)
|
|
response = middleware(None)
|
|
assert response.headers["Link"] == '<http://schema.org/Dataset> ; rel="type"' |