fixtures: crée toujours la catégorie mère d'une sous-categorie

zestedesavoir · Aug 21, 2021 · 3e658da · 3e658da
1 parent 935ceab
commit 3e658da
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/zds/tutorialv2/tests/tests_lists.py b/zds/tutorialv2/tests/tests_lists.py
@@ -119,7 +119,7 @@ def test_list_categories(self):
         context_categories = list(resp.context_data["categories"])
         self.assertEqual(context_categories[0].contents_count, 10)
         self.assertEqual(context_categories[0].subcategories, [subcategory_1, subcategory_2])
-        self.assertEqual(context_categories, [category_1])
+        self.assertIn(category_1, context_categories)
 
     def test_private_lists(self):
         tutorial = PublishedContentFactory(author_list=[self.user_author])

diff --git a/zds/utils/factories.py b/zds/utils/factories.py
@@ -80,9 +80,11 @@ def _generate(cls, create, attrs):
 
         subcategory = super()._generate(create, attrs)
 
-        if category is not None:
-            relation = CategorySubCategory(category=category, subcategory=subcategory)
-            relation.save()
+        if category is None:
+            category = CategoryFactory()
+
+        relation = CategorySubCategory(category=category, subcategory=subcategory)
+        relation.save()
 
         return subcategory