Root url should redirect if user is already logged in

atst/app.py

@@ -67,6 +67,7 @@ def make_app(config):
 def make_flask_callbacks(app):
     @app.before_request
     def _set_globals():
+        g.current_user = None
         g.dev = os.getenv("FLASK_ENV", "dev") == "dev"
         g.matchesPath = lambda href: re.match("^" + href, request.path)
         g.modal = request.args.get("modal", None)
@@ -74,7 +75,7 @@ def make_flask_callbacks(app):
 
     @app.after_request
     def _cleanup(response):
-        g.pop("current_user", None)
+        g.current_user = None
         return response
 
 

atst/routes/__init__.py

@@ -25,6 +25,9 @@ def root():
             "?{}".format(url.urlencode({"next": request.args.get("next")})),
         )
 
+    if g.current_user:
+        return redirect(url_for(".home"))
+
     return render_template(
         "login.html", redirect=bool(request.args.get("next")), redirect_url=redirect_url
     )

tests/routes/test_root.py

@@ -0,0 +1,6 @@
+from tests.factories import UserFactory
+
+def test_root_redirects_if_user_is_logged_in(client, user_session):
+    user_session(UserFactory.create())
+    response = client.get("/", follow_redirects=False)
+    assert "home" in response.location

tests/test_routes.py

@@ -4,7 +4,6 @@ import pytest
 @pytest.mark.parametrize(
     "path",
     (
-        "/",
         "/workspaces",
         "/requests",
         "/requests/new/1",