Update test factories to use latest factory_boy

incuna · Jun 16, 2015 · 8bb829d · 8bb829d
1 parent 654e4dc
commit 8bb829d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ colour-runner==0.0.4
 coverage==3.7.1
 dj-database-url==0.3.0
 dj-inmemorystorage==1.3.0
-factory_boy==2.4.1
+factory_boy==2.5.2
 flake8==2.3.0
 flake8-import-order==0.5.1
 incuna-test-utils==6.0.0

diff --git a/user_management/models/tests/factories.py b/user_management/models/tests/factories.py
@@ -6,12 +6,13 @@
 
 
 class UserFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = get_user_model()
-
     name = factory.Sequence('Test User {}'.format)
     email = factory.Sequence('email{}@example.com'.format)
     is_active = True
 
+    class Meta:
+        model = get_user_model()
+
     @factory.post_generation
     def password(self, create, extracted='default password', **kwargs):
         self.raw_password = extracted
@@ -21,7 +22,8 @@ def password(self, create, extracted='default password', **kwargs):
 
 
 class AuthTokenFactory(factory.DjangoModelFactory):
-    FACTORY_FOR = AuthToken
-
     key = factory.Sequence('key{}'.format)
     user = factory.SubFactory(UserFactory)
+
+    class Meta:
+        model = AuthToken