class Test(unittest.TestCase): def setUp(self): db_url = 'mysql://*****:*****@localhost/kew_pe?charset=utf8' self.control = Control(file_archive='',db_url=db_url) def tearDown(self): self.control._dispose_() def testSave(self): with self.control._session_context as session: deal = session.query(model.Deal).filter(model.Deal.name=="12 Eaton Place").first() if deal: data = deal.to_json() collections = [] self.control.genoa_save('Deal', data, collections, surrogates=[])
class Test(unittest.TestCase): def setUp(self): test_db_path = 'deal_test.db' if os.path.isfile(test_db_path): os.remove(test_db_path) test_db_path = '../deal_test.db' # db_url = "sqlite:///" + test_db_path db_url = 'mysql://*****:*****@localhost/kew_pe?charset=utf8' self.control = Control(file_archive='',db_url=db_url, echo=False) def tearDown(self): self.control._dispose_() def _testHoldings(self): with self.control._session_context as session: deal = session.query(model.Deal).get(1) if deal: for i, h in enumerate(deal.holdings): print i, h.id, h.name, h.deal_id def _testEntity(self): with self.control._session_context as session: obj = session.query(model.Entity).get(99) if obj: for i, o in enumerate(obj.accounts): print i, o def _testVersionInc(self): with self.control._session_context as session: deals = [34] version_date = datetime.date.today() self.control.roll_version(deals, version_date=version_date, version_label="New Report") def testColProps(self): with self.control._session_context as session: print "*"*20, datetime.datetime.now() deal_27 = session.query(model.Deal).get(27) deal_34 = session.query(model.Deal).get(34) print deal_27.version_id, deal_34.version_id print deal_27.current_value, deal_34.current_value print deal_27.current_invested_capital, deal_34.current_invested_capital print deal_27.current_pnl, deal_34.current_pnl print deal_27.cash_pnl, deal_34.cash_pnl #print '*'*20, deal.current_invested_capital #json = deal.to_json() #print json print "*"*20, datetime.datetime.now() print "genoa_item" #list = self.control.genoa_list(model='Cashflow', criteria={u'deal_id': 27, u'version_id': 27}, offset=None, limit=None, summary=None, order_by=None) data = self.control.genoa_item(model='Deal', pk=27, collections=['cashflows'], surrogates=[], version_id=None) print "*"*20, datetime.datetime.now() print "genoa_save" self.control.genoa_save('Deal', data, collections=[['cashflows', 'Cashflow', 'ONETOMANY']], surrogates=[]) print "*"*20, datetime.datetime.now() print "genoa_list" self.control.genoa_list('Cashflow', criteria={'deal_id':27, 'version_id':59}) print "*"*20, datetime.datetime.now()