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)
