def delete_wall() -> str: wall_id = request.args.get("wall_id") form = WarrantyForm() if form.validate_on_submit(): if form.no.data: flash("Wall has not been deleted.") next_page = request.args.get("next_page") if not next_page: next_page = url_for("masonry_works.walls") return redirect(next_page) elif form.yes.data: Wall.delete_wall(wall_id) flash("Wall has been deleted.") return redirect(url_for("masonry_works.walls")) return render_template( "warranty_form.html", title="Delete Wall", form=form, )
def test_delete_wall_when_no_wall(app_and_db): assert not Wall.query.filter_by(id=1).first() Wall.delete_wall(1)
def test_delete_wall(add_wall): Wall.add_hole(wall_id=1, width=1, height=2, amonunt=1) Wall.add_processing(wall_id=1, year=2020, month="December", done=0.5) assert Wall.query.filter_by(id=1).first() Wall.delete_wall(1) assert not Wall.query.filter_by(id=1).first()