def create(user: UserData) -> BaseData:
user_doc = user.to_doc()
mongo_crud_wrapper.create(COLLECTION_NAME, user_doc)
return UserData.from_doc(user_doc)
def update(user: UserData) -> UpdateResult:
user_doc = user.to_doc()
return mongo_crud_wrapper.update(COLLECTION_NAME, user_doc)
def test_get_doc(user: UserData, object_id):
user.uid = object_id
user_doc = user.to_doc()
data_obj = UserData.from_doc(user_doc)
assert data_obj == user