def test_add_routes_table_adds_table(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_routes_table() c = handler.conn.cursor() c.execute("SELECT NAME FROM sqlite_master WHERE " "TYPE='table' AND NAME='routes'") self.assertTrue(c.fetchone(), "routes table not created")
def test_construct_db_calls_add_route_table(self): handler = DatabaseHandler('unit_test_db.sqlite3') mock_add_addresses = Mock() handler._add_addresses_table = mock_add_addresses mock_add_stops = Mock() handler._add_stops_table = mock_add_stops mock_add_routes = Mock() handler._add_routes_table = mock_add_routes handler.initialize_db() self.assertTrue(mock_add_routes.called, "initialize_db did not call _add_routes_table")