def _check_database_quotes(self, symbol, total, check_date, total_date, is_eod, stock):
with database.connect_db(TEST_DB_NAME) as db:
db.execute("SELECT COUNT(*) FROM options WHERE symbol=?;", [symbol])
self.assertEqual(db.fetchall(), [(total,)])
if total > 0:
database_quotes = options.query_historical(symbol, check_date, TEST_DB_NAME)
self.assertEqual(len(database_quotes), total_date)
self._check_quotes(database_quotes, symbol, check_date, is_eod, stock)
def test_query_historical_non_existing(self):
test_date = dates.get_database_timestamp().date()
quotes = options.query_historical(NON_EXISTING_SYMBOL, test_date, TEST_DB_NAME)
self.assertEqual(len(quotes), 0)
def test_query_historical_existing(self):
test_date = dates.get_database_timestamp().date()
options.fetch_historical("VIX", TEST_DB_NAME)
quotes = options.query_historical("VIX", test_date, TEST_DB_NAME)
self._check_quotes(quotes, "VIX", test_date, True, None)