def setUp(self): """Setup test fixture for each model test method.""" try: new_attrs = {} new_attrs.update(self.attrs) new_attrs.update(self.do_get_dependencies()) self.obj = self.klass(**new_attrs) for k, v in self.after_creation().items( ): # usually klass does not have update method setattr(self.obj, k, v) DBSession.add(self.obj) DBSession.flush() return self.obj except: DBSession.rollback() raise
def tearDown(self): """Tear down test fixture for each model test method.""" DBSession.rollback()