Add tests for admin

Author: gasman

experiments/utils.py

@@ -10,5 +10,5 @@ def get_user_id(request):
 def percentage(fraction, population):
     try:
         return float(fraction) / float(population) * 100
-    except (ValueError, ZeroDivisionError):
+    except (ValueError, ZeroDivisionError, TypeError):
         return 0.0

tests/settings.py

@@ -22,6 +22,8 @@
 
 ROOT_URLCONF = 'tests.urls'
 
+STATIC_URL = '/static/'
+
 STATICFILES_FINDERS = (
     'django.contrib.staticfiles.finders.AppDirectoriesFinder',
 )
@@ -97,3 +99,5 @@
 PASSWORD_HASHERS = (
     'django.contrib.auth.hashers.MD5PasswordHasher',  # don't use the intentionally slow default password hasher
 )
+
+WAGTAIL_SITE_NAME = 'wagtail-experiments test'

tests/tests.py

@@ -1,5 +1,7 @@
 from __future__ import absolute_import, unicode_literals
 
+from django.contrib.auth.models import User
+from django.core.urlresolvers import reverse
 from django.test import TestCase
 from wagtail.wagtailcore.models import Page
 
@@ -239,3 +241,62 @@ def test_completed_status(self):
 
         self.assertEqual(response.status_code, 200)
         self.assertContains(response, "<p>Oh, it&#39;s you. What do you want?</p>")
+
+
+class TestAdmin(TestCase):
+    fixtures = ['test.json']
+
+    def setUp(self):
+        User.objects.create_superuser(username='admin', email='[email protected]', password='password')
+        self.assertTrue(
+            self.client.login(username='admin', password='password')
+        )
+        self.experiment = Experiment.objects.get(slug='homepage-text')
+        self.homepage_alternative_1 = Page.objects.get(url_path='/home/home-alternative-1/')
+
+    def test_experiments_menu_item(self):
+        response = self.client.get(reverse('wagtailadmin_home'))
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, 'href="/admin/experiments/experiment/"')
+
+    def test_experiments_index(self):
+        response = self.client.get('/admin/experiments/experiment/')
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, 'Homepage text')
+
+    def test_experiment_edit(self):
+        response = self.client.get('/admin/experiments/experiment/edit/%d/' % self.experiment.pk)
+        self.assertEqual(response.status_code, 200)
+
+    def test_experiment_delete(self):
+        response = self.client.get('/admin/experiments/experiment/delete/%d/' % self.experiment.pk)
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, "Are you sure you want to delete this experiment?")
+
+        response = self.client.post('/admin/experiments/experiment/delete/%d/' % self.experiment.pk)
+        self.assertRedirects(response, '/admin/experiments/experiment/')
+        self.assertFalse(Experiment.objects.filter(slug='homepage-text').exists())
+
+    def test_show_report(self):
+        response = self.client.get('/admin/experiments/experiment/report/%d/' % self.experiment.pk)
+        self.assertEqual(response.status_code, 200)
+
+    def test_select_winner(self):
+        response = self.client.post(
+            '/admin/experiments/experiment/select_winner/%d/%d/' % (
+                self.experiment.pk, self.homepage_alternative_1.pk
+            )
+        )
+        self.assertRedirects(
+            response,
+            '/admin/experiments/experiment/report/%d/' % self.experiment.pk
+        )
+        experiment = Experiment.objects.get(pk=self.experiment.pk)
+        self.assertEqual(experiment.status, 'completed')
+        self.assertEqual(experiment.winning_variation, self.homepage_alternative_1)
+
+    def test_preview(self):
+        response = self.client.get(
+            '/admin/experiments/experiment/report/preview/%d/' % self.homepage_alternative_1.pk
+        )
+        self.assertEqual(response.status_code, 200)