mybuddy/babybuddy/tests/tests_models.py

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

from babybuddy.models import Settings


class SettingsTestCase(TestCase):
    def setUp(self):
        call_command('migrate', verbosity=0)

    def test_settings(self):
        credentials = {
            'username': 'Test',
            'password': 'User'
        }
        user = User.objects.create_user(is_superuser=True, **credentials)
        self.assertIsInstance(user.settings, Settings)
        self.assertEqual(
            user.settings.dashboard_refresh_rate_milliseconds, 60000)

        user.settings.dashboard_refresh_rate = None
        user.save()
        self.assertIsNone(user.settings.dashboard_refresh_rate_milliseconds)

        user.settings.language = 'fr'
        user.save()
        self.assertEqual(user.settings.language, 'fr')