From 56222f9259ba81730405b2ff35cdc0b04ded99bc Mon Sep 17 00:00:00 2001
From: Karl Hobley <karl@torchbox.com>
Date: Mon, 17 Feb 2014 10:52:52 +0000
Subject: [PATCH] Search: Frontend tests

---
 .../wagtailsearch/tests/test_editorspicks.py  | 63 +++++++++++++++++++
 wagtail/wagtailsearch/tests/test_frontend.py  | 30 +++++++++
 wagtail/wagtailsearch/tests/test_queries.py   | 12 ++++
 wagtail/wagtailsearch/tests/utils.py          | 14 +++++
 4 files changed, 119 insertions(+)
 create mode 100644 wagtail/wagtailsearch/tests/test_frontend.py
 create mode 100644 wagtail/wagtailsearch/tests/utils.py

diff --git a/wagtail/wagtailsearch/tests/test_editorspicks.py b/wagtail/wagtailsearch/tests/test_editorspicks.py
index 31965e26f5..64e90bc87a 100644
--- a/wagtail/wagtailsearch/tests/test_editorspicks.py
+++ b/wagtail/wagtailsearch/tests/test_editorspicks.py
@@ -1,5 +1,68 @@
 from django.test import TestCase
+from .utils import get_default_host, login
+from wagtail.wagtailsearch import models
 
 
 class TestEditorsPicks(TestCase):
     pass
+
+
+class TestEditorsPicksIndexView(TestCase):
+    def setUp(self):
+        login(self.client)
+
+    def get(self, params={}):
+        return self.client.get('/admin/search/editorspicks/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
+
+    def test_search_status_code(self):
+        self.assertEqual(self.get({'q': "Hello"}).status_code, 200)
+
+    def test_search_template_context(self):
+        self.assertEqual(self.get({'q': "Hello"}).context['search_query'], "Hello")
+
+    def test_page_status_code(self):
+        self.assertEqual(self.get({'p': '1'}).status_code, 200)
+
+
+class TestEditorsPicksAddView(TestCase):
+    def setUp(self):
+        login(self.client)
+
+    def get(self, params={}):
+        return self.client.get('/admin/search/editorspicks/add/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
+
+
+class TestEditorsPicksEditView(TestCase):
+    def setUp(self):
+        login(self.client)
+
+        # Create an editors pick to edit
+        self.query = models.Query.get("Hello")
+        self.query.editors_picks.create(page_id=1, description="Root page")
+
+    def get(self, params={}):
+        return self.client.get('/admin/search/editorspicks/' + str(self.query.id) + '/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
+
+
+class TestEditorsPicksDeleteView(TestCase):
+    def setUp(self):
+        login(self.client)
+
+        # Create an editors pick to delete
+        self.query = models.Query.get("Hello")
+        self.query.editors_picks.create(page_id=1, description="Root page")
+
+    def get(self, params={}):
+        return self.client.get('/admin/search/editorspicks/' + str(self.query.id) + '/delete/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
diff --git a/wagtail/wagtailsearch/tests/test_frontend.py b/wagtail/wagtailsearch/tests/test_frontend.py
new file mode 100644
index 0000000000..7ac103004c
--- /dev/null
+++ b/wagtail/wagtailsearch/tests/test_frontend.py
@@ -0,0 +1,30 @@
+from django.test import TestCase
+from .utils import get_default_host
+
+
+class TestSearchView(TestCase):
+    def get(self, params={}):
+        return self.client.get('/search/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
+
+    def test_search_status_code(self):
+        self.assertEqual(self.get({'q': "Hello"}).status_code, 200)
+
+    def test_search_template_context(self):
+        self.assertEqual(self.get({'q': "Hello"}).context['query_string'], "Hello")
+
+    def test_page_status_code(self):
+        self.assertEqual(self.get({'p': '1'}).status_code, 200)
+
+
+class TestSuggestionsView(TestCase):
+    def get(self, params={}):
+        return self.client.get('/search/suggest/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
+
+    def test_search_status_code(self):
+        self.assertEqual(self.get({'q': "Hello"}).status_code, 200)
diff --git a/wagtail/wagtailsearch/tests/test_queries.py b/wagtail/wagtailsearch/tests/test_queries.py
index 6acfd7e135..fef6c94399 100644
--- a/wagtail/wagtailsearch/tests/test_queries.py
+++ b/wagtail/wagtailsearch/tests/test_queries.py
@@ -1,6 +1,7 @@
 from django.test import TestCase
 from django.core import management
 from wagtail.wagtailsearch import models
+from .utils import get_default_host, login
 from StringIO import StringIO
 import unittest
 
@@ -137,3 +138,14 @@ class TestGarbageCollectCommand(TestCase):
         management.call_command('search_garbage_collect', interactive=False, stdout=StringIO())
 
     # TODO: Test that this command is acctually doing its job
+
+
+class TestQueryChooserView(TestCase):
+    def setUp(self):
+        login(self.client)
+
+    def get(self, params={}):
+        return self.client.get('/admin/search/queries/chooser/', params, HTTP_HOST=get_default_host())
+
+    def test_status_code(self):
+        self.assertEqual(self.get().status_code, 200)
diff --git a/wagtail/wagtailsearch/tests/utils.py b/wagtail/wagtailsearch/tests/utils.py
new file mode 100644
index 0000000000..05a122f99f
--- /dev/null
+++ b/wagtail/wagtailsearch/tests/utils.py
@@ -0,0 +1,14 @@
+from wagtail.wagtailcore.models import Site
+from django.contrib.auth.models import User
+
+
+def get_default_host():
+    return Site.objects.filter(is_default_site=True).first().root_url.split('://')[1]
+
+
+def login(client):
+    # Create a user
+    User.objects.create_superuser(username='test', email='test@email.com', password='password')
+
+    # Login
+    client.login(username='test', password='password')