from django.test import TestCase

from accounts.models import CustomUser, EmailAddress


class CustomUserTestCase(TestCase):

    def setUp(self):
        self.user = CustomUser.objects.create(
            login_id="testuser",
            username="Test User",
            is_staff=True,
            is_active=True,
            is_superuser=False,
            password_changed=False,
        )
        EmailAddress.objects.create(user=self.user, email="testuser1@example.com", is_primary=False)
        EmailAddress.objects.create(user=self.user, email="testuser2@example.com", is_primary=True)

    def test_str(self):
        self.assertEqual(str(self.user), "Test User")

    def test_get_full_name(self):
        self.assertEqual(self.user.get_full_name(), "Test User")

    def test_get_short_name(self):
        self.assertEqual(self.user.get_short_name(), "Test User")

    def test_get_primary_email(self):
        self.assertEqual(self.user.get_primary_email(), "testuser2@example.com")