def test_get_session_inner_function_calls(self, mock_delarative_base,
mock_sessionmaker,
mock_create_engine):
# pdb.set_trace()
# mock_create_engine.return_value = sentinel.engine
# mock_session = mock_sessionmaker.return_value.return_value
# self.assertEqual(mock_session, get_session('any_path'))
# assert mock_session.connection.called
self.assertEqual(get_session('any_path'), mock_sessionmaker.return_value.return_value)
def show_shelter_details(shelter_id):
if request.method == 'GET':
pdb.set_trace()
session = get_session(test_db_string)
shelter = session.query(Shelter).filter_by(shelter_id=shelter_id).one()
if shelter is not None:
render_template('editdelete_shelter.html', hasValue=True,
shelter=shelter)
else:
render_template('editdelete_shelter.html', hasValue=False)
if request.method == 'POST': # edit
pass
if request.method == 'DELETE':
pass
def add_shelter():
if request.method == 'POST':
# print 'I got POST response' # protect code from sql and code injection
name = bleach.clean(request.form['shelter_name'])
street = bleach.clean(request.form['shelter_street'])
city = bleach.clean(request.form['shelter_city'])
state = bleach.clean(request.form['shelter_state'])
zipcode = int(bleach.clean((request.form['shelter_zipcode'])))
website = bleach.clean(request.form['shelter_website'])
email = bleach.clean(request.form['shelter_email'])
current_occupancy = int(bleach.clean(request.form['shelter_curroccupancy']))
max_occupancy = int(bleach.clean(request.form['shelter_maxoccupancy']))
shelter = Shelter(name, street, city, state, zipcode, website,
email, current_occupancy, max_occupancy)
session = get_session(test_db_string)
session.add(shelter)
# Get that object
session.commit()
return redirect(url_for('show_all_shelters'))
else:
return render_template('add_shelter.html')
def add_shelter():
if request.method == 'POST':
# print 'I got POST response' # protect code from sql and code injection
name = bleach.clean(request.form['shelter_name'])
street = bleach.clean(request.form['shelter_street'])
city = bleach.clean(request.form['shelter_city'])
state = bleach.clean(request.form['shelter_state'])
zipcode = int(bleach.clean((request.form['shelter_zipcode'])))
website = bleach.clean(request.form['shelter_website'])
email = bleach.clean(request.form['shelter_email'])
current_occupancy = int(
bleach.clean(request.form['shelter_curroccupancy']))
max_occupancy = int(bleach.clean(request.form['shelter_maxoccupancy']))
shelter = Shelter(name, street, city, state, zipcode, website, email,
current_occupancy, max_occupancy)
session = get_session(test_db_string)
session.add(shelter)
# Get that object
session.commit()
return redirect(url_for('show_all_shelters'))
else:
return render_template('add_shelter.html')
def test_get_session_for_exception_with_different_connection_strings(self):
self.assertRaises(OperationalError,
get_session(self.invalid_connection_str))
self.assertRaises(ArgumentError, get_session(self.unparsable_db_str))
self.assertIsNotNone(get_session(self.valid_connection_str),
'Session object cannot be None')
def test_get_session_for_exception_with_different_connection_strings(self):
self.assertRaises(OperationalError, get_session(
self.invalid_connection_str))
self.assertRaises(ArgumentError, get_session(self.unparsable_db_str))
self.assertIsNotNone(get_session(self.valid_connection_str),
'Session object cannot be None')
