fix error redirection on import

This commit is contained in:
Youe Graillot 2021-12-13 04:31:04 +01:00
parent 99a78f86fb
commit 946fa3e8e9
2 changed files with 4 additions and 8 deletions

View file

@ -74,13 +74,13 @@ class BaseTestCase(TestCase):
follow_redirects=follow_redirects,
)
def import_project(self, id, data, expected_code=200):
def import_project(self, id, data, success=True):
resp = self.client.post(
f"/{id}/import",
data=data,
follow_redirects=True,
# follow_redirects=True,
)
self.assertEqual(resp.status_code, expected_code)
self.assertEqual("/{id}/edit" in str(resp.response), not success)
def create_project(self, id, default_currency="XXX", name=None, password=None):
name = name or str(id)

View file

@ -444,7 +444,6 @@ def edit_project():
@main.route("/<project_id>/import", methods=["POST"])
def import_project():
form = ImportProjectForm()
result = 200
if form.validate():
try:
data = form.file.data
@ -497,13 +496,10 @@ def import_project():
return redirect(url_for("main.list_bills"))
except ValueError as b:
flash(b.args[0], category="danger")
result = 400
else:
for component, errors in form.errors.items():
flash(_(component + ": ") + ", ".join(errors), category="danger")
result = 400
return redirect(url_for(".edit_project"), code=result)
return redirect(request.headers.get("Referer") or url_for(".edit_project"))
@main.route("/<project_id>/delete", methods=["POST"])