如何在Django模型测试中正确处理ManyToMany字段？

问题场景 #

编写Django模型测试用例时，操作ManyToManyField字段连续出现报错：

初始测试代码 #

class TestVolunteeringModels(TestCase):
    def setUp(self) -> None:
        self.test_need = sample_need(title="Need", description="Description")
        self.test_opportunity = sample_opportunity(title="Opportunity", description="Description")
        for i in range(10):
            self.test_category = sample_category(name="Category")
            self.test_need.category = self.test_category
            self.test_opportunity.category = self.test_category

    def tearDown(self) -> None:
        self.test_need.delete()
        self.test_opportunity.delete()

第一次报错 #

TypeError: Direct assignment to the forward side of a many-to-many set is prohibited. Use category.set() instead.

修改后的代码（使用set()） #

self.test_need.category.set(self.test_category)
self.test_opportunity.category.set(self.test_category)

第二次报错 #

TypeError: 'Category' object is not iterable

解决方案 #

核心原因 #

Django的ManyToMany字段set()方法要求传入可迭代对象（如列表、元组、QuerySet），不能直接传入单个模型实例；同时原循环逻辑会重复覆盖单个分类变量，无法实现多分类关联的预期效果。

正确实现方式 #

场景1：关联单个分类

将单个分类实例放入列表中，再传入set()：

def setUp(self) -> None:
    self.test_need = sample_need(title="Need", description="Description")
    self.test_opportunity = sample_opportunity(title="Opportunity", description="Description")
    self.test_category = sample_category(name="Category")
    # 单个对象放入列表
    self.test_need.category.set([self.test_category])
    self.test_opportunity.category.set([self.test_category])

场景2：关联多个分类（原循环创建10个分类的需求）

先将所有创建的分类存入列表，再一次性关联：

def setUp(self) -> None:
    self.test_need = sample_need(title="Need", description="Description")
    self.test_opportunity = sample_opportunity(title="Opportunity", description="Description")
    # 用列表存储所有创建的分类
    test_categories = []
    for i in range(10):
        # 为分类添加唯一标识避免重名（可选）
        category = sample_category(name=f"Category {i}")
        test_categories.append(category)
    # 一次性关联所有分类
    self.test_need.category.set(test_categories)
    self.test_opportunity.category.set(test_categories)

额外优化 #

Django的TestCase会在每个测试方法执行后自动回滚数据库，无需手动在tearDown中删除实例，可直接移除tearDown方法。

内容的提问来源于stack exchange，提问作者Алексей Василенко