diff --git a/baobab/cron/tests/test_sn.py b/baobab/cron/tests/test_sn.py index be59782..2a5b864 100644 --- 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')