Misago/misago/users/tests/test_deleteprofilefield.py

from io import StringIO

from django.contrib.auth import get_user_model
from django.core.management import call_command
from django.test import TestCase

from misago.users.management.commands import deleteprofilefield


UserModel = get_user_model()


class DeleteProfileFieldTests(TestCase):
    def test_no_fieldname(self):
        """utility has no showstoppers when no fieldname is given"""
        out = StringIO()
        call_command(deleteprofilefield.Command(), stderr=out)
        command_output = out.getvalue().splitlines()[0].strip()

        self.assertEqual(command_output, "Specify fieldname to delete.")

    def test_no_fields_set(self):
        """utility has no showstoppers when no fields are set"""
        out = StringIO()
        call_command(deleteprofilefield.Command(), "gender", stdout=out)
        command_output = out.getvalue().splitlines()[0].strip()

        self.assertEqual(
            command_output, '"gender" profile field has been deleted from 0 users.'
        )

    def test_delete_fields(self):
        """utility has no showstoppers when no fields are set"""
        user = UserModel.objects.create_user("Bob", "bob@bob.com", "pass123")
        user.profile_fields = {"gender": "male", "bio": "Yup!"}
        user.save()

        out = StringIO()
        call_command(deleteprofilefield.Command(), "gender", stdout=out)
        command_output = out.getvalue().splitlines()[0].strip()

        self.assertEqual(
            command_output, '"gender" profile field has been deleted from 1 users.'
        )

        user = UserModel.objects.get(pk=user.pk)
        self.assertEqual(user.profile_fields, {"bio": "Yup!"})