def testDelete(self): apple = Position(ticker="AAPL", shares=100, account_id=3) apple.save() orange = Position(ticker="ORNG", shares=150, account_id=3) orange.save() apple.delete() with sqlite3.connect(DBPATH) as connection: cursor = connection.cursor() SQL = "SELECT * FROM positions;" cursor.execute(SQL) rows = cursor.fetchall() self.assertEqual(len(rows), 1, "delete should delete 1 row in the database")
def testPosition(self): position = Position(ticker_symbol="NYSE:T", number_of_share=200, account_pk=8) position.save() self.assertEqual(position.ticker_symbol, "NYSE:T") self.assertEqual(position.number_of_share, 200) self.assertEqual(position.account_pk, 8) position = Position(pk=1, ticker_symbol="NYSE:ACB", number_of_share=300, account_pk=2) position.save() self.assertEqual(position.ticker_symbol, "NYSE:ACB") self.assertEqual(position.number_of_share, 300) self.assertEqual(position.account_pk, 2) position = Position(pk=4) position.delete() self.assertEqual(position.ticker_symbol, None) position = Position().one_from_pk(pk=1) self.assertEqual(position.ticker_symbol, "NYSE:ACB") self.assertEqual(position.number_of_share, 300) self.assertEqual(position.account_pk, 2) position = Position().all_from_where_clause() self.assertEqual(position[1].ticker_symbol, "NASDAQ:EA") self.assertEqual(position[1].number_of_share, 100) self.assertEqual(position[1].account_pk, 1) position = Position().all_from_where_clause() self.assertEqual(position[2].ticker_symbol, "NYSE:T") self.assertEqual(position[2].number_of_share, 200) self.assertEqual(position[2].account_pk, 8)
def testDelete(self): position = Position(pk=2) position.delete() self.assertEqual(position.ticker, None)