MongoDB CRUD测试报错'$set'为空，如何解决？

问题概述 #

测试基于MongoDB的AnimalShelter类CRUD功能时，调用update方法传入空的更新数据，触发了如下WriteError：

WriteError: '$set' is empty. You must specify a field like so: {$set: {: ...}}

错误原因 #

1. AnimalShelter类的update方法缺少参数校验：当传入的updateData为空字典时，代码直接构造{"$set": updateData}，生成了MongoDB不允许的空$set操作（MongoDB要求$set必须包含至少一个键值对）。
2. 测试代码传入无效参数：调用update时传入了空字典{}作为更新数据，直接触发了上述MongoDB的校验错误。

修复方案 #

1. 完善AnimalShelter类的update方法 #

在执行更新操作前，增加对searchData和updateData的有效性校验，避免无效操作：

def update(self, searchData, updateData):
    # 校验搜索条件非空，且更新数据为非空字典
    if not searchData or not isinstance(updateData, dict) or len(updateData) == 0:
        return "错误：搜索条件不能为空，且更新数据必须为非空字典"
    try:
        result = self.database.animals.update_many(searchData, {"$set": updateData})
        return result.raw_result
    except Exception as e:
        return f"更新失败：{str(e)}"

同时建议优化create方法（原方法允许插入空文档，且使用了已废弃的insert方法）：

def create(self, data):
    if data and isinstance(data, dict):
        # 使用官方推荐的insert_one替代已废弃的insert
        insert_result = self.database.animals.insert_one(data)
        return insert_result.inserted_id
    else:
        raise Exception("无法保存：数据参数为空或不是字典类型")

2. 修改测试代码，传入有效参数 #

测试时需传入具体的搜索条件和更新数据，避免空参数触发错误或危险操作（比如删除所有文档）：

from AnimalShelter import AnimalShelter

# 初始化实例
test = AnimalShelter("aacuser","French")

# 测试create：传入有效动物数据
new_animal = {
    "animal_id": "A12345",
    "name": "Buddy",
    "type": "Dog",
    "age_upon_outcome": "2 years"
}
success_id = test.create(new_animal)
print(f"创建成功，文档ID：{success_id}")

# 测试read：搜索指定animal_id的文档
result = test.read({"animal_id": "A12345"})
for doc in result:
    print("查询结果：", doc)

# 测试update：更新指定文档的字段
update_result = test.update(
    {"animal_id": "A12345"},
    {"name": "Buddy Updated", "age_upon_outcome": "3 years"}
)
print("更新结果：", update_result)

# 测试delete：删除指定文档（可选，测试后可注释避免数据丢失）
# delete_result = test.delete({"animal_id": "A12345"})
# print("删除结果：", delete_result)

额外注意点 #

• 原delete方法同样存在风险：如果传入空的deleteData，会删除集合中所有文档，建议增加类似的参数校验。
• 原read方法中，当searchData为空时会返回所有文档，若集合数据量大可能影响性能，可根据需求调整逻辑。

内容的提问来源于stack exchange，提问作者GuillyStudiesCode