simple-login/app
1
1
Fork 0
mirror of https://github.com/simple-login/app.git synced 2025-02-25 08:13:16 +08:00
Code Issues Projects Releases Packages Wiki Activity

refactor test

Browse source
This commit is contained in:
Son NK 2021-03-24 15:54:03 +01:00
parent 84b4c11086
commit 10cc61b4a0
1 changed files with 5 additions and 31 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
tests/api/test_new_random_alias.py
View file

@ -3,23 +3,15 @@ import uuid
from flask import url_for from flask import url_for
from app.config import EMAIL_DOMAIN, MAX_NB_EMAIL_FREE_PLAN from app.config import EMAIL_DOMAIN, MAX_NB_EMAIL_FREE_PLAN
from app.extensions import db from app.models import Alias
from app.models import User, ApiKey, Alias from tests.utils import login
def test_success(flask_client): def test_success(flask_client):
user = User.create( login(flask_client)
email="a@b.c", password="password", name="Test User", activated=True
)
db.session.commit()
# create api_key
api_key = ApiKey.create(user.id, "for test")
db.session.commit()
r = flask_client.post( r = flask_client.post(
url_for("api.new_random_alias", hostname="www.test.com"), url_for("api.new_random_alias", hostname="www.test.com"),
headers={"Authentication": api_key.code},
) )
assert r.status_code == 201 assert r.status_code == 201
@ -39,19 +31,11 @@ def test_success(flask_client):
def test_custom_mode(flask_client): def test_custom_mode(flask_client):
user = User.create( login(flask_client)
email="a@b.c", password="password", name="Test User", activated=True
)
db.session.commit()
# create api_key
api_key = ApiKey.create(user.id, "for test")
db.session.commit()
# without note # without note
r = flask_client.post( r = flask_client.post(
url_for("api.new_random_alias", hostname="www.test.com", mode="uuid"), url_for("api.new_random_alias", hostname="www.test.com", mode="uuid"),
headers={"Authentication": api_key.code},
) )
assert r.status_code == 201 assert r.status_code == 201
@ -63,7 +47,6 @@ def test_custom_mode(flask_client):
# with note # with note
r = flask_client.post( r = flask_client.post(
url_for("api.new_random_alias", hostname="www.test.com", mode="uuid"), url_for("api.new_random_alias", hostname="www.test.com", mode="uuid"),
headers={"Authentication": api_key.code},
json={"note": "test note"}, json={"note": "test note"},
) )
@ -74,15 +57,7 @@ def test_custom_mode(flask_client):
def test_out_of_quota(flask_client): def test_out_of_quota(flask_client):
user = User.create( user = login(flask_client)
email="a@b.c", password="password", name="Test User", activated=True
)
user.trial_end = None
db.session.commit()
# create api_key
api_key = ApiKey.create(user.id, "for test")
db.session.commit()
# create MAX_NB_EMAIL_FREE_PLAN random alias to run out of quota # create MAX_NB_EMAIL_FREE_PLAN random alias to run out of quota
for _ in range(MAX_NB_EMAIL_FREE_PLAN): for _ in range(MAX_NB_EMAIL_FREE_PLAN):
@ -90,7 +65,6 @@ def test_out_of_quota(flask_client):
r = flask_client.post( r = flask_client.post(
url_for("api.new_random_alias", hostname="www.test.com"), url_for("api.new_random_alias", hostname="www.test.com"),
headers={"Authentication": api_key.code},
) )
assert r.status_code == 400 assert r.status_code == 400