This commit is contained in:
Matt Westcott 2015-09-03 12:00:47 +01:00
commit d85b0041dc
2 changed files with 9 additions and 1 deletions

View file

@ -1,6 +1,7 @@
from django import template
from wagtail.wagtailsearch.models import Query
from wagtail.contrib.wagtailsearchpromotions.models import SearchPromotion
register = template.Library()
@ -8,4 +9,7 @@ register = template.Library()
@register.assignment_tag()
def get_search_promotions(search_query):
return Query.get(search_query).editors_picks.all()
if search_query:
return Query.get(search_query).editors_picks.all()
else:
return SearchPromotion.objects.none()

View file

@ -71,6 +71,10 @@ class TestGetSearchPromotionsTemplateTag(TestCase):
search_picks = list(get_search_promotions("root page"))
self.assertEqual(search_picks, [pick])
def test_get_search_promotions_with_none_query_string(self):
search_picks = list(get_search_promotions(None))
self.assertEqual(search_picks, [])
class TestSearchPromotionsIndexView(TestCase, WagtailTestUtils):
def setUp(self):