mirror of https://github.com/logos-co/open-law.git
147 lines
3.6 KiB
Python
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
|