mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-11-29 15:31:28 +00:00
Reformat code with Black
This commit is contained in:
parent
8e0a5f5ec0
commit
69eb91a1ea
|
@ -147,23 +147,26 @@ class AsyncRestoreBackup(Resource):
|
||||||
401:
|
401:
|
||||||
description: Unauthorized
|
description: Unauthorized
|
||||||
"""
|
"""
|
||||||
backup_restoration_command = ["restic", "-r", "rclone:backblaze:sfbackup", "var", "--json"]
|
backup_restoration_command = [
|
||||||
|
"restic",
|
||||||
|
"-r",
|
||||||
|
"rclone:backblaze:sfbackup",
|
||||||
|
"var",
|
||||||
|
"--json",
|
||||||
|
]
|
||||||
|
|
||||||
with open("/tmp/backup.log", "w", encoding="utf-8") as backup_log_file_descriptor:
|
with open(
|
||||||
|
"/tmp/backup.log", "w", encoding="utf-8"
|
||||||
|
) as backup_log_file_descriptor:
|
||||||
with subprocess.Popen(
|
with subprocess.Popen(
|
||||||
backup_restoration_command,
|
backup_restoration_command,
|
||||||
shell=False,
|
shell=False,
|
||||||
stdout=subprocess.PIPE,
|
stdout=subprocess.PIPE,
|
||||||
stderr=backup_log_file_descriptor,
|
stderr=backup_log_file_descriptor,
|
||||||
) as backup_restoration_process_descriptor:
|
) as backup_restoration_process_descriptor:
|
||||||
backup_restoration_status = (
|
backup_restoration_status = "Backup restoration procedure started"
|
||||||
"Backup restoration procedure started"
|
|
||||||
)
|
|
||||||
|
|
||||||
return {
|
return {"status": 0, "message": backup_restoration_status}
|
||||||
"status": 0,
|
|
||||||
"message": backup_restoration_status
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
api.add_resource(ListAllBackups, "/restic/backup/list")
|
api.add_resource(ListAllBackups, "/restic/backup/list")
|
||||||
|
|
|
@ -32,28 +32,24 @@ class PullRepositoryChanges(Resource):
|
||||||
current_working_directory = os.getcwd()
|
current_working_directory = os.getcwd()
|
||||||
os.chdir("/etc/nixos")
|
os.chdir("/etc/nixos")
|
||||||
|
|
||||||
|
|
||||||
git_pull_process_descriptor = subprocess.Popen(
|
git_pull_process_descriptor = subprocess.Popen(
|
||||||
git_pull_command,
|
git_pull_command,
|
||||||
stdout=subprocess.PIPE,
|
stdout=subprocess.PIPE,
|
||||||
stderr=subprocess.STDOUT,
|
stderr=subprocess.STDOUT,
|
||||||
shell=False
|
shell=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
git_pull_process_descriptor.communicate()[0]
|
git_pull_process_descriptor.communicate()[0]
|
||||||
|
|
||||||
os.chdir(current_working_directory)
|
os.chdir(current_working_directory)
|
||||||
|
|
||||||
if git_pull_process_descriptor.returncode == 0:
|
if git_pull_process_descriptor.returncode == 0:
|
||||||
return {
|
return {"status": 0, "message": "Update completed successfully"}
|
||||||
"status": 0,
|
|
||||||
"message": "Update completed successfully"
|
|
||||||
}
|
|
||||||
elif git_pull_process_descriptor.returncode > 0:
|
elif git_pull_process_descriptor.returncode > 0:
|
||||||
return {
|
return {
|
||||||
"status": git_pull_process_descriptor.returncode,
|
"status": git_pull_process_descriptor.returncode,
|
||||||
"message": "Something went wrong"
|
"message": "Something went wrong",
|
||||||
}, 500
|
}, 500
|
||||||
|
|
||||||
|
|
||||||
api.add_resource(PullRepositoryChanges, "/update")
|
api.add_resource(PullRepositoryChanges, "/update")
|
||||||
|
|
|
@ -7,6 +7,7 @@ from flask import current_app
|
||||||
|
|
||||||
USERDATA_FILE = "/etc/nixos/userdata/userdata.json"
|
USERDATA_FILE = "/etc/nixos/userdata/userdata.json"
|
||||||
|
|
||||||
|
|
||||||
def get_domain():
|
def get_domain():
|
||||||
"""Get domain from /var/domain without trailing new line"""
|
"""Get domain from /var/domain without trailing new line"""
|
||||||
with open("/var/domain", "r", encoding="utf-8") as domain_file:
|
with open("/var/domain", "r", encoding="utf-8") as domain_file:
|
||||||
|
@ -18,9 +19,7 @@ class WriteUserData(object):
|
||||||
"""Write userdata.json with lock"""
|
"""Write userdata.json with lock"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.userdata_file = open(
|
self.userdata_file = open(USERDATA_FILE, "r+", encoding="utf-8")
|
||||||
USERDATA_FILE, "r+", encoding="utf-8"
|
|
||||||
)
|
|
||||||
portalocker.lock(self.userdata_file, portalocker.LOCK_EX)
|
portalocker.lock(self.userdata_file, portalocker.LOCK_EX)
|
||||||
self.data = json.load(self.userdata_file)
|
self.data = json.load(self.userdata_file)
|
||||||
|
|
||||||
|
@ -40,9 +39,7 @@ class ReadUserData(object):
|
||||||
"""Read userdata.json with lock"""
|
"""Read userdata.json with lock"""
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.userdata_file = open(
|
self.userdata_file = open(USERDATA_FILE, "r", encoding="utf-8")
|
||||||
USERDATA_FILE, "r", encoding="utf-8"
|
|
||||||
)
|
|
||||||
portalocker.lock(self.userdata_file, portalocker.LOCK_SH)
|
portalocker.lock(self.userdata_file, portalocker.LOCK_SH)
|
||||||
self.data = json.load(self.userdata_file)
|
self.data = json.load(self.userdata_file)
|
||||||
|
|
||||||
|
@ -60,4 +57,3 @@ def validate_ssh_public_key(key):
|
||||||
if not key.startswith("ssh-rsa"):
|
if not key.startswith("ssh-rsa"):
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
|
@ -5,10 +5,12 @@ from selfprivacy_api.app import create_app
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def app():
|
def app():
|
||||||
app = create_app({
|
app = create_app(
|
||||||
|
{
|
||||||
"AUTH_TOKEN": "TEST_TOKEN",
|
"AUTH_TOKEN": "TEST_TOKEN",
|
||||||
"ENABLE_SWAGGER": "0",
|
"ENABLE_SWAGGER": "0",
|
||||||
})
|
}
|
||||||
|
)
|
||||||
|
|
||||||
yield app
|
yield app
|
||||||
|
|
||||||
|
@ -17,6 +19,7 @@ def app():
|
||||||
def client(app):
|
def client(app):
|
||||||
return app.test_client()
|
return app.test_client()
|
||||||
|
|
||||||
|
|
||||||
class AuthorizedClient(testing.FlaskClient):
|
class AuthorizedClient(testing.FlaskClient):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super().__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
|
@ -28,11 +31,13 @@ class AuthorizedClient(testing.FlaskClient):
|
||||||
kwargs["headers"]["Authorization"] = f"Bearer {self.token}"
|
kwargs["headers"]["Authorization"] = f"Bearer {self.token}"
|
||||||
return super().open(*args, **kwargs)
|
return super().open(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def authorized_client(app):
|
def authorized_client(app):
|
||||||
app.test_client_class = AuthorizedClient
|
app.test_client_class = AuthorizedClient
|
||||||
return app.test_client()
|
return app.test_client()
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def runner(app):
|
def runner(app):
|
||||||
return app.test_cli_runner()
|
return app.test_cli_runner()
|
|
@ -1,43 +1,54 @@
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def bitwarden_off(mocker, datadir):
|
def bitwarden_off(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
||||||
assert read_json(datadir / "turned_off.json")["bitwarden"]["enable"] == False
|
assert read_json(datadir / "turned_off.json")["bitwarden"]["enable"] == False
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def bitwarden_on(mocker, datadir):
|
def bitwarden_on(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
||||||
assert read_json(datadir / "turned_on.json")["bitwarden"]["enable"] == True
|
assert read_json(datadir / "turned_on.json")["bitwarden"]["enable"] == True
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def bitwarden_enable_undefined(mocker, datadir):
|
def bitwarden_enable_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
|
||||||
|
)
|
||||||
assert "enable" not in read_json(datadir / "enable_undefined.json")["bitwarden"]
|
assert "enable" not in read_json(datadir / "enable_undefined.json")["bitwarden"]
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def bitwarden_undefined(mocker, datadir):
|
def bitwarden_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||||||
assert "bitwarden" not in read_json(datadir / "undefined.json")
|
assert "bitwarden" not in read_json(datadir / "undefined.json")
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_unauthorized(client, bitwarden_off, endpoint):
|
def test_unauthorized(client, bitwarden_off, endpoint):
|
||||||
response = client.post(f"/services/bitwarden/{endpoint}")
|
response = client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_illegal_methods(authorized_client, bitwarden_off, endpoint):
|
def test_illegal_methods(authorized_client, bitwarden_off, endpoint):
|
||||||
response = authorized_client.get(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.get(f"/services/bitwarden/{endpoint}")
|
||||||
|
@ -47,34 +58,68 @@ def test_illegal_methods(authorized_client, bitwarden_off, endpoint):
|
||||||
response = authorized_client.delete(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.delete(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_off(authorized_client, bitwarden_off, endpoint, target_file):
|
def test_switch_from_off(authorized_client, bitwarden_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(bitwarden_off / "turned_off.json") == read_json(bitwarden_off / target_file)
|
assert read_json(bitwarden_off / "turned_off.json") == read_json(
|
||||||
|
bitwarden_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_on(authorized_client, bitwarden_on, endpoint, target_file):
|
def test_switch_from_on(authorized_client, bitwarden_on, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(bitwarden_on / "turned_on.json") == read_json(bitwarden_on / target_file)
|
assert read_json(bitwarden_on / "turned_on.json") == read_json(
|
||||||
|
bitwarden_on / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_twice(authorized_client, bitwarden_off, endpoint, target_file):
|
def test_switch_twice(authorized_client, bitwarden_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(bitwarden_off / "turned_off.json") == read_json(bitwarden_off / target_file)
|
assert read_json(bitwarden_off / "turned_off.json") == read_json(
|
||||||
|
bitwarden_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_attribute_deleted(authorized_client, bitwarden_enable_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_attribute_deleted(
|
||||||
|
authorized_client, bitwarden_enable_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(bitwarden_enable_undefined / "enable_undefined.json") == read_json(bitwarden_enable_undefined / target_file)
|
assert read_json(bitwarden_enable_undefined / "enable_undefined.json") == read_json(
|
||||||
|
bitwarden_enable_undefined / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_bitwarden_undefined(authorized_client, bitwarden_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_bitwarden_undefined(
|
||||||
|
authorized_client, bitwarden_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
response = authorized_client.post(f"/services/bitwarden/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(bitwarden_undefined / "undefined.json") == read_json(bitwarden_undefined / target_file)
|
assert read_json(bitwarden_undefined / "undefined.json") == read_json(
|
||||||
|
bitwarden_undefined / target_file
|
||||||
|
)
|
||||||
|
|
|
@ -1,43 +1,54 @@
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def gitea_off(mocker, datadir):
|
def gitea_off(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
||||||
assert read_json(datadir / "turned_off.json")["gitea"]["enable"] == False
|
assert read_json(datadir / "turned_off.json")["gitea"]["enable"] == False
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def gitea_on(mocker, datadir):
|
def gitea_on(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
||||||
assert read_json(datadir / "turned_on.json")["gitea"]["enable"] == True
|
assert read_json(datadir / "turned_on.json")["gitea"]["enable"] == True
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def gitea_enable_undefined(mocker, datadir):
|
def gitea_enable_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
|
||||||
|
)
|
||||||
assert "enable" not in read_json(datadir / "enable_undefined.json")["gitea"]
|
assert "enable" not in read_json(datadir / "enable_undefined.json")["gitea"]
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def gitea_undefined(mocker, datadir):
|
def gitea_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||||||
assert "gitea" not in read_json(datadir / "undefined.json")
|
assert "gitea" not in read_json(datadir / "undefined.json")
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_unauthorized(client, gitea_off, endpoint):
|
def test_unauthorized(client, gitea_off, endpoint):
|
||||||
response = client.post(f"/services/gitea/{endpoint}")
|
response = client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_illegal_methods(authorized_client, gitea_off, endpoint):
|
def test_illegal_methods(authorized_client, gitea_off, endpoint):
|
||||||
response = authorized_client.get(f"/services/gitea/{endpoint}")
|
response = authorized_client.get(f"/services/gitea/{endpoint}")
|
||||||
|
@ -47,34 +58,64 @@ def test_illegal_methods(authorized_client, gitea_off, endpoint):
|
||||||
response = authorized_client.delete(f"/services/gitea/{endpoint}")
|
response = authorized_client.delete(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_off(authorized_client, gitea_off, endpoint, target_file):
|
def test_switch_from_off(authorized_client, gitea_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(gitea_off / "turned_off.json") == read_json(gitea_off / target_file)
|
assert read_json(gitea_off / "turned_off.json") == read_json(
|
||||||
|
gitea_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_on(authorized_client, gitea_on, endpoint, target_file):
|
def test_switch_from_on(authorized_client, gitea_on, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(gitea_on / "turned_on.json") == read_json(gitea_on / target_file)
|
assert read_json(gitea_on / "turned_on.json") == read_json(gitea_on / target_file)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_twice(authorized_client, gitea_off, endpoint, target_file):
|
def test_switch_twice(authorized_client, gitea_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(gitea_off / "turned_off.json") == read_json(gitea_off / target_file)
|
assert read_json(gitea_off / "turned_off.json") == read_json(
|
||||||
|
gitea_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_attribute_deleted(authorized_client, gitea_enable_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_attribute_deleted(
|
||||||
|
authorized_client, gitea_enable_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(gitea_enable_undefined / "enable_undefined.json") == read_json(gitea_enable_undefined / target_file)
|
assert read_json(gitea_enable_undefined / "enable_undefined.json") == read_json(
|
||||||
|
gitea_enable_undefined / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_on_gitea_undefined(authorized_client, gitea_undefined, endpoint, target_file):
|
def test_on_gitea_undefined(authorized_client, gitea_undefined, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
response = authorized_client.post(f"/services/gitea/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(gitea_undefined / "undefined.json") == read_json(gitea_undefined / target_file)
|
assert read_json(gitea_undefined / "undefined.json") == read_json(
|
||||||
|
gitea_undefined / target_file
|
||||||
|
)
|
||||||
|
|
|
@ -2,14 +2,18 @@ import base64
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r", encoding="utf-8") as f:
|
with open(file_path, "r", encoding="utf-8") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
class ProcessMock():
|
|
||||||
|
class ProcessMock:
|
||||||
"""Mock subprocess.Popen"""
|
"""Mock subprocess.Popen"""
|
||||||
|
|
||||||
def __init__(self, args, **kwargs):
|
def __init__(self, args, **kwargs):
|
||||||
self.args = args
|
self.args = args
|
||||||
self.kwargs = kwargs
|
self.kwargs = kwargs
|
||||||
|
@ -17,6 +21,7 @@ class ProcessMock():
|
||||||
def communicate():
|
def communicate():
|
||||||
return (b"I am a DKIM key", None)
|
return (b"I am a DKIM key", None)
|
||||||
|
|
||||||
|
|
||||||
class NoFileMock(ProcessMock):
|
class NoFileMock(ProcessMock):
|
||||||
def communicate():
|
def communicate():
|
||||||
return (b"", None)
|
return (b"", None)
|
||||||
|
@ -25,24 +30,36 @@ class NoFileMock(ProcessMock):
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def mock_subproccess_popen(mocker):
|
def mock_subproccess_popen(mocker):
|
||||||
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=ProcessMock)
|
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=ProcessMock)
|
||||||
mocker.patch("selfprivacy_api.resources.services.mailserver.get_domain", autospec=True, return_value="example.com")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.resources.services.mailserver.get_domain",
|
||||||
|
autospec=True,
|
||||||
|
return_value="example.com",
|
||||||
|
)
|
||||||
mocker.patch("os.path.exists", autospec=True, return_value=True)
|
mocker.patch("os.path.exists", autospec=True, return_value=True)
|
||||||
return mock
|
return mock
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def mock_no_file(mocker):
|
def mock_no_file(mocker):
|
||||||
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=NoFileMock)
|
mock = mocker.patch("subprocess.Popen", autospec=True, return_value=NoFileMock)
|
||||||
mocker.patch("selfprivacy_api.resources.services.mailserver.get_domain", autospec=True, return_value="example.com")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.resources.services.mailserver.get_domain",
|
||||||
|
autospec=True,
|
||||||
|
return_value="example.com",
|
||||||
|
)
|
||||||
mocker.patch("os.path.exists", autospec=True, return_value=False)
|
mocker.patch("os.path.exists", autospec=True, return_value=False)
|
||||||
return mock
|
return mock
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
def test_unauthorized(client, mock_subproccess_popen):
|
def test_unauthorized(client, mock_subproccess_popen):
|
||||||
"""Test unauthorized"""
|
"""Test unauthorized"""
|
||||||
response = client.get("/services/mailserver/dkim")
|
response = client.get("/services/mailserver/dkim")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
def test_illegal_methods(authorized_client, mock_subproccess_popen):
|
def test_illegal_methods(authorized_client, mock_subproccess_popen):
|
||||||
response = authorized_client.post("/services/mailserver/dkim")
|
response = authorized_client.post("/services/mailserver/dkim")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
@ -51,12 +68,17 @@ def test_illegal_methods(authorized_client, mock_subproccess_popen):
|
||||||
response = authorized_client.delete("/services/mailserver/dkim")
|
response = authorized_client.delete("/services/mailserver/dkim")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
|
|
||||||
def test_dkim_key(authorized_client, mock_subproccess_popen):
|
def test_dkim_key(authorized_client, mock_subproccess_popen):
|
||||||
"""Test DKIM key"""
|
"""Test DKIM key"""
|
||||||
response = authorized_client.get("/services/mailserver/dkim")
|
response = authorized_client.get("/services/mailserver/dkim")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert base64.b64decode(response.data) == b"I am a DKIM key"
|
assert base64.b64decode(response.data) == b"I am a DKIM key"
|
||||||
assert mock_subproccess_popen.call_args[0][0] == ["cat", "/var/dkim/example.com.selector.txt"]
|
assert mock_subproccess_popen.call_args[0][0] == [
|
||||||
|
"cat",
|
||||||
|
"/var/dkim/example.com.selector.txt",
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_no_dkim_key(authorized_client, mock_no_file):
|
def test_no_dkim_key(authorized_client, mock_no_file):
|
||||||
"""Test no DKIM key"""
|
"""Test no DKIM key"""
|
||||||
|
|
|
@ -1,43 +1,54 @@
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def nextcloud_off(mocker, datadir):
|
def nextcloud_off(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
||||||
assert read_json(datadir / "turned_off.json")["nextcloud"]["enable"] == False
|
assert read_json(datadir / "turned_off.json")["nextcloud"]["enable"] == False
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def nextcloud_on(mocker, datadir):
|
def nextcloud_on(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
||||||
assert read_json(datadir / "turned_on.json")["nextcloud"]["enable"] == True
|
assert read_json(datadir / "turned_on.json")["nextcloud"]["enable"] == True
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def nextcloud_enable_undefined(mocker, datadir):
|
def nextcloud_enable_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
|
||||||
|
)
|
||||||
assert "enable" not in read_json(datadir / "enable_undefined.json")["nextcloud"]
|
assert "enable" not in read_json(datadir / "enable_undefined.json")["nextcloud"]
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def nextcloud_undefined(mocker, datadir):
|
def nextcloud_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||||||
assert "nextcloud" not in read_json(datadir / "undefined.json")
|
assert "nextcloud" not in read_json(datadir / "undefined.json")
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_unauthorized(client, nextcloud_off, endpoint):
|
def test_unauthorized(client, nextcloud_off, endpoint):
|
||||||
response = client.post(f"/services/nextcloud/{endpoint}")
|
response = client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_illegal_methods(authorized_client, nextcloud_off, endpoint):
|
def test_illegal_methods(authorized_client, nextcloud_off, endpoint):
|
||||||
response = authorized_client.get(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.get(f"/services/nextcloud/{endpoint}")
|
||||||
|
@ -47,34 +58,66 @@ def test_illegal_methods(authorized_client, nextcloud_off, endpoint):
|
||||||
response = authorized_client.delete(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.delete(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_off(authorized_client, nextcloud_off, endpoint, target_file):
|
def test_switch_from_off(authorized_client, nextcloud_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(nextcloud_off / "turned_off.json") == read_json(nextcloud_off / target_file)
|
assert read_json(nextcloud_off / "turned_off.json") == read_json(
|
||||||
|
nextcloud_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_on(authorized_client, nextcloud_on, endpoint, target_file):
|
def test_switch_from_on(authorized_client, nextcloud_on, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(nextcloud_on / "turned_on.json") == read_json(nextcloud_on / target_file)
|
assert read_json(nextcloud_on / "turned_on.json") == read_json(
|
||||||
|
nextcloud_on / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_twice(authorized_client, nextcloud_off, endpoint, target_file):
|
def test_switch_twice(authorized_client, nextcloud_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(nextcloud_off / "turned_off.json") == read_json(nextcloud_off / target_file)
|
assert read_json(nextcloud_off / "turned_off.json") == read_json(
|
||||||
|
nextcloud_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_attribute_deleted(authorized_client, nextcloud_enable_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_attribute_deleted(
|
||||||
|
authorized_client, nextcloud_enable_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(nextcloud_enable_undefined / "enable_undefined.json") == read_json(nextcloud_enable_undefined / target_file)
|
assert read_json(nextcloud_enable_undefined / "enable_undefined.json") == read_json(
|
||||||
|
nextcloud_enable_undefined / target_file
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target", [("enable", True), ("disable", False)])
|
@pytest.mark.parametrize("endpoint,target", [("enable", True), ("disable", False)])
|
||||||
def test_on_nextcloud_undefined(authorized_client, nextcloud_undefined, endpoint, target):
|
def test_on_nextcloud_undefined(
|
||||||
|
authorized_client, nextcloud_undefined, endpoint, target
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
response = authorized_client.post(f"/services/nextcloud/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(nextcloud_undefined / "undefined.json")["nextcloud"]["enable"] == target
|
assert (
|
||||||
|
read_json(nextcloud_undefined / "undefined.json")["nextcloud"]["enable"]
|
||||||
|
== target
|
||||||
|
)
|
||||||
|
|
|
@ -1,43 +1,54 @@
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def ocserv_off(mocker, datadir):
|
def ocserv_off(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
||||||
assert read_json(datadir / "turned_off.json")["ocserv"]["enable"] == False
|
assert read_json(datadir / "turned_off.json")["ocserv"]["enable"] == False
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def ocserv_on(mocker, datadir):
|
def ocserv_on(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
||||||
assert read_json(datadir / "turned_on.json")["ocserv"]["enable"] == True
|
assert read_json(datadir / "turned_on.json")["ocserv"]["enable"] == True
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def ocserv_enable_undefined(mocker, datadir):
|
def ocserv_enable_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
|
||||||
|
)
|
||||||
assert "enable" not in read_json(datadir / "enable_undefined.json")["ocserv"]
|
assert "enable" not in read_json(datadir / "enable_undefined.json")["ocserv"]
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def ocserv_undefined(mocker, datadir):
|
def ocserv_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||||||
assert "ocserv" not in read_json(datadir / "undefined.json")
|
assert "ocserv" not in read_json(datadir / "undefined.json")
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_unauthorized(client, ocserv_off, endpoint):
|
def test_unauthorized(client, ocserv_off, endpoint):
|
||||||
response = client.post(f"/services/ocserv/{endpoint}")
|
response = client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_illegal_methods(authorized_client, ocserv_off, endpoint):
|
def test_illegal_methods(authorized_client, ocserv_off, endpoint):
|
||||||
response = authorized_client.get(f"/services/ocserv/{endpoint}")
|
response = authorized_client.get(f"/services/ocserv/{endpoint}")
|
||||||
|
@ -47,34 +58,66 @@ def test_illegal_methods(authorized_client, ocserv_off, endpoint):
|
||||||
response = authorized_client.delete(f"/services/ocserv/{endpoint}")
|
response = authorized_client.delete(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_off(authorized_client, ocserv_off, endpoint, target_file):
|
def test_switch_from_off(authorized_client, ocserv_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(ocserv_off / "turned_off.json") == read_json(ocserv_off / target_file)
|
assert read_json(ocserv_off / "turned_off.json") == read_json(
|
||||||
|
ocserv_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_from_on(authorized_client, ocserv_on, endpoint, target_file):
|
def test_switch_from_on(authorized_client, ocserv_on, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(ocserv_on / "turned_on.json") == read_json(ocserv_on / target_file)
|
assert read_json(ocserv_on / "turned_on.json") == read_json(ocserv_on / target_file)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
def test_switch_twice(authorized_client, ocserv_off, endpoint, target_file):
|
def test_switch_twice(authorized_client, ocserv_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(ocserv_off / "turned_off.json") == read_json(ocserv_off / target_file)
|
assert read_json(ocserv_off / "turned_off.json") == read_json(
|
||||||
|
ocserv_off / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_attribute_deleted(authorized_client, ocserv_enable_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_attribute_deleted(
|
||||||
|
authorized_client, ocserv_enable_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(ocserv_enable_undefined / "enable_undefined.json") == read_json(ocserv_enable_undefined / target_file)
|
assert read_json(ocserv_enable_undefined / "enable_undefined.json") == read_json(
|
||||||
|
ocserv_enable_undefined / target_file
|
||||||
|
)
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
def test_on_ocserv_undefined(authorized_client, ocserv_undefined, endpoint, target_file):
|
@pytest.mark.parametrize(
|
||||||
|
"endpoint,target_file",
|
||||||
|
[("enable", "turned_on.json"), ("disable", "turned_off.json")],
|
||||||
|
)
|
||||||
|
def test_on_ocserv_undefined(
|
||||||
|
authorized_client, ocserv_undefined, endpoint, target_file
|
||||||
|
):
|
||||||
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
response = authorized_client.post(f"/services/ocserv/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(ocserv_undefined / "undefined.json") == read_json(ocserv_undefined / target_file)
|
assert read_json(ocserv_undefined / "undefined.json") == read_json(
|
||||||
|
ocserv_undefined / target_file
|
||||||
|
)
|
||||||
|
|
|
@ -1,43 +1,54 @@
|
||||||
import json
|
import json
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
def read_json(file_path):
|
def read_json(file_path):
|
||||||
with open(file_path, "r") as f:
|
with open(file_path, "r") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def pleroma_off(mocker, datadir):
|
def pleroma_off(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_off.json")
|
||||||
assert read_json(datadir / "turned_off.json")["pleroma"]["enable"] == False
|
assert read_json(datadir / "turned_off.json")["pleroma"]["enable"] == False
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def pleroma_on(mocker, datadir):
|
def pleroma_on(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "turned_on.json")
|
||||||
assert read_json(datadir / "turned_on.json")["pleroma"]["enable"] == True
|
assert read_json(datadir / "turned_on.json")["pleroma"]["enable"] == True
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def pleroma_enable_undefined(mocker, datadir):
|
def pleroma_enable_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json")
|
mocker.patch(
|
||||||
|
"selfprivacy_api.utils.USERDATA_FILE", new=datadir / "enable_undefined.json"
|
||||||
|
)
|
||||||
assert "enable" not in read_json(datadir / "enable_undefined.json")["pleroma"]
|
assert "enable" not in read_json(datadir / "enable_undefined.json")["pleroma"]
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def pleroma_undefined(mocker, datadir):
|
def pleroma_undefined(mocker, datadir):
|
||||||
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||||||
assert "pleroma" not in read_json(datadir / "undefined.json")
|
assert "pleroma" not in read_json(datadir / "undefined.json")
|
||||||
return datadir
|
return datadir
|
||||||
|
|
||||||
|
|
||||||
###############################################################################
|
###############################################################################
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_unauthorized(client, pleroma_off, endpoint):
|
def test_unauthorized(client, pleroma_off, endpoint):
|
||||||
response = client.post(f"/services/pleroma/{endpoint}")
|
response = client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 401
|
assert response.status_code == 401
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
@pytest.mark.parametrize("endpoint", ["enable", "disable"])
|
||||||
def test_illegal_methods(authorized_client, pleroma_off, endpoint):
|
def test_illegal_methods(authorized_client, pleroma_off, endpoint):
|
||||||
response = authorized_client.get(f"/services/pleroma/{endpoint}")
|
response = authorized_client.get(f"/services/pleroma/{endpoint}")
|
||||||
|
@ -47,34 +58,68 @@ def test_illegal_methods(authorized_client, pleroma_off, endpoint):
|
||||||
response = authorized_client.delete(f"/services/pleroma/{endpoint}")
|
response = authorized_client.delete(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 405
|
assert response.status_code == 405
|
||||||
|
|
||||||
@pytest.mark.parametrize("endpoint,target_file", [("enable", "turned_on.json"), ("disable", "turned_off.json")])
|
|
||||||
|
@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):
|
def test_switch_from_off(authorized_client, pleroma_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(pleroma_off / "turned_off.json") == read_json(pleroma_off / target_file)
|
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")])
|
|
||||||
|
@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):
|
def test_switch_from_on(authorized_client, pleroma_on, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(pleroma_on / "turned_on.json") == read_json(pleroma_on / target_file)
|
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")])
|
|
||||||
|
@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):
|
def test_switch_twice(authorized_client, pleroma_off, endpoint, target_file):
|
||||||
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(pleroma_off / "turned_off.json") == read_json(pleroma_off / target_file)
|
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):
|
@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}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(pleroma_enable_undefined / "enable_undefined.json") == read_json(pleroma_enable_undefined / target_file)
|
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):
|
@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}")
|
response = authorized_client.post(f"/services/pleroma/{endpoint}")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(pleroma_undefined / "undefined.json") == read_json(pleroma_undefined / target_file)
|
assert read_json(pleroma_undefined / "undefined.json") == read_json(
|
||||||
|
pleroma_undefined / target_file
|
||||||
|
)
|
||||||
|
|
|
@ -7,6 +7,7 @@ def read_json(file_path):
|
||||||
with open(file_path, "r", encoding="utf-8") as f:
|
with open(file_path, "r", encoding="utf-8") as f:
|
||||||
return json.load(f)
|
return json.load(f)
|
||||||
|
|
||||||
|
|
||||||
def call_args_asserts(mocked_object):
|
def call_args_asserts(mocked_object):
|
||||||
assert mocked_object.call_count == 8
|
assert mocked_object.call_count == 8
|
||||||
assert mocked_object.call_args_list[0][0][0] == [
|
assert mocked_object.call_args_list[0][0][0] == [
|
||||||
|
@ -50,6 +51,7 @@ def call_args_asserts(mocked_object):
|
||||||
"pleroma.service",
|
"pleroma.service",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
class ProcessMock:
|
class ProcessMock:
|
||||||
"""Mock subprocess.Popen"""
|
"""Mock subprocess.Popen"""
|
||||||
|
|
||||||
|
|
|
@ -172,91 +172,143 @@ def test_set_settings_undefined(authorized_client, undefined_settings, settings)
|
||||||
if "passwordAuthentication" in settings:
|
if "passwordAuthentication" in settings:
|
||||||
assert data["passwordAuthentication"] == settings["passwordAuthentication"]
|
assert data["passwordAuthentication"] == settings["passwordAuthentication"]
|
||||||
|
|
||||||
|
|
||||||
def test_add_root_key(authorized_client, ssh_on):
|
def test_add_root_key(authorized_client, ssh_on):
|
||||||
response = authorized_client.put(f"/services/ssh/key/send", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.put(
|
||||||
|
f"/services/ssh/key/send", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 201
|
assert response.status_code == 201
|
||||||
assert read_json(ssh_on / "turned_on.json")["ssh"]["rootKeys"] == [
|
assert read_json(ssh_on / "turned_on.json")["ssh"]["rootKeys"] == [
|
||||||
"ssh-rsa KEY test@pc",
|
"ssh-rsa KEY test@pc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_add_root_key_one_more(authorized_client, root_and_admin_have_keys):
|
def test_add_root_key_one_more(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.put(f"/services/ssh/key/send", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.put(
|
||||||
|
f"/services/ssh/key/send", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 201
|
assert response.status_code == 201
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"]["rootKeys"] == [
|
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"][
|
||||||
|
"rootKeys"
|
||||||
|
] == [
|
||||||
"ssh-ed25519 KEY test@pc",
|
"ssh-ed25519 KEY test@pc",
|
||||||
"ssh-rsa KEY test@pc",
|
"ssh-rsa KEY test@pc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_add_existing_root_key(authorized_client, root_and_admin_have_keys):
|
def test_add_existing_root_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.put(f"/services/ssh/key/send", json={"public_key": "ssh-ed25519 KEY test@pc"})
|
response = authorized_client.put(
|
||||||
|
f"/services/ssh/key/send", json={"public_key": "ssh-ed25519 KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 409
|
assert response.status_code == 409
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"]["rootKeys"] == [
|
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"][
|
||||||
|
"rootKeys"
|
||||||
|
] == [
|
||||||
"ssh-ed25519 KEY test@pc",
|
"ssh-ed25519 KEY test@pc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_add_invalid_root_key(authorized_client, ssh_on):
|
def test_add_invalid_root_key(authorized_client, ssh_on):
|
||||||
response = authorized_client.put(f"/services/ssh/key/send", json={"public_key": "INVALID KEY test@pc"})
|
response = authorized_client.put(
|
||||||
|
f"/services/ssh/key/send", json={"public_key": "INVALID KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 400
|
assert response.status_code == 400
|
||||||
|
|
||||||
|
|
||||||
def test_add_root_key_via_wrong_endpoint(authorized_client, ssh_on):
|
def test_add_root_key_via_wrong_endpoint(authorized_client, ssh_on):
|
||||||
response = authorized_client.post(f"/services/ssh/keys/root", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.post(
|
||||||
|
f"/services/ssh/keys/root", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 400
|
assert response.status_code == 400
|
||||||
|
|
||||||
|
|
||||||
def test_get_root_key(authorized_client, root_and_admin_have_keys):
|
def test_get_root_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.get(f"/services/ssh/keys/root")
|
response = authorized_client.get(f"/services/ssh/keys/root")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert response.json == ["ssh-ed25519 KEY test@pc"]
|
assert response.json == ["ssh-ed25519 KEY test@pc"]
|
||||||
|
|
||||||
|
|
||||||
def test_get_root_key_when_none(authorized_client, ssh_on):
|
def test_get_root_key_when_none(authorized_client, ssh_on):
|
||||||
response = authorized_client.get(f"/services/ssh/keys/root")
|
response = authorized_client.get(f"/services/ssh/keys/root")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert response.json == []
|
assert response.json == []
|
||||||
|
|
||||||
|
|
||||||
def test_delete_root_key(authorized_client, root_and_admin_have_keys):
|
def test_delete_root_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.delete(f"/services/ssh/keys/root", json={"public_key": "ssh-ed25519 KEY test@pc"})
|
response = authorized_client.delete(
|
||||||
|
f"/services/ssh/keys/root", json={"public_key": "ssh-ed25519 KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"]["rootKeys"] == []
|
assert (
|
||||||
|
read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"][
|
||||||
|
"rootKeys"
|
||||||
|
]
|
||||||
|
== []
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_delete_root_nonexistent_key(authorized_client, root_and_admin_have_keys):
|
def test_delete_root_nonexistent_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.delete(f"/services/ssh/keys/root", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.delete(
|
||||||
|
f"/services/ssh/keys/root", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 404
|
assert response.status_code == 404
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"]["rootKeys"] == [
|
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["ssh"][
|
||||||
|
"rootKeys"
|
||||||
|
] == [
|
||||||
"ssh-ed25519 KEY test@pc",
|
"ssh-ed25519 KEY test@pc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_get_admin_key(authorized_client, root_and_admin_have_keys):
|
def test_get_admin_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.get(f"/services/ssh/keys/tester")
|
response = authorized_client.get(f"/services/ssh/keys/tester")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert response.json == ["ssh-rsa KEY test@pc"]
|
assert response.json == ["ssh-rsa KEY test@pc"]
|
||||||
|
|
||||||
|
|
||||||
def test_get_admin_key_when_none(authorized_client, ssh_on):
|
def test_get_admin_key_when_none(authorized_client, ssh_on):
|
||||||
response = authorized_client.get(f"/services/ssh/keys/tester")
|
response = authorized_client.get(f"/services/ssh/keys/tester")
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert response.json == []
|
assert response.json == []
|
||||||
|
|
||||||
|
|
||||||
def test_delete_admin_key(authorized_client, root_and_admin_have_keys):
|
def test_delete_admin_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.delete(f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.delete(
|
||||||
|
f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["sshKeys"] == []
|
assert (
|
||||||
|
read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["sshKeys"]
|
||||||
|
== []
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def test_add_admin_key(authorized_client, ssh_on):
|
def test_add_admin_key(authorized_client, ssh_on):
|
||||||
response = authorized_client.post(f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.post(
|
||||||
|
f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 201
|
assert response.status_code == 201
|
||||||
assert read_json(ssh_on / "turned_on.json")["sshKeys"] == [
|
assert read_json(ssh_on / "turned_on.json")["sshKeys"] == [
|
||||||
"ssh-rsa KEY test@pc",
|
"ssh-rsa KEY test@pc",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
def test_add_admin_key_one_more(authorized_client, root_and_admin_have_keys):
|
def test_add_admin_key_one_more(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.post(f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY_2 test@pc"})
|
response = authorized_client.post(
|
||||||
|
f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY_2 test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 201
|
assert response.status_code == 201
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["sshKeys"] == [
|
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")[
|
||||||
"ssh-rsa KEY test@pc",
|
"sshKeys"
|
||||||
"ssh-rsa KEY_2 test@pc"
|
] == ["ssh-rsa KEY test@pc", "ssh-rsa KEY_2 test@pc"]
|
||||||
]
|
|
||||||
|
|
||||||
def test_add_existing_admin_key(authorized_client, root_and_admin_have_keys):
|
def test_add_existing_admin_key(authorized_client, root_and_admin_have_keys):
|
||||||
response = authorized_client.post(f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"})
|
response = authorized_client.post(
|
||||||
|
f"/services/ssh/keys/tester", json={"public_key": "ssh-rsa KEY test@pc"}
|
||||||
|
)
|
||||||
assert response.status_code == 409
|
assert response.status_code == 409
|
||||||
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")["sshKeys"] == [
|
assert read_json(root_and_admin_have_keys / "root_and_admin_have_keys.json")[
|
||||||
|
"sshKeys"
|
||||||
|
] == [
|
||||||
"ssh-rsa KEY test@pc",
|
"ssh-rsa KEY test@pc",
|
||||||
]
|
]
|
Loading…
Reference in a new issue