Use the new Meta API introduced in factory_boy 2.4.0 in factories.

python · May 13, 2015 · 0c80c39 · 0c80c39
1 parent a50d041
commit 0c80c39
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 11 deletions.
diff --git a/companies/factories.py b/companies/factories.py
@@ -8,8 +8,10 @@
 
 
 class CompanyFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = Company
-    FACTORY_DJANGO_GET_OR_CREATE = ('name',)
+
+    class Meta:
+        model = Company
+        django_get_or_create = ('name',)
 
     name = factory.Sequence(lambda n: 'Company {}'.format(n))
     email = factory.Sequence(lambda n: 'zombie_{}@python.org'.format(n))
diff --git a/jobs/factories.py b/jobs/factories.py
@@ -9,21 +9,27 @@
 
 
 class JobCategoryFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = JobCategory
-    FACTORY_DJANGO_GET_OR_CREATE = ('name',)
+
+    class Meta:
+        model = JobCategory
+        django_get_or_create = ('name',)
 
     name = factory.Sequence(lambda n: 'Job Category {}'.format(n))
 
 
 class JobTypeFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = JobType
-    FACTORY_DJANGO_GET_OR_CREATE = ('name',)
+
+    class Meta:
+        model = JobType
+        django_get_or_create = ('name',)
 
     name = factory.Sequence(lambda n: 'Job Type {}'.format(n))
 
 
 class JobFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = Job
+
+    class Meta:
+        model = Job
 
     category = factory.SubFactory(JobCategoryFactory)
     job_title = factory.Sequence(lambda n: 'Job Title #{}'.format(n))

diff --git a/users/factories.py b/users/factories.py
@@ -4,8 +4,10 @@
 
 
 class UserFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = User
-    FACTORY_DJANGO_GET_OR_CREATE = ('username',)
+
+    class Meta:
+        model = User
+        django_get_or_create = ('username',)
 
     username = factory.Sequence(lambda n: 'zombie{}'.format(n))
     email = factory.Sequence(lambda n: "zombie%s@example.com" % n)
@@ -19,8 +21,10 @@ class StaffUserFactory(UserFactory):
 
 
 class MembershipFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = Membership
-    FACTORY_DJANGO_GET_OR_CREATE = ('creator',)
+
+    class Meta:
+        model = Membership
+        django_get_or_create = ('creator',)
 
     psf_code_of_conduct = True
     psf_announcements = True