Misago/misago/users/tests/test_namechanges.py

from django.contrib.auth import get_user_model
from django.test import TestCase

from misago.users.namechanges import get_available_namechanges_data


UserModel = get_user_model()


class UsernameChangesTests(TestCase):
    def test_username_changes_helper(self):
        """username changes are tracked correctly"""
        test_user = UserModel.objects.create_user('Bob', 'bob@bob.com', 'pass123')

        namechanges = get_available_namechanges_data(test_user)
        self.assertEqual(namechanges, {
            'changes_left': 2,
            'next_change_on': None,
        })

        self.assertEqual(test_user.namechanges.count(), 0)

        test_user.set_username('Boberson')
        test_user.save(update_fields=['username', 'slug'])

        namechanges = get_available_namechanges_data(test_user)
        self.assertEqual(namechanges, {
            'changes_left': 1,
            'next_change_on': None,
        })

        self.assertEqual(test_user.namechanges.count(), 1)