Convert Users, Roles, WorkspaceUsers, TaskOrders to use classmethods

This commit is contained in:
richard-dds
2018-08-03 13:12:54 -04:00
committed by dandds
parent 7b5d76e260
commit b72a16569f
8 changed files with 86 additions and 107 deletions

View File

@@ -5,69 +5,64 @@ from atst.domain.users import Users
from atst.domain.exceptions import NotFoundError, AlreadyExistsError
@pytest.fixture()
def users_repo(session):
return Users(session)
@pytest.fixture(scope="function")
def user_id():
return uuid4()
def test_create_user(users_repo, user_id):
user = users_repo.create(user_id, "developer")
def test_create_user(user_id):
user = Users.create(user_id, "developer")
assert user.id == user_id
def test_create_user_with_nonexistent_role(users_repo, user_id):
def test_create_user_with_nonexistent_role(user_id):
with pytest.raises(NotFoundError):
users_repo.create(user_id, "nonexistent")
Users.create(user_id, "nonexistent")
def test_create_already_existing_user(users_repo, user_id):
users_repo.create(user_id, "developer")
def test_create_already_existing_user(user_id):
Users.create(user_id, "developer")
with pytest.raises(AlreadyExistsError):
users_repo.create(user_id, "developer")
Users.create(user_id, "developer")
def test_get_or_create_nonexistent_user(users_repo, user_id):
user = users_repo.get_or_create(user_id, atat_role_name="developer")
def test_get_or_create_nonexistent_user(user_id):
user = Users.get_or_create(user_id, atat_role_name="developer")
assert user.id == user_id
def test_get_or_create_existing_user(users_repo, user_id):
users_repo.get_or_create(user_id, atat_role_name="developer")
user = users_repo.get_or_create(user_id, atat_role_name="developer")
def test_get_or_create_existing_user(user_id):
Users.get_or_create(user_id, atat_role_name="developer")
user = Users.get_or_create(user_id, atat_role_name="developer")
assert user
def test_get_user(users_repo, user_id):
users_repo.create(user_id, "developer")
user = users_repo.get(user_id)
def test_get_user(user_id):
Users.create(user_id, "developer")
user = Users.get(user_id)
assert user.id == user_id
def test_get_nonexistent_user(users_repo, user_id):
users_repo.create(user_id, "developer")
def test_get_nonexistent_user(user_id):
Users.create(user_id, "developer")
with pytest.raises(NotFoundError):
users_repo.get(uuid4())
Users.get(uuid4())
def test_update_user(users_repo, user_id):
users_repo.create(user_id, "developer")
updated_user = users_repo.update(user_id, "ccpo")
def test_update_user(user_id):
Users.create(user_id, "developer")
updated_user = Users.update(user_id, "ccpo")
assert updated_user.atat_role.name == "ccpo"
def test_update_nonexistent_user(users_repo, user_id):
users_repo.create(user_id, "developer")
def test_update_nonexistent_user(user_id):
Users.create(user_id, "developer")
with pytest.raises(NotFoundError):
users_repo.update(uuid4(), "ccpo")
Users.update(uuid4(), "ccpo")
def test_update_existing_user_with_nonexistent_role(users_repo, user_id):
users_repo.create(user_id, "developer")
def test_update_existing_user_with_nonexistent_role(user_id):
Users.create(user_id, "developer")
with pytest.raises(NotFoundError):
users_repo.update(user_id, "nonexistent")
Users.update(user_id, "nonexistent")