diff options
author | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2011-05-13 15:24:53 +0200 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2011-05-13 15:24:53 +0200 |
commit | f57a0c213ffe12325e20e69e5cc03ff27633d06c (patch) | |
tree | 690e107617e228ca5f7027a50f5bca0885a74b28 | |
parent | d3835fdc4e556ca5abfdd98e89437a96ab54e4c7 (diff) | |
download | factory-boy-f57a0c213ffe12325e20e69e5cc03ff27633d06c.tar factory-boy-f57a0c213ffe12325e20e69e5cc03ff27633d06c.tar.gz |
Trailing whitespace
Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>
-rw-r--r-- | tests.py | 120 |
1 files changed, 60 insertions, 60 deletions
@@ -11,14 +11,14 @@ class TestObject(object): self.four = four class FakeDjangoModel(object): - class FakeDjangoManager(object): + class FakeDjangoManager(object): def create(self, **kwargs): fake_model = FakeDjangoModel(**kwargs) fake_model.id = 1 return fake_model - + objects = FakeDjangoManager() - + def __init__(self, **kwargs): for name, value in kwargs.iteritems(): setattr(self, name, value) @@ -31,73 +31,73 @@ class FactoryTestCase(unittest.TestCase): def testAttribute(self): class TestObjectFactory(Factory): one = 'one' - + test_object = TestObjectFactory.build() - self.assertEqual(test_object.one, 'one') - + self.assertEqual(test_object.one, 'one') + def testSequence(self): class TestObjectFactory(Factory): one = Sequence(lambda n: 'one' + n) two = Sequence(lambda n: 'two' + n) - + test_object0 = TestObjectFactory.build() self.assertEqual(test_object0.one, 'one0') self.assertEqual(test_object0.two, 'two0') - + test_object1 = TestObjectFactory.build() self.assertEqual(test_object1.one, 'one1') self.assertEqual(test_object1.two, 'two1') - + def testLazyAttribute(self): class TestObjectFactory(Factory): one = LazyAttribute(lambda a: 'abc' ) two = LazyAttribute(lambda a: a.one + ' xyz') - + test_object = TestObjectFactory.build() self.assertEqual(test_object.one, 'abc') self.assertEqual(test_object.two, 'abc xyz') - + def testLazyAttributeNonExistentParam(self): class TestObjectFactory(Factory): one = LazyAttribute(lambda a: a.does_not_exist ) - + try: TestObjectFactory() self.fail() except AttributeError as e: self.assertTrue('does not exist' in str(e)) - + def testLazyAttributeSequence(self): class TestObjectFactory(Factory): one = LazyAttributeSequence(lambda a, n: 'abc' + n) two = LazyAttributeSequence(lambda a, n: a.one + ' xyz' + n) - + test_object0 = TestObjectFactory.build() self.assertEqual(test_object0.one, 'abc0') self.assertEqual(test_object0.two, 'abc0 xyz0') - + test_object1 = TestObjectFactory.build() self.assertEqual(test_object1.one, 'abc1') self.assertEqual(test_object1.two, 'abc1 xyz1') - + def testLazyAttributeDecorator(self): class TestObjectFactory(Factory): @lazy_attribute def one(a): return 'one' - + test_object = TestObjectFactory.build() self.assertEqual(test_object.one, 'one') - + def testSequenceDecorator(self): class TestObjectFactory(Factory): @sequence def one(n): return 'one' + n - + test_object = TestObjectFactory.build() self.assertEqual(test_object.one, 'one0') - + def testLazyAttributeSequenceDecorator(self): class TestObjectFactory(Factory): @lazy_attribute_sequence @@ -106,126 +106,126 @@ class FactoryTestCase(unittest.TestCase): @lazy_attribute_sequence def two(a, n): return a.one + ' two' + n - + test_object = TestObjectFactory.build() self.assertEqual(test_object.one, 'one0') self.assertEqual(test_object.two, 'one0 two0') - + def testBuildWithParameters(self): class TestObjectFactory(Factory): one = Sequence(lambda n: 'one' + n) two = Sequence(lambda n: 'two' + n) - + test_object0 = TestObjectFactory.build(three='three') self.assertEqual(test_object0.one, 'one0') self.assertEqual(test_object0.two, 'two0') self.assertEqual(test_object0.three, 'three') - + test_object1 = TestObjectFactory.build(one='other') self.assertEqual(test_object1.one, 'other') self.assertEqual(test_object1.two, 'two1') - + def testCreate(self): class TestModelFactory(Factory): one = 'one' - + test_model = TestModelFactory.create() self.assertEqual(test_model.one, 'one') self.assertTrue(test_model.id) - + def testInheritance(self): class TestObjectFactory(Factory): one = 'one' two = LazyAttribute(lambda a: a.one + ' two') - + class TestObjectFactory2(TestObjectFactory): FACTORY_FOR = TestObject - + three = 'three' four = LazyAttribute(lambda a: a.three + ' four') - + test_object = TestObjectFactory2.build() self.assertEqual(test_object.one, 'one') self.assertEqual(test_object.two, 'one two') self.assertEqual(test_object.three, 'three') self.assertEqual(test_object.four, 'three four') - + def testInheritanceWithInheritedClass(self): class TestObjectFactory(Factory): one = 'one' two = LazyAttribute(lambda a: a.one + ' two') - + class TestFactory(TestObjectFactory): three = 'three' four = LazyAttribute(lambda a: a.three + ' four') - + test_object = TestFactory.build() self.assertEqual(test_object.one, 'one') self.assertEqual(test_object.two, 'one two') self.assertEqual(test_object.three, 'three') self.assertEqual(test_object.four, 'three four') - + def testSetCreationFunction(self): def creation_function(class_to_create, **kwargs): return "This doesn't even return an instance of {0}".format(class_to_create.__name__) - + class TestModelFactory(Factory): pass - + TestModelFactory.set_creation_function(creation_function) - + test_object = TestModelFactory.create() self.assertEqual(test_object, "This doesn't even return an instance of TestModel") class FactoryDefaultStrategyTestCase(unittest.TestCase): def setUp(self): self.default_strategy = Factory.default_strategy - + def tearDown(self): Factory.default_strategy = self.default_strategy - + def testBuildStrategy(self): Factory.default_strategy = BUILD_STRATEGY - + class TestModelFactory(Factory): one = 'one' - + test_model = TestModelFactory() self.assertEqual(test_model.one, 'one') self.assertFalse(test_model.id) - + def testCreateStrategy(self): # Default default_strategy - + class TestModelFactory(Factory): one = 'one' test_model = TestModelFactory() self.assertEqual(test_model.one, 'one') self.assertTrue(test_model.id) - + def testStubStrategy(self): Factory.default_strategy = STUB_STRATEGY - + class TestModelFactory(Factory): one = 'one' test_model = TestModelFactory() self.assertEqual(test_model.one, 'one') self.assertFalse(hasattr(test_model, 'id')) # We should have a plain old object - + def testUnknownStrategy(self): Factory.default_strategy = 'unknown' - + class TestModelFactory(Factory): one = 'one' self.assertRaises(Factory.UnknownStrategy, TestModelFactory) - - def testStubWithNonStubStrategy(self): + + def testStubWithNonStubStrategy(self): class TestModelFactory(StubFactory): one = 'one' - + TestModelFactory.default_strategy = CREATE_STRATEGY self.assertRaises(StubFactory.UnsupportedStrategy, TestModelFactory) @@ -234,21 +234,21 @@ class FactoryCreationTestCase(unittest.TestCase): def testFactoryFor(self): class TestFactory(Factory): FACTORY_FOR = TestObject - + self.assertTrue(isinstance(TestFactory.build(), TestObject)) - + def testAutomaticAssociatedClassDiscovery(self): class TestObjectFactory(Factory): pass - + self.assertTrue(isinstance(TestObjectFactory.build(), TestObject)) - + def testStub(self): class TestFactory(StubFactory): pass - + self.assertEqual(TestFactory.default_strategy, STUB_STRATEGY) - + def testInheritanceWithStub(self): class TestObjectFactory(StubFactory): pass @@ -257,9 +257,9 @@ class FactoryCreationTestCase(unittest.TestCase): pass self.assertEqual(TestFactory.default_strategy, STUB_STRATEGY) - + # Errors - + def testNoAssociatedClassWithAutodiscovery(self): try: class TestFactory(Factory): @@ -267,7 +267,7 @@ class FactoryCreationTestCase(unittest.TestCase): self.fail() except Factory.AssociatedClassError as e: self.assertTrue('autodiscovery' in str(e)) - + def testNoAssociatedClassWithoutAutodiscovery(self): try: class Test(Factory): @@ -275,7 +275,7 @@ class FactoryCreationTestCase(unittest.TestCase): self.fail() except Factory.AssociatedClassError as e: self.assertTrue('autodiscovery' not in str(e)) - + def testInheritanceFromMoreThanOneFactory(self): class TestObjectFactory(StubFactory): pass |