mybuddy/babybuddy/tests/tests_commands.py

25 lines
892 B
Python
Raw Normal View History

2017-11-11 22:25:28 +00:00
# -*- coding: utf-8 -*-
from django.test import TransactionTestCase
2017-11-11 22:25:28 +00:00
from django.contrib.auth.models import User
from django.core.management import call_command
from core.models import Child
class CommandsTestCase(TransactionTestCase):
2017-11-11 22:25:28 +00:00
def test_migrate(self):
2022-02-10 00:00:30 +00:00
call_command("migrate", verbosity=0)
self.assertIsInstance(User.objects.get(username="admin"), User)
2017-11-11 22:25:28 +00:00
def test_fake(self):
2022-02-10 00:00:30 +00:00
call_command("migrate", verbosity=0)
call_command("fake", children=1, days=7, verbosity=0)
2017-11-11 22:25:28 +00:00
self.assertEqual(Child.objects.count(), 1)
2022-02-10 00:00:30 +00:00
call_command("fake", children=2, days=7, verbosity=0)
2017-11-11 22:25:28 +00:00
self.assertEqual(Child.objects.count(), 3)
def test_reset(self):
2022-02-10 00:00:30 +00:00
call_command("reset", verbosity=0, interactive=False)
self.assertIsInstance(User.objects.get(username="admin"), User)
self.assertEqual(Child.objects.count(), 1)