Add unit tests for wagtailforms backend

wagtail/tests/fixtures/test.json

@@ -229,6 +229,19 @@
         ]
     }
 },
+{
+    "pk": 5,
+    "model": "auth.group",
+    "fields": {
+        "name": "Site-wide editors",
+        "permissions": [
+            ["access_admin", "wagtailadmin", "admin"],
+            ["add_image", "wagtailimages", "image"],
+            ["change_image", "wagtailimages", "image"],
+            ["delete_image", "wagtailimages", "image"]
+        ]
+    }
+},
 {
     "pk": 1,
     "model": "wagtailcore.grouppagepermission",
@@ -265,6 +278,15 @@
         "permission_type": "publish"
     }
 },
+{
+    "pk": 5,
+    "model": "wagtailcore.grouppagepermission",
+    "fields": {
+        "group": ["Site-wide editors"],
+        "page": 2,
+        "permission_type": "edit"
+    }
+},
 
 {
     "pk": 1,
@@ -336,5 +358,42 @@
         "password": "md5$seasalt$1e9bf2bf5606aa5c39852cc30f0f6f22",
         "email": "inactiveuser@example.com"
     }
+},
+{
+    "pk": 5,
+    "model": "auth.user",
+    "fields": {
+        "username": "siteeditor",
+        "first_name": "",
+        "last_name": "",
+        "is_active": true,
+        "is_superuser": false,
+        "is_staff": false,
+        "groups": [
+            ["Site-wide editors"]
+        ],
+        "user_permissions": [],
+        "password": "md5$seasalt$1e9bf2bf5606aa5c39852cc30f0f6f22",
+        "email": "siteeditor@example.com"
+    }
+},
+
+{
+    "pk": 1,
+    "model": "wagtailforms.formsubmission",
+    "fields": {
+        "form_data": "{\"your-email\": \"old@example.com\", \"your-message\": \"this is a really old message\"}",
+        "page": 7,
+        "submit_time": "2013-01-01T12:00:00.000Z"
+    }
+},
+{
+    "pk": 2,
+    "model": "wagtailforms.formsubmission",
+    "fields": {
+        "form_data": "{\"your-email\": \"new@example.com\", \"your-message\": \"this is a fairly new message\"}",
+        "page": 7,
+        "submit_time": "2014-01-01T12:00:00.000Z"
+    }
 }
 ]

wagtail/wagtailforms/tests.py

@@ -28,3 +28,36 @@ class TestFormSubmission(TestCase):
         form_page = Page.objects.get(url_path='/home/contact-us/')
 
         self.assertTrue(FormSubmission.objects.filter(page=form_page, form_data__contains='hello world').exists())
+
+
+class TestFormsBackend(TestCase):
+    fixtures = ['test.json']
+
+    def test_cannot_see_forms_without_permission(self):
+        form_page = Page.objects.get(url_path='/home/contact-us/')
+
+        self.client.login(username='eventeditor', password='password')
+        response = self.client.get('/admin/forms/')
+        self.assertFalse(response.context['form_pages'].filter(id=form_page.id).exists())
+
+    def test_can_see_forms_with_permission(self):
+        form_page = Page.objects.get(url_path='/home/contact-us/')
+
+        self.client.login(username='siteeditor', password='password')
+        response = self.client.get('/admin/forms/')
+        self.assertTrue(response.context['form_pages'].filter(id=form_page.id).exists())
+
+    def test_can_get_submissions(self):
+        form_page = Page.objects.get(url_path='/home/contact-us/')
+
+        self.client.login(username='siteeditor', password='password')
+
+        response = self.client.get('/admin/forms/submissions/%d/' % form_page.id)
+        self.assertEqual(len(response.context['data_rows']), 2)
+
+        response = self.client.get('/admin/forms/submissions/%d/?date_from=01%%2F01%%2F2014' % form_page.id)
+        self.assertEqual(len(response.context['data_rows']), 1)
+
+        response = self.client.get('/admin/forms/submissions/%d/?date_from=01%%2F01%%2F2014&action=CSV' % form_page.id)
+        data_line = response.content.split("\n")[1]
+        self.assertTrue('new@example.com' in data_line)