open-law/tests/test_users.py

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