from forge import Forge
with Forge.define('color') as f:
f.name = 'Red'
def setUp(self):
Forge.configure(models='forge.test.support.models')
if 'user' not in Forge._registry:
Forge.define('user', name='Matte', age=25)
def should_support_with_context_format(self):
with Forge.define('pet') as p:
p.breed = 'Husky'
assert Forge._registry['pet'] == {'breed': 'Husky'}
def should_create_associations(self):
Forge.define('car', owner=Forge.build('user'))
assert Forge.build('car').owner.name == 'Matte'
def should_not_allow_duplicate_factory_names(self):
with self.assertRaises(DuplicateFactoryError):
Forge.define('user', color='red')
