def test_create_and_get(self, user: UserData): new_user = mongo_user_wrapper.create(user) user.uid = new_user.uid self.assertEqual(user, new_user) retrieved_user = mongo_user_wrapper.get(new_user.uid) self.assertEqual(retrieved_user, new_user)
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
def test_to_json(user: UserData, object_id): user.uid = object_id user_json = user.to_json() loaded_json = UserData.from_json(user_json) assert loaded_json == user