Add core model tests.

core/tests/tests_models.py

@@ -0,0 +1,177 @@
+# -*- coding: utf-8 -*-
+from __future__ import unicode_literals
+
+from django.contrib.auth.models import User
+from django.core.management import call_command
+from django.test import TestCase
+from django.utils import timezone
+
+from core import models
+
+
+class ChildTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+
+    def test_child_create(self):
+        child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+        self.assertEqual(child, models.Child.objects.get(first_name='First'))
+        self.assertEqual(child.slug, 'first-last')
+        self.assertEqual(str(child), 'First Last')
+        self.assertEqual(child.name(), 'First Last')
+        self.assertEqual(child.name(reverse=True), 'Last, First')
+
+
+class DiaperChangeTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+        self.change = models.DiaperChange.objects.create(
+            child=self.child,
+            time=timezone.localtime() - timezone.timedelta(days=1),
+            wet=1,
+            solid=1,
+            color='black'
+        )
+
+    def test_diaperchange_create(self):
+        self.assertEqual(self.change, models.DiaperChange.objects.first())
+        self.assertEqual(str(self.change), 'Diaper Change')
+
+    def test_diaperchange_attributes(self):
+        self.assertListEqual(
+            self.change.attributes(), ['wet', 'solid', 'black'])
+
+
+class FeedingChangeTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+
+    def test_feeding_create(self):
+        feeding = models.Feeding.objects.create(
+            child=self.child,
+            start=timezone.localtime() - timezone.timedelta(minutes=30),
+            end=timezone.localtime(),
+            type='formula',
+            method='bottle',
+            amount=2
+        )
+        self.assertEqual(feeding, models.Feeding.objects.first())
+        self.assertEqual(str(feeding), 'Feeding')
+        self.assertEqual(feeding.duration, timezone.timedelta(minutes=30))
+
+
+class NoteTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+
+    def test_note_create(self):
+        note = models.Note.objects.create(
+            child=self.child, note='Note', time=timezone.localtime())
+        self.assertEqual(note, models.Note.objects.first())
+        self.assertEqual(str(note), 'Note')
+
+
+class SleepTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+
+    def test_sleep_create(self):
+        sleep = models.Sleep.objects.create(
+            child=self.child,
+            start=timezone.localtime() - timezone.timedelta(minutes=30),
+            end=timezone.localtime(),
+        )
+        self.assertEqual(sleep, models.Sleep.objects.first())
+        self.assertEqual(str(sleep), 'Sleep')
+        self.assertEqual(sleep.duration, timezone.timedelta(minutes=30))
+
+
+class TimerTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.named = models.Timer.objects.create(
+            name='Named',
+            end=timezone.localtime(),
+            user=User.objects.first()
+        )
+        self.unnamed = models.Timer.objects.create(
+            end=timezone.localtime(),
+            user=User.objects.first()
+        )
+
+    def test_timer_create(self):
+        self.assertEqual(self.named, models.Timer.objects.get(name='Named'))
+        self.assertEqual(str(self.named), 'Named')
+        self.assertEqual(self.unnamed, models.Timer.objects.get(name=None))
+        self.assertEqual(
+            str(self.unnamed), 'Timer #{}'.format(self.unnamed.id))
+
+    def test_timer_restart(self):
+        self.named.restart()
+        self.assertIsNone(self.named.end)
+        self.assertIsNone(self.named.duration)
+        self.assertTrue(self.named.active)
+
+    def test_timer_stop(self):
+        stop_time = timezone.localtime()
+        self.unnamed.stop(end=stop_time)
+        self.assertEqual(self.unnamed.end, stop_time)
+        self.assertEqual(
+            self.unnamed.duration, self.unnamed.start - self.unnamed.end)
+        self.assertFalse(self.unnamed.active)
+
+    def test_timer_current_duration(self):
+        timer = models.Timer.objects.create(user=User.objects.first())
+        # Timer.start uses auto_now_add, so it cannot be set in create().
+        timer.start = timezone.localtime() - timezone.timedelta(minutes=30)
+        timer.save()
+
+        self.assertEqual(
+            timer.current_duration(), timezone.timedelta(minutes=30))
+        timer.stop()
+        self.assertEqual(
+            timer.current_duration(), timezone.timedelta(minutes=30))
+
+
+class TummyTimeTestCase(TestCase):
+    def setUp(self):
+        call_command('migrate', verbosity=0)
+        self.child = models.Child.objects.create(
+            first_name='First',
+            last_name='Last',
+            birth_date=timezone.localdate()
+        )
+
+    def test_tummytime_create(self):
+        tummy_time = models.TummyTime.objects.create(
+            child=self.child,
+            start=timezone.localtime() - timezone.timedelta(minutes=30),
+            end=timezone.localtime(),
+        )
+        self.assertEqual(tummy_time, models.TummyTime.objects.first())
+        self.assertEqual(str(tummy_time), 'Tummy Time')
+        self.assertEqual(tummy_time.duration, timezone.timedelta(minutes=30))