add tests

Georiviere · May 15, 2024 · 13dd449 · 13dd449
1 parent fbf3aa1
commit 13dd449
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/georiviere/river/tests/test_managers.py b/georiviere/river/tests/test_managers.py
@@ -0,0 +1,15 @@
+from django.test import TestCase
+
+from georiviere.river.models import Stream
+from georiviere.river.tests.factories import StreamFactory
+
+
+class RiverManagerTestCase(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        StreamFactory.create_batch(5)
+
+    def test_truncate(self):
+        self.assertEqual(Stream.objects.count(), 5)
+        Stream.objects.truncate()
+        self.assertEqual(Stream.objects.count(), 0)