From 217518c00e7afc2a850afd4f6090ce0c6853e41f Mon Sep 17 00:00:00 2001
From: Son <nguyenkims@users.noreply.github.com>
Date: Thu, 14 Apr 2022 18:37:55 +0200
Subject: [PATCH] refactor

---
 tests/api/test_alias_options.py | 21 ++++++---------------
 1 file changed, 6 insertions(+), 15 deletions(-)

diff --git a/tests/api/test_alias_options.py b/tests/api/test_alias_options.py
index 6bd3c037..dfe41511 100644
--- a/tests/api/test_alias_options.py
+++ b/tests/api/test_alias_options.py
@@ -2,6 +2,7 @@ from flask import url_for
 
 from app.db import Session
 from app.models import User, ApiKey, AliasUsedOn, Alias
+from tests.utils import login
 
 
 def test_different_scenarios_v4(flask_client):
@@ -98,18 +99,11 @@ def test_different_scenarios_v4_2(flask_client):
 
 
 def test_different_scenarios_v5(flask_client):
-    user = User.create(
-        email="a@b.c", password="password", name="Test User", activated=True
-    )
-    Session.commit()
-
-    # create api_key
-    api_key = ApiKey.create(user.id, "for test")
-    Session.commit()
+    user = login(flask_client)
 
     # <<< without hostname >>>
     r = flask_client.get(
-        "/api/v5/alias/options", headers={"Authentication": api_key.code}
+        "/api/v5/alias/options"
     )
 
     assert r.status_code == 200
@@ -129,15 +123,13 @@ def test_different_scenarios_v5(flask_client):
 
     # <<< with hostname >>>
     r = flask_client.get(
-        "/api/v5/alias/options?hostname=www.test.com",
-        headers={"Authentication": api_key.code},
+        "/api/v5/alias/options?hostname=www.test.com"
     )
     assert r.json["prefix_suggestion"] == "test"
 
     # <<< with hostname with 2 parts TLD, for example wwww.numberoneshoes.co.nz >>>
     r = flask_client.get(
-        "/api/v5/alias/options?hostname=wwww.numberoneshoes.co.nz",
-        headers={"Authentication": api_key.code},
+        "/api/v5/alias/options?hostname=wwww.numberoneshoes.co.nz"
     )
     assert r.json["prefix_suggestion"] == "numberoneshoes"
 
@@ -150,8 +142,7 @@ def test_different_scenarios_v5(flask_client):
     Session.commit()
 
     r = flask_client.get(
-        url_for("api.options_v4", hostname="www.test.com"),
-        headers={"Authentication": api_key.code},
+        url_for("api.options_v4", hostname="www.test.com")
     )
     assert r.json["recommendation"]["alias"] == alias.email
     assert r.json["recommendation"]["hostname"] == "www.test.com"