Used setupTestData in m2m_through tests.

django · Jan 7, 2016 · 5ccb7f5 · 5ccb7f5
1 parent 7b8e454
commit 5ccb7f5
Showing 1 changed file with 16 additions and 14 deletions.
diff --git a/tests/m2m_through/tests.py b/tests/m2m_through/tests.py
@@ -13,12 +13,13 @@
 
 
 class M2mThroughTests(TestCase):
-    def setUp(self):
-        self.bob = Person.objects.create(name='Bob')
-        self.jim = Person.objects.create(name='Jim')
-        self.jane = Person.objects.create(name='Jane')
-        self.rock = Group.objects.create(name='Rock')
-        self.roll = Group.objects.create(name='Roll')
+    @classmethod
+    def setUpTestData(cls):
+        cls.bob = Person.objects.create(name='Bob')
+        cls.jim = Person.objects.create(name='Jim')
+        cls.jane = Person.objects.create(name='Jane')
+        cls.rock = Group.objects.create(name='Rock')
+        cls.roll = Group.objects.create(name='Roll')
 
     def test_retrieve_intermediate_items(self):
         Membership.objects.create(person=self.jim, group=self.rock)
@@ -432,14 +433,15 @@ def test_through_fields_self_referential(self):
 
 
 class M2mThroughToFieldsTests(TestCase):
-    def setUp(self):
-        self.pea = Ingredient.objects.create(iname='pea')
-        self.potato = Ingredient.objects.create(iname='potato')
-        self.tomato = Ingredient.objects.create(iname='tomato')
-        self.curry = Recipe.objects.create(rname='curry')
-        RecipeIngredient.objects.create(recipe=self.curry, ingredient=self.potato)
-        RecipeIngredient.objects.create(recipe=self.curry, ingredient=self.pea)
-        RecipeIngredient.objects.create(recipe=self.curry, ingredient=self.tomato)
+    @classmethod
+    def setUpTestData(cls):
+        cls.pea = Ingredient.objects.create(iname='pea')
+        cls.potato = Ingredient.objects.create(iname='potato')
+        cls.tomato = Ingredient.objects.create(iname='tomato')
+        cls.curry = Recipe.objects.create(rname='curry')
+        RecipeIngredient.objects.create(recipe=cls.curry, ingredient=cls.potato)
+        RecipeIngredient.objects.create(recipe=cls.curry, ingredient=cls.pea)
+        RecipeIngredient.objects.create(recipe=cls.curry, ingredient=cls.tomato)
 
     def test_retrieval(self):
         # Forward retrieval