class DatabaseTestCase(unittest.TestCase): def setUp(self): self.persistence_manager = PersistenceManager(True) #just keep a top level reference to these guys for ease of use self.session = db_connector.get_session() #and then create the database schema, reference tables self.persistence_manager.create_schema() self.persistence_manager.populate_reference_tables() self.session.commit() def tearDown(self): # self.session.flush() self.session.close_all() self.persistence_manager.drop_schema()
self.assertEqual( DiceFace.EVADE, adjustment.to_dice.dice_face) result = throw.results[2] self.assertEqual( 3, result.dice_num) self.assertEqual( DiceType.GREEN, result.dice.dice_type) self.assertEqual( DiceFace.FOCUS, result.dice.dice_face) self.assertEqual( DiceFace.EVADE, result.final_dice.dice_face) self.assertEqual( 1, len (result.adjustments)) adjustment = result.adjustments[0] self.assertEqual( adjustment.base_result_id, result.id) self.assertEqual( DiceThrowAdjustmentType.CONVERT, adjustment.adjustment_type) self.assertEqual( DiceFace.FOCUS, adjustment.from_dice.dice_face ) self.assertEqual( DiceFace.EVADE, adjustment.to_dice.dice_face) if __name__ == "__main__": if len (sys.argv) == 1: unittest.main() elif sys.argv[1] == 'create': pm = PersistenceManager(db_connector) pm.create_schema() pm.populate_reference_tables() db_connector.get_session().commit() db_connector.get_session().close_all() elif sys.argv[1] == 'destroy': pm = PersistenceManager(db_connector) pm.drop_schema() db_connector.get_session().commit() db_connector.get_session().close_all()