Adding tests to make sure that custom autofixture classes are used wh…

…en creating m2m instances.

autofixture_tests/models.py

@@ -196,3 +196,8 @@ class DummyStorage(FileSystemStorage):
 class ImageModel(models.Model):
     imgfield = models.ImageField(upload_to='_autofixtures',
                                  storage=dummy_storage)
+
+
+class RelationWithCustomAutofixtureModel(models.Model):
+    user = models.ForeignKey('auth.User', related_name='user1+')
+    users = models.ManyToManyField('auth.User', related_name='user2+')

autofixture_tests/tests/test_base.py

@@ -16,7 +16,10 @@
     M2MModel, ThroughModel, M2MModelThrough, SelfReferencingModel,
     UniqueNullFieldModel, UniqueTogetherNullFieldModel,
     MultipleUniqueTogetherNullFieldModel, SelfReferencingModelNoNull, GFKModel,
-    GRModel)
+    GRModel, RelationWithCustomAutofixtureModel)
+
+
+autofixture.autodiscover()
 
 
 if sys.version_info[0] < 3:
@@ -256,6 +259,19 @@ def test_generate_m2m(self):
         self.assertEqual(SimpleModel.objects.count(), len(all_m2m))
         self.assertEqual(OtherSimpleModel.objects.count(), len(all_secondm2m))
 
+    def test_generate_m2m_with_custom_autofixture(self):
+        filler = AutoFixture(RelationWithCustomAutofixtureModel,
+            generate_fk=True,
+            generate_m2m=(1, 1))
+        instance = filler.create_one()
+        self.assertEqual(instance.users.count(), 1)
+        user = instance.users.get()
+
+        # Detect that the UserFixture was used.
+        self.assertTrue(' ' not in user.username)
+        self.assertTrue(' ' not in user.first_name)
+        self.assertTrue(' ' not in user.last_name)
+
     def test_generate_only_some_m2m(self):
         filler = AutoFixture(
             M2MModel,