Add factory.build(), factory.create(), factory.stub() methods.

Those are shortcuts for writing a full Factory class.

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

1 files changed, 21 insertions, 0 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index 01855c3..5855682 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -55,6 +55,27 @@ class SafetyTestCase(unittest.TestCase):
         self.assertRaises(RuntimeError, base.BaseFactory)
 
 
+class SimpleBuildTestCase(unittest.TestCase):
+    """Tests the minimalist 'factory.build/create' functions."""
+
+    def test_build(self):
+        obj = base.build(TestObject, two=2)
+        self.assertEqual(obj.one, None)
+        self.assertEqual(obj.two, 2)
+        self.assertEqual(obj.three, None)
+        self.assertEqual(obj.four, None)
+
+    def test_create(self):
+        obj = base.create(FakeDjangoModel, foo='bar')
+        self.assertEqual(obj.id, 1)
+        self.assertEqual(obj.foo, 'bar')
+
+    def test_stub(self):
+        obj = base.stub(TestObject, three=3)
+        self.assertEqual(obj.three, 3)
+        self.assertFalse(hasattr(obj, 'two'))
+
+
 class FactoryTestCase(unittest.TestCase):
     def testAttribute(self):
         class TestObjectFactory(base.Factory):