atst/tests/models/test_user.py

27 lines
667 B
Python

import pytest
from sqlalchemy.exc import InternalError
from atst.models.user import User
from tests.factories import UserFactory
def test_profile_complete_with_all_info():
user = UserFactory.create()
assert user.profile_complete
@pytest.mark.parametrize("missing_field", User.REQUIRED_FIELDS)
def test_profile_complete_with_missing_info(missing_field):
user = UserFactory.create()
setattr(user, missing_field, None)
assert not user.profile_complete
def test_cannot_update_dod_id(session):
user = UserFactory.create()
user.dod_id = "23403498202"
session.add(user)
with pytest.raises(InternalError):
session.commit()