summaryrefslogtreecommitdiff
path: root/tests/test_using.py
diff options
context:
space:
mode:
authorRaphaël Barrois <raphael.barrois@polytechnique.org>2015-03-26 22:53:15 +0100
committerRaphaël Barrois <raphael.barrois@polytechnique.org>2015-03-26 22:53:15 +0100
commit636ca46951d710a4b9d9fd61ec1da02294806d3d (patch)
tree23b5bd7bfd7938cccd39d72e829562ca6c515137 /tests/test_using.py
parent72fd943513b0e516f06c53b13ff35ca814b0a4a0 (diff)
downloadfactory-boy-636ca46951d710a4b9d9fd61ec1da02294806d3d.tar
factory-boy-636ca46951d710a4b9d9fd61ec1da02294806d3d.tar.gz
Add support for multidb with Django (Closes #171).
The ``factory.django.DjangoModelFactory`` now takes an extra option: ``` class MyFactory(factory.django.DjangoModelFactory): class Meta: model = models.MyModel database = 'replica' ``` This will create all instances of ``models.Model`` in the ``'replica'`` database.
Diffstat (limited to 'tests/test_using.py')
-rw-r--r--tests/test_using.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/test_using.py b/tests/test_using.py
index 7318f2e..1d7977f 100644
--- a/tests/test_using.py
+++ b/tests/test_using.py
@@ -69,6 +69,9 @@ class FakeModel(object):
def order_by(self, *args, **kwargs):
return [1]
+ def using(self, db):
+ return self
+
objects = FakeModelManager()
def __init__(self, **kwargs):
@@ -1490,6 +1493,9 @@ class BetterFakeModelManager(object):
instance.id = 2
return instance, True
+ def using(self, db):
+ return self
+
class BetterFakeModel(object):
@classmethod