mirror of https://github.com/snachodog/mybuddy.git
25 lines
892 B
Python
25 lines
892 B
Python
# -*- 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)
|