Add test for reports view

tobami · tobami · commit 471b2dc54b05 · 2014-08-03T22:49:35.000+02:00
diff --git a/codespeed/feeds.py b/codespeed/feeds.py
@@ -3,6 +3,7 @@
 from django.conf import settings
 from django.db.models import Q
 
+
 class ResultFeed(Feed):
     title = settings.WEBSITE_NAME
     link = "/changes/"
@@ -13,7 +14,8 @@ def items(self):
             .order_by('-revision__date')[:10]
 
     def item_title(self, item):
-        return "%s: %s" % (item.revision.get_short_commitid(), item.item_description())
+        return "%s: %s" % (item.revision.get_short_commitid(),
+                           item.item_description())
 
     description_template = "codespeed/changes_table.html"
 
@@ -31,15 +33,17 @@ def get_context_data(self, **kwargs):
             'env': report.environment,
         }
 
+
 class LatestEntries(ResultFeed):
     description = "Last benchmark runs"
 
     def result_filter(self):
         return Q(revision__branch__name=settings.DEF_BRANCH)
 
+
 class LatestSignificantEntries(ResultFeed):
     description = "Last benchmark runs with significant changes"
 
     def result_filter(self):
         return Q(revision__branch__name=settings.DEF_BRANCH,
-                 colorcode__in = ('red','green'))
+                 colorcode__in=('red', 'green'))
diff --git a/codespeed/tests/test_views.py b/codespeed/tests/test_views.py
@@ -0,0 +1,32 @@
+from django.test import TestCase
+from django.core.urlresolvers import reverse
+
+from codespeed.models import Environment
+
+
+class TestAddResult(TestCase):
+
+    def setUp(self):
+        self.path = reverse('codespeed.views.reports')
+        self.e = Environment.objects.create(name='Dual Core', cpu='Core 2 Duo 8200')
+        self.data = {
+            'commitid': 'abcd1',
+            'branch': 'default',
+            'project': 'MyProject',
+            'executable': 'myexe O3 64bits',
+            'benchmark': 'float',
+            'environment': 'Dual Core',
+            'result_value': 200,
+        }
+        resp = self.client.post(reverse('codespeed.views.add_result'), self.data)
+        self.assertEqual(resp.status_code, 202)
+        self.data['commitid'] = "abcd2"
+        self.client.post(reverse('codespeed.views.add_result'), self.data)
+        assert resp.status_code == 202
+
+    def test_reports(self):
+        response = self.client.get(self.path)
+
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('Latest Results', response.content)
+        self.assertIn(self.data['commitid'], response.content)
diff --git a/codespeed/tests/tests.py b/codespeed/tests/tests.py
@@ -18,8 +18,7 @@ class TestAddResult(TestCase):
 
     def setUp(self):
         self.path = reverse('codespeed.views.add_result')
-        self.e = Environment(name='Dual Core', cpu='Core 2 Duo 8200')
-        self.e.save()
+        self.e = Environment.objects.create(name='Dual Core', cpu='Core 2 Duo 8200')
         temp = datetime.today()
         self.cdate = datetime(
             temp.year, temp.month, temp.day, temp.hour, temp.minute, temp.second)
