25 lines
819 B
Python
25 lines
819 B
Python
from django.test import TestCase
|
|
from uuid import UUID
|
|
from accounts.models import EmailWhiteList
|
|
|
|
|
|
class EmailWhiteListModelTestCase(TestCase):
|
|
|
|
def setUp(self):
|
|
email_a = EmailWhiteList(email='user_a@example.com')
|
|
email_a.save()
|
|
EmailWhiteList.objects.create(email='user_b@example.com')
|
|
|
|
def test_email_white_list_model(self):
|
|
white_listed_emails = EmailWhiteList.objects.all()
|
|
|
|
self.assertTrue(len(white_listed_emails) == 2)
|
|
|
|
email_a = white_listed_emails.filter(pk=1)
|
|
self.assertTrue(len(email_a) == 0)
|
|
|
|
for i, x in enumerate(white_listed_emails):
|
|
self.assertTrue(isinstance(white_listed_emails[i].pk, UUID))
|
|
self.assertTrue(isinstance(white_listed_emails[i].email, str))
|
|
self.assertEquals(str(x), x.email)
|