def editHotel(hotel_id): logger.info("Got an Edit hotel page request: %s" % request) db = AndrewDB() form = CUHotelForm() form.csrf_enabled = False if current_user.is_hotel_admin(): logger.info("Validating the Create and Update hotel form") if form.validate_on_submit(): img_name = imgName(form.img.data.filename) if img_name: db.insert_location_if_not_exists(form.country.data, form.city.data) old_img = db.get_image_name_by_hotel_id(hotel_id) img_path = '/static/img/hotels/' + img_name db.update_hotel_by_id(hotel_id, form.city.data, form.address.data, form.hotel_name.data, form.stars.data, form.description.data, img_path) form.img.data.save( os.path.join(app.config['UPLOAD_FOLDER'], img_name)) os.remove(os.path.abspath('app' + old_img)) return redirect(url_for('myHotels')) res = db.get_hotel_and_address_by_id(hotel_id) logger.info("Rendering the Edit hotel page") return render_template('edit_hotel.html', form=form, hotel=res) else: flash("Access error") logger.info("Access error, Redirecting to login page") return redirect(url_for('login'))
def test_get_image_name_by_hotel_id(mock_connect): with allure.step('Get hotel image'): with app.app_context(): db = AndrewDB() expected = {'img': "noname.png"} mock_connect().cursor.return_value.fetchone.return_value = expected result = db.get_image_name_by_hotel_id({}) assert result == expected['img']