36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
from django.test import TestCase
|
||
|
|
||
|
from .models import Category, Gauge
|
||
|
|
||
|
class TestCategoryModel(TestCase):
|
||
|
|
||
|
def test_str(self):
|
||
|
cat = Category(name='test')
|
||
|
|
||
|
self.assertEqual('test', str(cat))
|
||
|
|
||
|
class TestGaugeModel(TestCase):
|
||
|
|
||
|
def test_str(self):
|
||
|
cat = Category.objects.create(name='eau')
|
||
|
gauge = Gauge.objects.create(serial_number='test', category=cat)
|
||
|
|
||
|
self.assertEqual('eau (test)', str(gauge))
|
||
|
|
||
|
def test_default_ordering(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)
|
||
|
gauge_3 = Gauge.objects.create(serial_number='undefined', category=cat_gaz)
|
||
|
gauge_4 = Gauge.objects.create(serial_number='undefined', category=cat_water)
|
||
|
|
||
|
gauges_list = Gauge.objects.all()
|
||
|
|
||
|
self.assertEqual(gauge_1, gauges_list[0])
|
||
|
self.assertEqual(gauge_4, gauges_list[1])
|
||
|
self.assertEqual(gauge_2, gauges_list[2])
|
||
|
self.assertEqual(gauge_3, gauges_list[3])
|
||
|
|