时间戳

首页 > 解决方案 > pytest django 泛型关系老对象出现

python - pytest django 泛型关系老对象出现

问题描述

我有一个模型：

class Product(models.Model):
    url = models.TextField(blank=True)
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
    object_id = models.PositiveIntegerField()
    content_object = GenericForeignKey('content_type', 'object_id')

class Store(models.Model):
    product = GenericRelation('Product')

以及为商店创建产品的 celery 任务：

def my_celery_task(store_obj_pk):
    store_obj = Store.objects.get(pk=store_obj_pk)
    Product.objects.create(
                      content_type=ContentType.objects.get_for_model(store_obj),
                        object_id=store_obj.pk,
                    )

我想用 pytest 测试这个任务

class TestStoreTask:

    def test_store_one(self):
        self.__test_store(store_product={
            'name': 'one'
        })

    def test_store_two(self):
        self.__test_store(store_product={
            'name': 'two'
        })

    def __test_store(self, store_product):

        store_obj = Store.objects.create()
        my_celery_task(store_obj_pk=store_obj.pk)
        print(store_obj.product.all())

所以我希望每个测试都有一个对象。但是当我在其中运行两个测试时，test_store_two我有两个对象。这个对象的 id 为 2 和 3。

所以我的输出test_store_one：

<QuerySet [<Product: Product object (1)>]>

对于test_store_two：

<QuerySet [<Product: Product object (2)>, <Product: Product object (3)>]>

我不明白为什么 id 被更改以及为什么对象出现在新的测试用例中。

标签： pythondjangocelerypytest

解决方案

@hoefling 感谢您的帮助。这是非常奇怪的行为。所以在任务中创建对象，是另一个类的方法，比如：

    class SomeManagerClass:
        products = []

        def generate_products(self):
            self.products.append('one')

        def create_products(self, store_obj):
             for title in self.products:
                 Product.objects.create(
                      title=title,
                      content_type=ContentType.objects.get_for_model(store_obj),
                      object_id=store_obj.pk,
                    )

在任务中，我像这样使用它：

def my_celery_task(store_obj_pk):
    store_obj = Store.objects.get(pk=store_obj_pk)
    manager = SomeManagerClass()
    manager.generate_products()
    manager.create_products(store_obj)

当我调用my_celery_task两次时，我得到 3 个对象，而不是两个。因为self.products以前的结果保持活力。

我通过修改我的generate_products函数来解决这个问题：

def generate_products(self):
    self.products = []
    self.products.append('one')

推荐阅读