43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
|
|
from mayan.apps.common.tests.base import BaseTestCase
|
|
from mayan.apps.common.tests.utils import mute_stdout
|
|
|
|
from ..models import AutoAdminSingleton
|
|
from ..settings import setting_username
|
|
|
|
from .literals import TEST_ADMIN_USER_PASSWORD
|
|
|
|
|
|
class AutoAdminHandlerTestCase(BaseTestCase):
|
|
def test_post_admin_creation(self):
|
|
logging.disable(logging.INFO)
|
|
|
|
with mute_stdout():
|
|
AutoAdminSingleton.objects.create_autoadmin()
|
|
|
|
self.assertEqual(
|
|
AutoAdminSingleton.objects.get().account.username,
|
|
setting_username.value
|
|
)
|
|
|
|
user = AutoAdminSingleton.objects.get().account
|
|
|
|
user.set_password(TEST_ADMIN_USER_PASSWORD)
|
|
user.save(update_fields=['password'])
|
|
|
|
self.assertEqual(AutoAdminSingleton.objects.get().account, None)
|
|
|
|
def test_double_creation(self):
|
|
with mute_stdout():
|
|
AutoAdminSingleton.objects.create_autoadmin()
|
|
|
|
self.assertEqual(AutoAdminSingleton.objects.count(), 1)
|
|
|
|
logging.disable(logging.ERROR)
|
|
|
|
AutoAdminSingleton.objects.create_autoadmin()
|
|
self.assertEqual(AutoAdminSingleton.objects.count(), 1)
|