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')