Add missing tests. Fixes #14 and #22.

1 files changed, 8 insertions, 0 deletions

diff --git a/budget/web.py b/budget/web.py
index 78886d8..c69189e 100644
--- a/budget/web.py
+++ b/budget/web.py
@@ -173,6 +173,14 @@ def add_member():
     form = MemberForm(g.project)
     if request.method == "POST":
         if form.validate():
+            # if the user is already bound to the project, just reactivate him
+            person = Person.query.filter(Person.name == form.name.data)\
+                        .filter(Project.id == g.project.id).all()
+            if person:
+                person[0].activated = True
+                db.session.commit()
+                return redirect(url_for("list_bills"))
+
             db.session.add(Person(name=form.name.data, project=g.project))
             db.session.commit()
             return redirect(url_for("list_bills"))