add category property test

This commit is contained in:
Fred 2016-08-04 21:52:20 +02:00
parent b768f535e0
commit b6d218553c
1 changed files with 12 additions and 2 deletions

View File

@ -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): class TestCategoryModel(TestCase):
@ -32,4 +35,11 @@ class TestGaugeModel(TestCase):
self.assertEqual(gauge_4, gauges_list[1]) self.assertEqual(gauge_4, gauges_list[1])
self.assertEqual(gauge_2, gauges_list[2]) self.assertEqual(gauge_2, gauges_list[2])
self.assertEqual(gauge_3, gauges_list[3]) 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)