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)