def test_remove_stock_successful(self):
with self.client:
add_stock('NASDAQ', 'AAPL')
remove_stock('NASDAQ', 'AAPL')
stock = StockListEntry.query.filter_by(market='NASDAQ',
ticker='AAPL').first()
self.assertFalse(stock)
def test_add_stock_successful(self):
with self.client:
add_stock('NASDAQ', 'AAPL')
stock = StockListEntry.query.filter_by(market='NASDAQ',
ticker='AAPL').first()
self.assertTrue(stock)
self.assertEqual(str(stock),
'<Stock List Entry Object NASDAQ:AAPL>')
def test_object_to_dictionary(self):
with self.client:
add_stock('NASDAQ', 'AAPL')
stock = StockListEntry.query.filter_by(market='NASDAQ',
ticker='AAPL').first()
stock_dict = object_to_dict(stock)
self.assertEqual(stock_dict['market'], 'NASDAQ')
self.assertEqual(stock_dict['ticker'], 'AAPL')
self.assertEqual(stock_dict['current_ratio'], 'N/A')
def test_get_stocks_from_db(self):
with self.client:
stocks = get_stocks_from_db()
self.assertEqual(stocks, [])
add_stock('NASDAQ', 'AAPL')
stocks = get_stocks_from_db()
self.assertEqual(stocks[0]['market'], 'NASDAQ')
self.assertEqual(stocks[0]['ticker'], 'AAPL')
self.assertEqual(stocks[0]['price'], 'N/A')
def test_add_stock_fail(self):
with self.client:
stock = add_stock('NASDAQ', 'AAPLLL')
stock = StockListEntry.query.filter_by(market='NASDAQ',
ticker='AAPLLL').first()
self.assertFalse(stock)
def add(market, ticker):
add_stock(market, ticker)
return redirect('/')