From 070d613b29359465dc8ffdd1dc3ffb116717fc3b Mon Sep 17 00:00:00 2001
From: Alban Gruin
Date: Sat, 29 Sep 2018 20:32:26 +0200
Subject: tests: vérification du nombre de cours dans le test du parseur

Dans certaines conditions, un nouveau cours doit être créé dans la
base de données, mais parfois non. Pour vérifier qu’un seul cours a
bien été rajouté (ou non, en fonction des cas), on teste aussi le
nombre de cours présent dans la base de données.

Signed-off-by: Alban Gruin <alban@pa1ch.fr>
---
 tests.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'tests.py')

diff --git a/tests.py b/tests.py
index a6c84d7..8ec90ba 100644
--- a/tests.py
+++ b/tests.py
@@ -391,6 +391,7 @@ class UPS2018ParserTestCase(TestCase):
 
     def test_get_event(self):
         get_event = self.parser._Parser__get_event
+        count = Course.objects.count()
 
         event = get_event(
             {"start": "2018-09-21T10:00:00", "end": "2018-09-21T12:00:00",
@@ -419,6 +420,9 @@ class UPS2018ParserTestCase(TestCase):
         self.assertEqual(event.end, timezone.make_aware(
             datetime.datetime(2018, 9, 21, 12, 0, 0)))
 
+        self.assertEqual(count, Course.objects.count() - 1)
+        count += 1
+
         events = [
             {
                 "text": "(10:00-12:00)<br>COURS/TD<br>Cours quelconque"
@@ -505,6 +509,9 @@ class UPS2018ParserTestCase(TestCase):
             else:
                 self.assertIsNone(event.notes)
 
+            self.assertEqual(count, Course.objects.count() - 1)
+            count += 1
+
         event = get_event(
             {"start": "2018-09-21T10:00:00", "end": "2018-09-21T12:00:00",
              "text": "Global Event"},
@@ -513,6 +520,7 @@ class UPS2018ParserTestCase(TestCase):
             timezone.make_aware(datetime.datetime(2018, 10, 1)),
             2018, 38)
         self.assertIsNone(event)
+        self.assertEqual(count, Course.objects.count())
 
         event = get_event(
             {"start": "2018-09-21T10:00:00", "end": "2018-09-21T12:00:00",
@@ -522,6 +530,7 @@ class UPS2018ParserTestCase(TestCase):
             timezone.make_aware(datetime.datetime(2018, 10, 1)),
             2018, 39)
         self.assertIsNone(event)
+        self.assertEqual(count, Course.objects.count())
 
     @mock.patch("requests.get", side_effect=mock_requests_get)
     def test_get_events(self, *args, **kwargs):
-- 
cgit v1.2.1