Pytest django 泛型关系旧对象出现

Question

我有一个模型：

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')

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

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 被更改以及为什么对象出现在新的测试用例中。

Answer 1

@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')