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.assertTrue(True) 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() elif sys.argv[1] == 'alter': pm = PersistenceManager(db_connector) #update all records in the ship_pilot table meta = XWingMetaData() #upgrades = meta.upgrades() #pm.create_upgrade_table() # upgrade_objects = [] # for ut in upgrades.keys(): # uvs = upgrades[ut] # for uv in uvs: # upgrade = Upgrade( upgrade_type=UpgradeType.from_string( ut ), name=uv['name'], cost=uv['cost']) # upgrade_objects.append( upgrade )
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()