28 lines
783 B
Python
28 lines
783 B
Python
import pytest
|
|
from django.contrib.auth import get_user_model
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
@pytest.mark.django_db
|
|
class TestUserModel:
|
|
def test_create_seeker(self):
|
|
user = User.objects.create_user(
|
|
username='seeker1', password='pass123', role='seeker'
|
|
)
|
|
assert user.role == 'seeker'
|
|
assert user.is_seeker is True
|
|
assert user.is_admin is False
|
|
|
|
def test_create_admin(self):
|
|
user = User.objects.create_user(
|
|
username='admin1', password='pass123', role='admin'
|
|
)
|
|
assert user.is_admin is True
|
|
|
|
def test_create_superadmin(self):
|
|
user = User.objects.create_user(
|
|
username='super1', password='pass123', role='superadmin'
|
|
)
|
|
assert user.is_superadmin is True
|