use .first() instead of a get with a guessing id

Gandi · Oct 27, 2015 · 1aa0ce2 · 1aa0ce2
1 parent d9bf7e6
commit 1aa0ce2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/baobab/cron/tests/test_sn.py b/baobab/cron/tests/test_sn.py
@@ -48,7 +48,7 @@ def test_03_event_started_with_end(self):
         event = self._create_event(now, end)
         cron_social_network()
         self.assertEqual(SnEvent.objects.count(), 1)
-        sn_event = SnEvent.objects.get(id=1)
+        sn_event = SnEvent.objects.first()
         self.assertEqual(sn_event.event, event)
         self.assertEqual(sn_event.name, MockSN.name)
         self.assertEqual(sn_event.sn_id, '1')
@@ -58,7 +58,7 @@ def test_04_event_started_without_end(self):
         event = self._create_event(now)
         cron_social_network()
         self.assertEqual(SnEvent.objects.count(), 1)
-        sn_event = SnEvent.objects.get(id=1)
+        sn_event = SnEvent.objects.first()
         self.assertEqual(sn_event.event, event)
         self.assertEqual(sn_event.name, MockSN.name)
         self.assertEqual(sn_event.sn_id, '1')
@@ -87,7 +87,7 @@ def test_06_eventlog_default(self):
 
         cron_social_network()
         self.assertEqual(SnEventLog.objects.count(), 1)
-        sn_eventlog = SnEventLog.objects.get(id=1)
+        sn_eventlog = SnEventLog.objects.first()
         self.assertEqual(sn_eventlog.eventlog, eventlog)
         self.assertEqual(sn_eventlog.name, MockSN.name)
         self.assertEqual(sn_eventlog.sn_id, '1')