class AircraftTestCase(unittest.TestCase):
adapter = None
repository = None
def setUp(self):
self.adapter = SqliteAdapter.get_instance()
self.mapper = AircraftDbMapper(self.adapter)
self.repository = AircraftRepository(self.adapter)
def tearDown(self):
self.mapper.delete_all()
self.adapter.connection.close()
def test_fetch_identical_returns_none_when_aircraft_does_not_exist(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft = self.repository.fetch_identical(aircraft)
self.assertEqual(None, aircraft)
def test_fetch_identical_returns_aircraft_when_exists(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
self.mapper.insert(aircraft)
aircraft = self.repository.fetch_identical(aircraft)
self.assertNotEqual(None, aircraft)
def handle(self, aircraft):
# fetch the aircraft image if missing
if not self.exists(aircraft):
self.logger.info('Downloading "{}" aircraft image.'.format(
aircraft.registration))
url = self.get_url(aircraft)
if url:
aircraft.image = self.download_image(url, aircraft)
AircraftDbMapper(self.adapter).update(aircraft)
self.logger.info('Aircraft image downloaded.')
self.logger.info('Optimizing image.')
self.optimize(aircraft.image)
self.logger.info('Creating thumbnail.')
self.create_thumbnail(aircraft.image)
return
self.logger.warning(
'Aircraft "{}" image URL could not be fetched.'.format(
aircraft.registration))
return
self.logger.info('Aircraft "{}" image already exists.'.format(
aircraft.registration))
return
def index(self):
"""Render index page."""
root_path = environment.web_root_path
adapter = SqliteAdapter.get_instance()
aircrafts = AircraftDbMapper(adapter).fetch_all(('last_seen', 'DESC'),
10)
adapter.connection.close()
return template('index.tpl', aircrafts=aircrafts, root_path=root_path)
def setUp(self):
self.adapter = SqliteAdapter.get_instance()
self.mapper = AircraftDbMapper(self.adapter)
self.repository = AircraftRepository(self.adapter)
class AircraftTestCase(unittest.TestCase):
adapter = None
mapper = None
def setUp(self):
self.adapter = SqliteAdapter.get_instance()
self.mapper = AircraftDbMapper(self.adapter)
def tearDown(self):
self.mapper.delete_all()
self.adapter.connection.close()
def test_insert_inserts_aircraft_and_returns_row_id(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft)
aircrafts = self.mapper.fetch_all()
self.assertEqual(1, len(aircrafts))
self.assertIsNotNone(aircraft_id)
def test_fetch_returns_none_when_aircraft_does_not_exist(self):
aircraft = self.mapper.fetch(1)
self.assertIsNone(aircraft)
def test_fetch_returns_aircraft_when_exists(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft)
aircraft = self.mapper.fetch(aircraft_id)
self.assertIsNotNone(aircraft)
def test_fetch_all_returns_empty_array_when_no_aircrafts_exist(self):
aircrafts = self.mapper.fetch_all()
self.assertEqual([], aircrafts)
def test_fetch_all_returns_all_aircrafts_when_exist(self):
seen_datetime = datetime.utcnow()
self.mapper.insert(
Aircraft([None, 'foo', None, seen_datetime, seen_datetime]))
self.mapper.insert(
Aircraft([None, 'bar', None, seen_datetime, seen_datetime]))
aircrafts = self.mapper.fetch_all()
self.assertEqual(2, len(aircrafts))
def test_update_does_nothing_when_aircraft_does_not_exist(self):
seen_datetime = datetime.utcnow()
aircraft_existing = Aircraft(
[None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft_existing)
aircraft_non_existing = Aircraft(
[0, 'foo', 'bar', seen_datetime, seen_datetime])
self.mapper.update(aircraft_non_existing)
# assert that existing aircraft was not affected
aircraft = self.mapper.fetch(aircraft_id)
self.assertIsNone(aircraft.image)
def test_update_updates_aircraft_when_exists(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft)
aircraft = self.mapper.fetch(aircraft_id)
aircraft.image = 'bar'
self.mapper.update(aircraft)
# assert that image column was updated
aircraft = self.mapper.fetch(aircraft_id)
self.assertEqual('bar', aircraft.image)
def test_delete_does_nothing_when_aircraft_does_not_exist(self):
seen_datetime = datetime.utcnow()
aircraft_existing = Aircraft(
[None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft_existing)
aircraft_non_existing = Aircraft(
[0, 'foo', 'bar', seen_datetime, seen_datetime])
self.mapper.delete(aircraft_non_existing)
# assert that existing aircraft was not affected
aircraft = self.mapper.fetch(aircraft_id)
self.assertIsNotNone(aircraft)
def test_delete_deletes_aircraft_when_exists(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft)
aircraft = self.mapper.fetch(aircraft_id)
self.mapper.delete(aircraft)
aircraft = self.mapper.fetch(aircraft_id)
self.assertIsNone(aircraft)
def test_delete_all_does_nothing_when_no_aircrafts_exist(self):
self.mapper.delete_all()
def test_delete_all_deletes_aircrafts_when_exist(self):
seen_datetime = datetime.utcnow()
aircraft = Aircraft([None, 'foo', None, seen_datetime, seen_datetime])
aircraft_id = self.mapper.insert(aircraft)
self.mapper.delete_all()
aircraft = self.mapper.fetch(aircraft_id)
self.assertIsNone(aircraft)