diff --git a/tests/test_app.py b/tests/test_app.py
index 1edeba1..25b4976 100644
--- a/tests/test_app.py
+++ b/tests/test_app.py
@@ -16,15 +16,30 @@ app.config["DISCORD_REDIRECT_URI"] = "http://127.0.0.1:5000/callback"
discord = DiscordOAuth2Session(app)
+HYPERLINK = '{}'
+
+
@app.route("/")
def index():
+ if not discord.authorized:
+ return HYPERLINK.format(url_for(".login"), "Login")
+ return f"""
+ {HYPERLINK.format(url_for(".me"), "@ME")}
+ {HYPERLINK.format(url_for(".logout"), "Logout")}
+ {HYPERLINK.format(url_for(".user_guilds"), "My Servers")}
+ {HYPERLINK.format(url_for(".add_to_guild", guild_id=475549041741135881), "Add bot to 475549041741135881.")}
+ """
+
+
+@app.route("/login/")
+def login():
return discord.create_session()
@app.route("/callback/")
def callback():
discord.callback()
- return redirect(url_for(".me"))
+ return redirect(url_for(".index"))
@app.route("/me/")