OpenSlides/tests/unit/users/test_models.py

from unittest import TestCase
from unittest.mock import MagicMock, call, patch

from django.core.exceptions import ObjectDoesNotExist

from openslides.users.models import UserManager


class UserManagerTest(TestCase):
    def test_create_user(self):
        """
        Tests that create_user saves a new user with a set password.
        """
        user = MagicMock()
        user_manager = UserManager()
        user_manager.model = MagicMock(return_value=user)
        user_manager._db = "my_test_db"

        return_user = user_manager.create_user(
            "test_username", "test_password", test_kwarg="test_kwarg"
        )

        user_manager.model.assert_called_once_with(
            username="test_username", test_kwarg="test_kwarg"
        )
        user.set_password.assert_called_once_with("test_password")
        user.save.assert_called_once_with(using="my_test_db", skip_autoupdate=False)
        self.assertEqual(
            return_user, user, "The returned user is not the created user."
        )


class UserManagerGenerateUsername(TestCase):
    """
    Tests for the manager method generate_username.
    """

    def setUp(self):
        self.exists_mock = MagicMock()
        self.filter_mock = MagicMock(return_value=self.exists_mock)
        self.manager = UserManager()
        self.manager.filter = self.filter_mock

    def test_clear_strings(self):
        self.exists_mock.exists.return_value = False

        self.assertEqual(
            self.manager.generate_username(
                "wiaf9eecu9mooJiZ3Lah", "ieHaVe9ci7mooPhe0AuY"
            ),
            "wiaf9eecu9mooJiZ3Lah ieHaVe9ci7mooPhe0AuY",
        )

    def test_unstripped_strings(self):
        self.exists_mock.exists.return_value = False

        self.assertEqual(
            self.manager.generate_username(
                "ouYeuwai0pheukeeShah ", "  Waefa8gahj8ohRaeroca\n"
            ),
            "ouYeuwai0pheukeeShah Waefa8gahj8ohRaeroca",
            "The returned value should only have one whitespace between the " "names.",
        )

    def test_empty_second_string(self):
        self.exists_mock.exists.return_value = False

        self.assertEqual(
            self.manager.generate_username("foobar", ""),
            "foobar",
            "The returned value should not have whitespaces at the end.",
        )

    def test_empty_first_string(self):
        self.exists_mock.exists.return_value = False

        self.assertEqual(
            self.manager.generate_username("", "foobar"),
            "foobar",
            "The returned value should not have whitespaces at the beginning.",
        )

    def test_two_empty_strings(self):
        self.exists_mock.exists.return_value = False

        with self.assertRaises(ValueError, msg="A ValueError should be raised."):
            self.manager.generate_username("", "")

    def test_used_username(self):
        self.exists_mock.exists.side_effect = (True, False)

        self.assertEqual(
            self.manager.generate_username("user", "name"),
            "user name 1",
            "If the username already exists, a number should be added to the " "name.",
        )

    def test_two_used_username(self):
        self.exists_mock.exists.side_effect = (True, True, False)

        self.assertEqual(
            self.manager.generate_username("user", "name"),
            "user name 2",
            "If the username with a number already exists, a higher number "
            "should be added to the name.",
        )

    def test_umlauts(self):
        self.exists_mock.exists.return_value = False

        self.assertEqual(
            self.manager.generate_username("äöü", "ßüäö"),
            "äöü ßüäö",
            "The method gen_username has also to work with umlauts.",
        )


@patch("openslides.users.models.choice")
class UserManagerGeneratePassword(TestCase):
    def test_normal(self, mock_choice):
        """
        Test normal run of the method.
        """
        mock_choice.side_effect = tuple("test_password")

        self.assertEqual(UserManager().generate_password(), "test_pas")
        # choice has to be called 8 times
        mock_choice.assert_has_calls(
            [
                call("abcdefghijkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ23456789")
                for _ in range(8)
            ]
        )


@patch("openslides.users.models.Permission")
@patch("openslides.users.models.Group")
class UserManagerCreateOrResetAdminUser(TestCase):
    def test_add_admin_group(self, mock_group, mock_permission):
        """
        Tests that the Group with pk=2 (Admin group) is added to the admin.
        """
        admin_user = MagicMock()
        manager = UserManager()
        manager.get = MagicMock(return_value=(admin_user))

        manager.create_or_reset_admin_user()

        admin_user.groups.add.assert_called_once_with(
            2
        )  # the admin should be added to the admin group with pk=2

    def test_password_set_to_admin(self, mock_group, mock_permission):
        """
        Tests that the password of the admin is set to 'admin'.
        """
        admin_user = MagicMock()
        manager = UserManager()
        manager.get = MagicMock(return_value=(admin_user))

        staff_group = MagicMock(name="Staff")
        mock_group.objects.get_or_create = MagicMock(return_value=(staff_group, True))
        mock_permission.get = MagicMock()

        manager.create_or_reset_admin_user()

        self.assertEqual(admin_user.default_password, "admin")
        admin_user.save.assert_called_once_with(skip_autoupdate=True)

    @patch("openslides.users.models.User")
    def test_return_value(self, mock_user, mock_group, mock_permission):
        """
        Tests that the method returns True when a user is created.
        """
        manager = UserManager()
        manager.get = MagicMock(side_effect=ObjectDoesNotExist)
        manager.model = mock_user

        staff_group = MagicMock(name="Staff")
        mock_group.objects.get_or_create = MagicMock(return_value=(staff_group, True))
        mock_permission.get = MagicMock()

        self.assertEqual(
            manager.create_or_reset_admin_user(),
            True,
            "The method create_or_reset_admin_user should return True when a "
            "new user is created.",
        )

    @patch("openslides.users.models.User")
    def test_attributes_of_created_user(self, mock_user, mock_group, mock_permission):
        """
        Tests username and last_name of the created admin user.
        """
        admin_user = MagicMock(username="admin", last_name="Administrator")
        manager = UserManager()
        manager.get = MagicMock(side_effect=ObjectDoesNotExist)
        manager.model = mock_user

        staff_group = MagicMock(name="Staff")
        mock_group.objects.get_or_create = MagicMock(return_value=(staff_group, True))
        mock_permission.get = MagicMock()

        manager.create_or_reset_admin_user()

        self.assertEqual(
            admin_user.username,
            "admin",
            "The username of a new created admin should be 'admin'.",
        )
        self.assertEqual(
            admin_user.last_name,
            "Administrator",
            "The last_name of a new created admin should be 'Administrator'.",
        )