def user_schema() -> UserDBSchema:
return UserDBSchema(
user_id=1,
email="*****@*****.**",
full_name="test",
is_active=True,
is_superuser=False,
created_date="1/1/2020",
)
async def test_user_list_users(
self, user_model: UserModel, user_schema: UserCreateSchema
) -> None:
user_service = UserService(UserQueriesDummy())
result = await user_service.list_users()
assert result == [UserDBSchema(
user_id=1,
email="*****@*****.**",
full_name="test",
is_active=True,
is_superuser=False,
created_date="1/1/2020",
)]
async def create_user(self, user: UserCreateSchema) -> UserDBSchema:
new_user = await self.__user_queries.create_user(user)
return UserDBSchema.from_orm(new_user)
async def remove_user(self, user_id: int) -> UserDBSchema:
user_removed = await self.__user_queries.delete_user(user_id)
return UserDBSchema.from_orm(user_removed)
async def update_user(self, user_id: int,
new_user: UserUpdateSchema) -> UserDBSchema:
old_user = await self.__user_queries.get_user_byid(user_id)
user_updated = await self.__user_queries.update_user(
old_user, new_user)
return UserDBSchema.from_orm(user_updated)
async def get_user_by_id(self, user_id: int) -> Optional[UserDBSchema]:
user = await self.__user_queries.get_user_byid(user_id)
if user:
return UserDBSchema.from_orm(user)
else:
return None
async def list_users(self) -> List[UserDBSchema]:
users = await self.__user_queries.get_all_users()
users_schema = list(map(lambda x: UserDBSchema.from_orm(x), users))
return users_schema