def test_addPortfolio(self):
nj = self.makeTradeJsons(trades_json)
db = DB('test.db')
db.dropTables()
pf_j = self.fillPortfolioVars(db, nj)
if not (db.addPortfolio(pf_j)):
self.fail()
if (db.addPortfolio(pf_j)):
self.fail()
def test_getTrades(self):
nj = TestDB.makeTradeJsons()
db = DB('test.db')
db.dropTables()
db.addClient(test_DB.client_json)
pf_j = TestDB.fillPortfolioVars(db, nj)
self.failIf(not (db.addPortfolio(pf_j)))
cp = ClientPortfolios(db,'*****@*****.**')
a = [p['portfolioName'] for p in cp.getPortfolios()]
self.failIf( len(a) <= 0)
self.failIf(len([t['tradeId'] for t in cp.getTrades('My portfolio')]) <= 0)
