mybuddy/babybuddy/tests/tests_commands.py

# -*- coding: utf-8 -*-
from django.test import TransactionTestCase
from django.contrib.auth.models import User
from django.core.management import call_command

from core.models import Child


class CommandsTestCase(TransactionTestCase):
    def test_migrate(self):
        call_command('migrate', verbosity=0)
        self.assertIsInstance(User.objects.get(username='admin'), User)

    def test_fake(self):
        call_command('migrate', verbosity=0)
        call_command('fake', children=1, days=7, verbosity=0)
        self.assertEqual(Child.objects.count(), 1)
        call_command('fake', children=2, days=7, verbosity=0)
        self.assertEqual(Child.objects.count(), 3)

    def test_reset(self):
        call_command('reset', verbosity=0, interactive=False)
        self.assertIsInstance(User.objects.get(username='admin'), User)
        self.assertEqual(Child.objects.count(), 1)