diff --git a/sherlock/tests.py b/sherlock/tests.py index b89bace..52a7d86 100644 --- a/sherlock/tests.py +++ b/sherlock/tests.py @@ -1,6 +1,9 @@ -from django.test import TestCase +from datetime import datetime -from .models import Category, Gauge +from django.test import TestCase +from django.utils import timezone + +from .models import Category, Gauge, Mesure class TestCategoryModel(TestCase): @@ -32,4 +35,11 @@ class TestGaugeModel(TestCase): self.assertEqual(gauge_4, gauges_list[1]) self.assertEqual(gauge_2, gauges_list[2]) self.assertEqual(gauge_3, gauges_list[3]) + + def test_category_property(self): + cat = Category.objects.create(name='eau') + gauge = Gauge.objects.create(serial_number='undefined', category=cat) + mesure = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=gauge) + + self.assertEqual(cat, mesure.category)