def test_add_addresses_table_adds_table(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() c = handler.conn.cursor() c.execute("SELECT NAME FROM sqlite_master WHERE " "TYPE='table' and NAME='addresses'") self.assertTrue(c.fetchone(), "addresses table not created")
def test_add_addresses_table_adds_table(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() c = handler.conn.cursor() c.execute("SELECT NAME FROM sqlite_master WHERE " "TYPE='table' and NAME='addresses'") self.assertTrue(c.fetchone(), "addresses table not created")
def test_add_address_uses_MapLocation_id_if_nonzero(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() address_location = MapLocation(latitude=0.33, longitude=4, id=100) handler.add_address(address_location) c = handler.conn.cursor() c.execute("SELECT * FROM addresses") self.assertEqual(100, c.fetchone()[0])
def test_add_address_adds_to_address_table(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() handler.add_address(location=MapLocation(latitude=0.56, longitude=9.5)) c = handler.conn.cursor() c.execute("SELECT latitude, longitude FROM addresses") row = c.fetchone() self.assertEqual((0.56, 9.5), row)
def test_add_address_uses_MapLocation_id_if_nonzero(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() address_location = MapLocation(latitude=0.33, longitude=4, id=100) handler.add_address(address_location) c = handler.conn.cursor() c.execute("SELECT * FROM addresses") self.assertEqual(100, c.fetchone()[0])
def test_add_address_adds_to_address_table(self, mock_init_db): handler = DatabaseHandler('unit_test_db.sqlite3') handler._add_addresses_table() handler.add_address(location=MapLocation(latitude=0.56, longitude=9.5)) c = handler.conn.cursor() c.execute("SELECT latitude, longitude FROM addresses") row = c.fetchone() self.assertEqual((0.56, 9.5), row)
def test_construct_db_calls_add_stop_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 handler.initialize_db() self.assertTrue(mock_add_stops.called, "initialize_db did not call _add_stops_table")
def test_construct_db_calls_add_stop_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 handler.initialize_db() self.assertTrue(mock_add_stops.called, "initialize_db did not call _add_stops_table")