Add last changes in the time we had

feldroy · Oct 4, 2020 · 4ecd09f · 4ecd09f
1 parent a2cb52e
commit 4ecd09f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/everycheese/cheeses/tests/factories.py b/everycheese/cheeses/tests/factories.py
@@ -1,3 +1,5 @@
+import pytest
+
 from django.template.defaultfilters import slugify
 
 import factory
@@ -20,4 +22,9 @@ class CheeseFactory(factory.django.DjangoModelFactory):
     creator = factory.SubFactory(UserFactory)
 
     class Meta:
-        model = Cheese
+        model = Cheese
+
+
+@pytest.fixture
+def cheese():
+    return CheeseFactory()
diff --git a/everycheese/cheeses/tests/test_views.py b/everycheese/cheeses/tests/test_views.py
@@ -5,7 +5,7 @@
 
 
 from .factories import UserFactory
-from .factories import CheeseFactory
+from .factories import CheeseFactory, cheese
 from ..models import Cheese
 from ..views import (
     CheeseListView,
@@ -40,8 +40,7 @@ def test_good_cheese_list_view(rf):
     assertContains(response, 'Cheese List')    
 
 
-def test_good_cheese_detail_view(rf):
-    cheese = CheeseFactory()
+def test_good_cheese_detail_view(rf, cheese):
     url = reverse('cheeses:detail',
             kwargs={'slug': cheese.slug}
     )