yunohost/src/tests/test_service.py
2021-11-10 17:56:02 +01:00

142 lines
3.7 KiB
Python

import os
from .conftest import raiseYunohostError
from yunohost.service import (
_get_services,
_save_services,
service_status,
service_add,
service_remove,
service_log,
service_reload_or_restart,
)
def setup_function(function):
clean()
def teardown_function(function):
clean()
def clean():
# To run these tests, we assume ssh(d) service exists and is running
assert os.system("pgrep sshd >/dev/null") == 0
services = _get_services()
assert "ssh" in services
if "dummyservice" in services:
del services["dummyservice"]
if "networking" in services:
del services["networking"]
_save_services(services)
if os.path.exists("/etc/nginx/conf.d/broken.conf"):
os.remove("/etc/nginx/conf.d/broken.conf")
os.system("systemctl reload-or-restart nginx")
def test_service_status_all():
status = service_status()
assert "ssh" in status.keys()
assert status["ssh"]["status"] == "running"
def test_service_status_single():
status = service_status("ssh")
assert "status" in status.keys()
assert status["status"] == "running"
def test_service_log():
logs = service_log("ssh")
assert "journalctl" in logs.keys()
assert "/var/log/auth.log" in logs.keys()
def test_service_status_unknown_service(mocker):
with raiseYunohostError(mocker, "service_unknown"):
service_status(["ssh", "doesnotexists"])
def test_service_add():
service_add("dummyservice", description="A dummy service to run tests")
assert "dummyservice" in service_status().keys()
def test_service_add_real_service():
service_add("networking")
assert "networking" in service_status().keys()
def test_service_remove():
service_add("dummyservice", description="A dummy service to run tests")
assert "dummyservice" in service_status().keys()
service_remove("dummyservice")
assert "dummyservice" not in service_status().keys()
def test_service_remove_service_that_doesnt_exists(mocker):
assert "dummyservice" not in service_status().keys()
with raiseYunohostError(mocker, "service_unknown"):
service_remove("dummyservice")
assert "dummyservice" not in service_status().keys()
def test_service_update_to_add_properties():
service_add("dummyservice", description="dummy")
assert not _get_services()["dummyservice"].get("test_status")
service_add("dummyservice", description="dummy", test_status="true")
assert _get_services()["dummyservice"].get("test_status") == "true"
def test_service_update_to_change_properties():
service_add("dummyservice", description="dummy", test_status="false")
assert _get_services()["dummyservice"].get("test_status") == "false"
service_add("dummyservice", description="dummy", test_status="true")
assert _get_services()["dummyservice"].get("test_status") == "true"
def test_service_update_to_remove_properties():
service_add("dummyservice", description="dummy", test_status="false")
assert _get_services()["dummyservice"].get("test_status") == "false"
service_add("dummyservice", description="dummy", test_status="")
assert not _get_services()["dummyservice"].get("test_status")
def test_service_conf_broken():
os.system("echo pwet > /etc/nginx/conf.d/broken.conf")
status = service_status("nginx")
assert status["status"] == "running"
assert status["configuration"] == "broken"
assert "broken.conf" in status["configuration-details"][0]
# Service reload-or-restart should check that the conf ain't valid
# before reload-or-restart, hence the service should still be running
service_reload_or_restart("nginx")
assert status["status"] == "running"
os.remove("/etc/nginx/conf.d/broken.conf")