from django.test import TestCase

from accounts.models import CustomUser, EmailAddress


class EmailAddressTestCase(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,
        )
        self.email1 = EmailAddress.objects.create(
            user=self.user, email="testuser@example.com", is_primary=True
        )

    def test_str(self):
        self.assertEqual(str(self.email1), "testuser@example.com (primary)")