open-law/tests/test_users.py

147 lines
3.6 KiB
Python

from flask import current_app as app
from flask.testing import FlaskClient, FlaskCliRunner
from click.testing import Result
from werkzeug.datastructures import FileStorage
from app import models as m, db
from tests.utils import login, create_book
def test_create_admin(runner: FlaskCliRunner):
res: Result = runner.invoke(args=["create-admin"])
assert "admin created" in res.output
assert m.User.query.filter_by(username=app.config["ADMIN_USERNAME"]).first()
def test_search_user(populate: FlaskClient, runner: FlaskCliRunner):
_, current_user = login(populate)
MAX_SEARCH_RESULTS = populate.application.config["MAX_SEARCH_RESULTS"]
response = populate.get("/user/search")
assert response.status_code == 422
assert response.json["message"] == "q parameter is required"
q = current_user.username
response = populate.get(f"/user/search?q={q}")
assert response.json
users = response.json.get("users")
assert users
assert len(users) <= MAX_SEARCH_RESULTS
for user in users:
assert q in user["username"]
assert user["username"] != current_user
q = "user"
response = populate.get(f"/user/search?q={q}")
assert response.json
users = response.json.get("users")
assert users
assert len(users) <= MAX_SEARCH_RESULTS
for user in users:
assert q in user["username"]
q = "user1"
response = populate.get(f"/user/search?q={q}")
assert response.json
users = response.json.get("users")
assert users
assert len(users) <= MAX_SEARCH_RESULTS
user = users[0]
assert user["username"] == q
q = "booboo"
response = populate.get(f"/user/search?q={q}")
assert response.json
users = response.json.get("users")
assert not users
# add dummmy data
runner.invoke(args=["db-populate"])
response = populate.get("/user/search?q=dummy&book_id=1")
assert response.json
book_1 = db.session.get(m.Book, 1)
contributors_ids = [contributor.user_id for contributor in book_1.contributors]
users = response.json.get("users")
assert users
for user in users:
user_id = user.get("id")
assert user_id not in contributors_ids
def test_profile(client):
login(client)
book = create_book(client)
user: m.User = m.User(
wallet_id="nsagqklfhqwef84r23hr34r35jfn", password="password"
).save()
assert user
# assert default values
assert user.username
assert not user.is_activated
assert not user.avatar_img
avatar_img = FileStorage(
stream=open("tests/testing_data/1.jpg", "rb"),
filename="1.jpg",
content_type="img/jpg",
)
login(client, username=user.username, password="password")
# edit_profile
new_name = "Some_other_name"
res = client.post(
"/user/edit_profile",
data={
"username": new_name,
"avatar_img": avatar_img,
},
follow_redirects=True,
)
assert res.status_code == 200
assert user.username == new_name
assert user.is_activated
assert user.avatar_img
assert book
# profile page
res = client.get(
"/user/2/profile",
follow_redirects=True,
)
assert res
assert res.status_code == 200
assert str.encode(new_name) in res.data
# delete_avatar
assert user.avatar_img
res = client.post(
"/user/delete_avatar",
follow_redirects=True,
)
assert res
assert not user.avatar_img
# delete_profile
res = client.post(
"/user/profile_delete",
follow_redirects=True,
)
assert res
assert user.is_deleted