70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import json
|
|
import time
|
|
from httmock import urlmatch, HTTMock
|
|
|
|
from hobo.context_processors import theme_base
|
|
|
|
import django
|
|
from django.core.cache import cache
|
|
from django.test import override_settings
|
|
|
|
|
|
def test_theme_base(settings, rf):
|
|
settings.THEME_SKELETON_URL = 'http://combo.example.com/_skeleton_/'
|
|
seen_urls = []
|
|
TEMPLATE = 'Feeling lucky, punk?'
|
|
TEMPLATE_PAGE2 = 'Response for page 2'
|
|
|
|
@urlmatch(netloc=r'combo.example.com$')
|
|
def combo_mock(url, request):
|
|
seen_urls.append(url.geturl())
|
|
status_code = 200
|
|
content = TEMPLATE
|
|
if 'page2' in url.query:
|
|
content = TEMPLATE_PAGE2
|
|
elif 'page1' in url.query:
|
|
status_code = 500
|
|
content = 'No template sorry'
|
|
return {
|
|
'status_code': status_code,
|
|
'content': content,
|
|
'headers': {'X-Combo-Skeleton-Pages': json.dumps(
|
|
{'1': 'http://testserver/foo',
|
|
'2': 'http://testserver/page1/page2/'})},
|
|
}
|
|
|
|
cache.clear()
|
|
|
|
def check(context, value):
|
|
if django.VERSION >= (1, 10, 0):
|
|
assert context['theme_base']().source == value
|
|
else:
|
|
assert context['theme_base']().origin.source == value
|
|
|
|
with HTTMock(combo_mock), override_settings(INSTALLED_APPS=[]):
|
|
context = theme_base(rf.get('/'))
|
|
check(context, TEMPLATE)
|
|
assert seen_urls[0] == 'http://combo.example.com/_skeleton_/?source=http%3A%2F%2Ftestserver%2F'
|
|
for i in range(10):
|
|
# wait for the other requests, made from a thread, to happen
|
|
time.sleep(0.1)
|
|
if len(seen_urls) == 4:
|
|
break
|
|
assert len(seen_urls) == 4
|
|
# requested page + root + pages from X-Combo-Skeleton-Pages header
|
|
seen_urls = []
|
|
context = theme_base(rf.get('/'))
|
|
check(context, TEMPLATE)
|
|
assert len(seen_urls) == 0
|
|
seen_urls = []
|
|
context = theme_base(rf.get('/page1/page2/'))
|
|
check(context, TEMPLATE_PAGE2)
|
|
assert len(seen_urls) == 0
|
|
seen_urls = []
|
|
context = theme_base(rf.get('/page1/'))
|
|
check(context, TEMPLATE)
|
|
assert len(seen_urls) == 0
|
|
context = theme_base(rf.get('/page1/page2/'))
|
|
check(context, TEMPLATE_PAGE2)
|
|
assert len(seen_urls) == 0
|