python - 为具有多对多关系的 Django 模型编写测试

Question

我想为具有多对多关系的 Django 模型编写测试，但出现此错误:

ValueError: "< Tour: tour >" needs to have a value for field "id" before this many-to-many relationship can be used.

我的测试:

class ModelTestCase(TestCase):
    def setUp(self):
        self.mock_file = mock.MagicMock(File)
        self.mock_file.name = 'MockFile'
        self.before_count = Tour.objects.count()
        self.tour_agency = TourAgency.objects.create(
            name='agency',
            username='agency')
        self.tour_category = TourCategory.objects.create(name='category')
        self.tour_equipment = TourEquipment.objects.create(name='equipment')
        self.tour_leader = TourLeader.objects.create(
            name='leader',
            email='leader@sample.com',
            username='leader',
            bio='sample text',)
        self.tour_tag = TourTag.objects.create(name='tag')

    def test_model_can_create_tour(self):
        self.tour = Tour.objects.create(
            name='tour',
            description='description',
            summary='summary',
            agency=self.tour_agency,
            equipment=self.tour_equipment,
            category=self.tour_category,
            tags=self.tour_tag,
            activity_type='activity type',
            date=datetime.now,
            photo=self.mock_file)
        self.tour.save()
        self.tour.save_m2m()
        count = Tour.objects.count()
        self.assertNotEqual(self.before_count, count)

我将尝试使用 .save() 保存对象，但它不起作用。

Answer 1

在添加多对多关系之前，您首先需要保存游览模型。查看代码我认为多对多是字段“标签”

def test_model_can_create_tour(self):
    self.tour = Tour.objects.create(
        name='tour',
        description='description',
        summary='summary',
        agency=self.tour_agency,
        equipment=self.tour_equipment,
        category=self.tour_category,
        activity_type='activity type',
        date=datetime.now,
        photo=self.mock_file)

    # Adding the tour later.
    self.tour.tags.add(self.tour_tag)
    count = Tour.objects.count()
    self.assertNotEqual(self.before_count, count)

这应该可以通过测试。