def setUp(self):
self.checker = PostgresDatabaseHealthChecker("dbname='stockcat' user='******' host='localhost' password='******'")
class PostgresDatabaseTest(unittest.TestCase):
def setUp(self):
self.checker = PostgresDatabaseHealthChecker("dbname='stockcat' user='******' host='localhost' password='******'")
def tearDown(self):
self.checker = None
def test_check_connection(self):
self.checker.check_connection()
def test_check_table_existed(self):
self.checker.check_table_existed('stock_symbol')
self.checker.check_table_existed('balance_sheet')
self.checker.check_table_existed('income_statement')
self.checker.check_table_existed('cash_flow')
self.checker.check_table_existed('operating_revenue')
self.checker.check_table_existed('capital_increase_history')
self.checker.check_table_existed('dividend_policy')
def test_check_balance_sheet_entry_existed(self):
entry = {
'table' : 'balance_sheet',
'stock_symbol' : '2330',
'stmt_date' : datetime.date(2010, 9, 30),
}
self.checker.check_entry_existed(entry)