Add a '@use_strategy' decorator for forcing alternate strategies.

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

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index 8da655e..a548b56 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -141,6 +141,14 @@ class FactoryDefaultStrategyTestCase(unittest.TestCase):
         TestModelFactory.default_strategy = base.BUILD_STRATEGY
         self.assertRaises(base.StubFactory.UnsupportedStrategy, TestModelFactory)
 
+    def test_change_strategy(self):
+        @base.use_strategy(base.CREATE_STRATEGY)
+        class TestModelFactory(base.StubFactory):
+            one = 'one'
+
+        self.assertEqual(base.CREATE_STRATEGY, TestModelFactory.default_strategy)
+
+
 class FactoryCreationTestCase(unittest.TestCase):
     def testFactoryFor(self):
         class TestFactory(base.Factory):