selfprivacy-rest-api/tests/test_rest_endpoints/services/test_pleroma.py

import json
import pytest


def read_json(file_path):
    with open(file_path, "r") as f:
        return json.load(f)


###############################################################################


@pytest.fixture
def pleroma_off(mocker, datadir):
    mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
    assert read_json(datadir / "turned_off.json")["pleroma"]["enable"] == False
    return datadir


@pytest.fixture
def pleroma_on(mocker, datadir):
    mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
    assert read_json(datadir / "turned_on.json")["pleroma"]["enable"] == True
    return datadir


@pytest.fixture
def pleroma_enable_undefined(mocker, datadir):
    mocker.patch(
        "selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
    )
    assert "enable" not in read_json(datadir / "enable_undefined.json")["pleroma"]
    return datadir


@pytest.fixture
def pleroma_undefined(mocker, datadir):
    mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
    assert "pleroma" not in read_json(datadir / "undefined.json")
    return datadir


###############################################################################


@pytest.mark.parametrize("endpoint", ["enable", "disable"])
def test_unauthorized(client, pleroma_off, endpoint):
    response = client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 401


@pytest.mark.parametrize("endpoint", ["enable", "disable"])
def test_illegal_methods(authorized_client, pleroma_off, endpoint):
    response = authorized_client.get(f"/services/pleroma/{endpoint}")
    assert response.status_code == 405
    response = authorized_client.put(f"/services/pleroma/{endpoint}")
    assert response.status_code == 405
    response = authorized_client.delete(f"/services/pleroma/{endpoint}")
    assert response.status_code == 405


@pytest.mark.parametrize(
    "endpoint,target_file",
    [("enable", "turned_on.json"), ("disable", "turned_off.json")],
)
def test_switch_from_off(authorized_client, pleroma_off, endpoint, target_file):
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    assert read_json(pleroma_off / "turned_off.json") == read_json(
        pleroma_off / target_file
    )


@pytest.mark.parametrize(
    "endpoint,target_file",
    [("enable", "turned_on.json"), ("disable", "turned_off.json")],
)
def test_switch_from_on(authorized_client, pleroma_on, endpoint, target_file):
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    assert read_json(pleroma_on / "turned_on.json") == read_json(
        pleroma_on / target_file
    )


@pytest.mark.parametrize(
    "endpoint,target_file",
    [("enable", "turned_on.json"), ("disable", "turned_off.json")],
)
def test_switch_twice(authorized_client, pleroma_off, endpoint, target_file):
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    assert read_json(pleroma_off / "turned_off.json") == read_json(
        pleroma_off / target_file
    )


@pytest.mark.parametrize(
    "endpoint,target_file",
    [("enable", "turned_on.json"), ("disable", "turned_off.json")],
)
def test_on_attribute_deleted(
    authorized_client, pleroma_enable_undefined, endpoint, target_file
):
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    assert read_json(pleroma_enable_undefined / "enable_undefined.json") == read_json(
        pleroma_enable_undefined / target_file
    )


@pytest.mark.parametrize(
    "endpoint,target_file",
    [("enable", "turned_on.json"), ("disable", "turned_off.json")],
)
def test_on_pleroma_undefined(
    authorized_client, pleroma_undefined, endpoint, target_file
):
    response = authorized_client.post(f"/services/pleroma/{endpoint}")
    assert response.status_code == 200
    assert read_json(pleroma_undefined / "undefined.json") == read_json(
        pleroma_undefined / target_file
    )