define setUp in TestMesureModel

This commit is contained in:
Fred 2016-08-05 21:32:08 +02:00
parent 45aa06be2a
commit 8ac0150914
1 changed files with 11 additions and 13 deletions

View File

@ -38,22 +38,20 @@ class TestGaugeModel(TestCase):
class TestMesureModel(TestCase): class TestMesureModel(TestCase):
def test_category_property(self): def setUp(self):
cat = Category.objects.create(name='eau') self.cat_water = Category.objects.create(name='eau')
gauge = Gauge.objects.create(serial_number='undefined', category=cat) self.cat_gaz = Category.objects.create(name='gaz')
mesure = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=gauge)
self.assertEqual(cat, mesure.category) self.gauge_1 = Gauge.objects.create(serial_number='undefined', category=self.cat_water)
self.gauge_2 = Gauge.objects.create(serial_number='undefined', category=self.cat_gaz)
self.mesure_1 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=self.gauge_1)
self.mesure_2 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.002, gauge=self.gauge_2)
def test_category_property(self):
self.assertEqual(self.cat_water, self.mesure_1.category)
def test_diff_raises_exception(self): def test_diff_raises_exception(self):
cat_water = Category.objects.create(name='eau')
cat_gaz = Category.objects.create(name='gaz')
gauge_1 = Gauge.objects.create(serial_number='undefined', category=cat_water)
gauge_2 = Gauge.objects.create(serial_number='undefined', category=cat_gaz)
mesure_1 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.001, gauge=gauge_1)
mesure_2 = Mesure.objects.create(moment=timezone.make_aware(datetime.now()), value=0.002, gauge=gauge_2)
with self.assertRaises(Exception): with self.assertRaises(Exception):
mesure_1.diff(mesure_2) mesure_1.diff(mesure_2)