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_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")
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")
