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.person_dao = PersonDao()
def test_calls_super_init(self, mock_init_method):
PersonDao()
try:
mock_init_method.assert_called_once_with(Person, 'email')
except:
print(
'\nERROR: PersonDao.__init__() should call super().__init__()',
file=sys.stderr)
raise
class PersonDaoTest(TestCase):
"""
Integration tests for PersonDao
"""
# -------------------------------------------------------------------------
# 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.person_dao = PersonDao()
def tearDown(self):
self.session.close_all()
drop_db_tables(db_filename)
# -------------------------------------------------------------------------
# Test Cases
# -------------------------------------------------------------------------
def test_extends_base_dao(self):
self.assertTrue(issubclass(PersonDao, BaseDao),
'PersonDao should be a subclass of BaseDao')
@patch.object(BaseDao, '__init__')
def test_calls_super_init(self, mock_init_method):
PersonDao()
try:
mock_init_method.assert_called_once_with(Person, 'email')
except:
print(
'\nERROR: PersonDao.__init__() should call super().__init__()',
file=sys.stderr)
raise
@patch.object(BaseDao, 'get')
def test_get_calls_superclass_method(self, mock_get_method):
self.person_dao.get('*****@*****.**', self.session)
try:
mock_get_method.assert_called_once_with('*****@*****.**',
self.session)
except:
print('\nERROR: PersonDao should delegate get() call to BaseDao',
file=sys.stderr)
raise
def test_get_person_found(self):
person = self.person_dao.get('*****@*****.**', self.session)
self.assertEqual(101, person.id)
self.assertEqual('Springfield', person.city)
self.assertEqual('USA', person.country)
self.assertEqual('54321', person.post_code)
self.assertEqual('OR', person.state)
self.assertEqual('123 Python St', person.street)
self.assertEqual('*****@*****.**', person.email)
self.assertEqual('Homer', person.first_name)
self.assertEqual('Simpson', person.last_name)
self.assertEqual('Virgil', person.middles)
def test_get_person_not_found(self):
self.assertRaises(PersistenceError, self.person_dao.delete, 999,
self.session)
def test_add_person_ok(self):
person = Person(id=103,
city='Wigan',
country='UK',
post_code='WG7 7FU',
street='62 West Wallaby St',
email='*****@*****.**',
first_name='Wallace',
middles='Peter',
last_name='Sallis')
self.person_dao.add(person, self.session)
self.session.commit()
rows = execute_select(db_filename,
'select * from people where id = 103')
self.assertEqual('Wallace', rows[0][7])
def test_update_person_ok(self):
person = Person(id=103,
city='Wigan',
country='UK',
post_code='WG7 7FU',
street='62 West Wallaby St',
email='*****@*****.**',
first_name='Wallace',
middles='Dwight',
last_name='Schultz')
self.person_dao.update(person, self.session)
self.session.commit()
rows = execute_select(db_filename,
'select * from people where id = 103')
self.assertEqual('Dwight', rows[0][9])
self.assertEqual('Schultz', rows[0][8])
def test_delete_person_found(self):
self.person_dao.delete('*****@*****.**', self.session)
self.session.commit()
rows = execute_select(
db_filename,
"select * from people where email = '*****@*****.**'")
self.assertEqual(0, len(rows))
def test_delete_person_not_found(self):
self.assertRaises(PersistenceError, self.person_dao.delete, 999,
self.session)
# -------------------------------------------------------------------------
# Utility Methods
# -------------------------------------------------------------------------
def populate_db_tables(self):
execute_insert(
db_filename,
'people',
(101, 'Springfield', 'USA', '54321', 'OR', '123 Python St',
'*****@*****.**', 'Homer', 'Simpson', 'Virgil'),
(102, 'Springfield', 'USA', '54321', 'OR', '125 Python St',
'*****@*****.**', 'Ned', 'Flanders', 'Micah'),
)