add update method to Users repo

This commit is contained in:
dandds
2018-10-15 13:17:47 -04:00
parent 30318b68bb
commit ec7504fb20
3 changed files with 45 additions and 9 deletions

View File

@@ -2,7 +2,7 @@ import pytest
from uuid import uuid4
from atst.domain.users import Users
from atst.domain.exceptions import NotFoundError, AlreadyExistsError
from atst.domain.exceptions import NotFoundError, AlreadyExistsError, UnauthorizedError
DOD_ID = "my_dod_id"
@@ -52,20 +52,32 @@ def test_get_user_by_dod_id():
assert user == new_user
def test_update_user():
def test_update_role():
new_user = Users.create(DOD_ID, "developer")
updated_user = Users.update(new_user.id, "ccpo")
updated_user = Users.update_role(new_user.id, "ccpo")
assert updated_user.atat_role.name == "ccpo"
def test_update_nonexistent_user():
def test_update_role_with_nonexistent_user():
Users.create(DOD_ID, "developer")
with pytest.raises(NotFoundError):
Users.update(uuid4(), "ccpo")
Users.update_role(uuid4(), "ccpo")
def test_update_existing_user_with_nonexistent_role():
new_user = Users.create(DOD_ID, "developer")
with pytest.raises(NotFoundError):
Users.update(new_user.id, "nonexistent")
Users.update_role(new_user.id, "nonexistent")
def test_update_user():
new_user = Users.create(DOD_ID, "developer")
updated_user = Users.update(new_user, {"first_name": "Jabba"})
assert updated_user.first_name == "Jabba"
def test_update_user_with_dod_id():
new_user = Users.create(DOD_ID, "developer")
with pytest.raises(UnauthorizedError):
Users.update(new_user, {"dod_id": "1234567890"})