def setUp(self):
create_db_tables(db_filename)
self.populate_db_tables()
settings = {'sqlalchemy.url': 'sqlite:///' + db_filename}
# Create the SQLAlchemy DB Engine
engine = engine_from_config(settings, 'sqlalchemy.')
Session = sessionmaker(bind=engine)
self.session = Session()
self.venue_dao = VenueDao()
def setUp(self):
if os.path.exists(db_filename):
os.remove(db_filename)
create_db_tables(db_filename)
self.populate_db_tables()
settings = {'sqlalchemy.url': 'sqlite:///' + db_filename}
# Create the SQLAlchemy DB Engine
engine = engine_from_config(settings, 'sqlalchemy.')
Session = sessionmaker(bind=engine)
self.session = Session()
self.venue_dao = VenueDao()
class VenueDaoTest(TestCase):
"""
Integration tests for VenueDao
"""
# -------------------------------------------------------------------------
# Set Up and Tear Down Methods
# -------------------------------------------------------------------------
def setUp(self):
if os.path.exists(db_filename):
os.remove(db_filename)
create_db_tables(db_filename)
self.populate_db_tables()
settings = {'sqlalchemy.url': 'sqlite:///' + db_filename}
# Create the SQLAlchemy DB Engine
engine = engine_from_config(settings, 'sqlalchemy.')
Session = sessionmaker(bind=engine)
self.session = Session()
self.venue_dao = VenueDao()
def tearDown(self):
self.session.close_all()
drop_db_tables(db_filename)
# -------------------------------------------------------------------------
# Test Cases
# -------------------------------------------------------------------------
def test_extends_base_dao(self):
self.assertTrue(issubclass(VenueDao, BaseDao),
'VenueDao should be a subclass of BaseDao')
@patch.object(BaseDao, '__init__')
def test_calls_super_init(self, mock_init_method):
VenueDao()
try:
mock_init_method.assert_called_once_with(Venue, 'id')
except:
print('\nERROR: Venue.__init__() should call super().__init__()',
file=sys.stderr)
raise
@patch.object(BaseDao, 'get')
def test_get_calls_superclass_method(self, mock_get_method):
self.venue_dao.get('*****@*****.**', self.session)
try:
mock_get_method.assert_called_once_with('*****@*****.**',
self.session)
except:
print('\nERROR: Venue should delegate get() call to BaseDao',
file=sys.stderr)
raise
def test_get_venue_found(self):
venue = self.venue_dao.get(101, self.session)
self.assertEqual(101, venue.id)
self.assertEqual('Chicago', venue.city)
self.assertEqual('USA', venue.country)
self.assertEqual(41.8369, venue.latitude)
self.assertEqual(-87.6847, venue.longitude)
self.assertEqual('Auditorium Theatre', venue.name)
self.assertEqual('IL', venue.prov_state)
self.assertEqual('E Congress Pkwy, Chicago, IL 60605',
venue.street_address)
self.assertEqual(2, len(venue.events))
self.assertEqual(201, venue.events[0].id)
self.assertEqual(202, venue.events[1].id)
def test_get_venue_by_name_found(self):
venue = self.venue_dao.get_venue_by_name('Auditorium Theatre',
self.session)
self.assertEqual(101, venue.id)
self.assertEqual('Chicago', venue.city)
self.assertEqual('USA', venue.country)
self.assertEqual(41.8369, venue.latitude)
self.assertEqual(-87.6847, venue.longitude)
self.assertEqual('Auditorium Theatre', venue.name)
self.assertEqual('IL', venue.prov_state)
self.assertEqual('E Congress Pkwy, Chicago, IL 60605',
venue.street_address)
self.assertEqual(2, len(venue.events))
self.assertEqual(201, venue.events[0].id)
self.assertEqual(202, venue.events[1].id)
def test_get_venue_and_events_found(self):
venue = self.venue_dao.get_venue_and_events(101, self.session)
self.assertEqual(101, venue.id)
self.assertEqual('Chicago', venue.city)
self.assertEqual('USA', venue.country)
self.assertEqual(41.8369, venue.latitude)
self.assertEqual(-87.6847, venue.longitude)
self.assertEqual('Auditorium Theatre', venue.name)
self.assertEqual('IL', venue.prov_state)
self.assertEqual('E Congress Pkwy, Chicago, IL 60605',
venue.street_address)
self.assertEqual(2, len(venue.events))
self.assertEqual(201, venue.events[0].id)
self.assertEqual(202, venue.events[1].id)
def test_get_venue_not_found(self):
self.assertRaises(PersistenceError, self.venue_dao.delete, 999,
self.session)
def test_add_venue_ok(self):
venue = Venue(id=678, city='London', country='UK', latitude=51.5072,
longitude=-0.1275, name='Royal Albert Hall',
street_address='Kensington Gore, London SW7 2AP')
self.venue_dao.add(venue, self.session)
self.session.commit()
rows = execute_select(db_filename,
'select * from venues where id = 678')
self.assertEqual('Royal Albert Hall', rows[0][5])
def test_update_venue_ok(self):
venue = Venue(id=678, city='London', country='United Kingdom',
latitude=51.5072, longitude=-0.1275,
name='Royal Albert Hall',
street_address='Kensington Gore, London SW7 2AP')
self.venue_dao.update(venue, self.session)
self.session.commit()
rows = execute_select(db_filename,
'select * from venues where id = 678')
self.assertEqual('United Kingdom', rows[0][2])
def test_delete_venue_found(self):
self.venue_dao.delete(101, self.session)
self.session.commit()
rows = execute_select(db_filename,
'select * from venues where id = 101')
self.assertEqual(0, len(rows))
def test_delete_venue_not_found(self):
self.assertRaises(PersistenceError, self.venue_dao.delete, 999,
self.session)
# -------------------------------------------------------------------------
# Utility Methods
# -------------------------------------------------------------------------
def populate_db_tables(self):
execute_insert(db_filename, 'events',
(201, '2015-12-31 20:00:00.000', 101, None),
(202, '2016-01-01 20:00:00.000', 101, None),
(203, '2015-12-31 21:00:00.000', 102, None),
(204, '2015-12-25 10:00:00.000', 102, None)
)
execute_insert(db_filename, 'venues',
(101, 'Chicago', 'USA', 41.8369, -87.6847, 'Auditorium Theatre',
'IL', 'E Congress Pkwy, Chicago, IL 60605'),
(102, 'New York', 'USA', 40.7127, -74.0059, 'Carnegie Hall',
'NY', '881 7th Ave, New York, NY 10019')
)