Remember created project

ihatemoney/tests/budget_test.py

@@ -119,6 +119,16 @@ def test_invite(self):
         resp = self.client.get("/raclette/join/token.invalid", follow_redirects=True)
         self.assertIn("Provided token is invalid", resp.data.decode("utf-8"))
 
+    def test_create_should_remember_project(self):
+        """Test that creating a project adds it to the "logged in project" list,
+        as it does for authentication
+        """
+        self.login("raclette")
+        self.post_project("raclette")
+        self.post_project("tartiflette")
+        data = self.client.get("/raclette/").data.decode("utf-8")
+        self.assertEqual(data.count('href="/tartiflette/"'), 1)
+
     def test_multiple_join(self):
         """Test that joining multiple times a project
         doesn't add it multiple times in the session"""

ihatemoney/web.py

@@ -322,8 +322,7 @@ def create_project():
             db.session.commit()
 
             # create the session object (authenticate)
-            session[project.id] = True
-            session.update()
+            set_authorized_project(project)
 
             # send reminder email
             g.project = project