Unit test AMP pages (#249)

.gitignore

@@ -30,3 +30,4 @@ testproject*/
 testmedia*/
 teststatic*/
 *.sqlite3
+media/

coderedcms/models/tests/test_page_models.py

@@ -1,5 +1,6 @@
 from django.test import Client
 from wagtail.tests.utils import WagtailPageTests
+from wagtail.core.models import Site
 
 from coderedcms.models.page_models import (
     CoderedArticleIndexPage,
@@ -25,6 +26,9 @@ from coderedcms.tests.testapp.models import (
     StreamFormPage,
     WebPage
 )
+from coderedcms.models.wagtailsettings_models import (
+    SeoSettings
+)
 
 
 class BasicPageTestCase():
@@ -161,6 +165,15 @@ class CoderedStreamFormPageTestCase(AbstractPageTestCase, WagtailPageTests):
 class ArticlePageTestCase(ConcreteBasicPageTestCase, WagtailPageTests):
     model = ArticlePage
 
+    def test_amp(self):
+        site = Site.objects.filter(is_default_site=True)[0]
+        settings = SeoSettings.for_site(site)
+        settings.amp_pages = True
+        settings.save()
+
+        response = self.client.get(self.basic_page.url + '?amp')
+        self.assertEqual(response.status_code, 200)
+
 
 class ArticleIndexPageTestCase(ConcreteBasicPageTestCase, WagtailPageTests):
     model = ArticleIndexPage