Tests for template tag get_back_button

oscar/templatetags/history_tags.py

@@ -46,19 +46,19 @@ def get_back_button(context):
 
     referrer = request.META.get('HTTP_REFERER', None)
     if not referrer:
-        return False
+        return None
 
     try:
         url = urlparse.urlparse(referrer)
     except:
-        return False
+        return None
 
     if request.get_host() != url.netloc:
         try:
             Site.objects.get(domain=url.netloc)
         except Site.DoesNotExist:
             # Came from somewhere else, don't show back button:
-            return False
+            return None
 
     try:
         match = resolve(url.path)

tests/functional/customer/history_tests.py

@@ -5,6 +5,7 @@
 
 from oscar_testsupport.factories import create_product
 from oscar.apps.customer.history_helpers import get_recently_viewed_product_ids
+from oscar.templatetags.history_tags import get_back_button
 from django.http import HttpRequest
 
 
@@ -24,6 +25,20 @@ def test_id_gets_added_to_cookie(self):
         request.COOKIES['oscar_recently_viewed_products'] = response.cookies['oscar_recently_viewed_products'].value
         self.assertTrue(self.product.id in get_recently_viewed_product_ids(request))
 
+    def test_get_back_button(self):
+        request = HttpRequest()
+
+        request.META['SERVER_NAME'] = 'test'
+        request.META['SERVER_PORT'] = 8000
+        request.META['HTTP_REFERER'] = 'http://www.google.com'
+        backbutton = get_back_button({'request': request})
+        self.assertEqual(backbutton, None)
+
+        request.META['HTTP_REFERER'] = 'http://test:8000/search/'
+        backbutton = get_back_button({'request': request})
+        self.assertTrue(backbutton)
+        self.assertEqual(backbutton['title'], 'Back to search results')
+
 
 class TestAUserWhoLogsOut(TestCase):
     username = 'customer'