185 lines
6.7 KiB
Python
185 lines
6.7 KiB
Python
"""Test the OpenFlexureSystem Thing *without* connecting it to a LabThings Server."""
|
|
|
|
import os
|
|
from signal import SIGTERM
|
|
from uuid import UUID
|
|
|
|
import pytest
|
|
|
|
from labthings_fastapi.testing import create_thing_without_server
|
|
|
|
from openflexure_microscope_server.things import system
|
|
from openflexure_microscope_server.utilities import VersionData, robust_version_strings
|
|
|
|
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
FAKE_OS_FILE = os.path.join(THIS_DIR, "assets", "os-release")
|
|
BROKEN_OS_FILE = os.path.join(THIS_DIR, "assets", "os-release-broken")
|
|
MISSING_OS_FILE = os.path.join(THIS_DIR, "assets", "this-does-not-exist")
|
|
|
|
|
|
@pytest.fixture
|
|
def system_thing():
|
|
"""Return a OpenFlexureSystem with a mocked server interface."""
|
|
return create_thing_without_server(system.OpenFlexureSystem)
|
|
|
|
|
|
def _is_raspberrypi() -> bool:
|
|
"""Check if the machine running the test is a Raspberry Pi.
|
|
|
|
This information is needed to check if the ``is_raspberrypi`` property of the Thing
|
|
is correct. This is implemented in a different way to in the main code.
|
|
"""
|
|
# Check if the file specifying the Raspberry Pi model exists
|
|
if not os.path.isfile("/proc/device-tree/model"):
|
|
return False
|
|
with open("/proc/device-tree/model", "r", encoding="utf-8") as model_file:
|
|
model_name = model_file.read()
|
|
return "Raspberry Pi" in model_name
|
|
|
|
|
|
def test_is_raspberry(system_thing):
|
|
"""Check the thing property reports whether this is a Raspberry Pi correctly."""
|
|
assert system_thing.is_raspberrypi == _is_raspberrypi()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("version_file", "expected_result"),
|
|
[
|
|
(MISSING_OS_FILE, "unknown"),
|
|
(FAKE_OS_FILE, "trixie"),
|
|
(BROKEN_OS_FILE, "unknown"),
|
|
],
|
|
)
|
|
def test_os_version_pi(version_file, expected_result, system_thing, mocker):
|
|
"""Check reading the operating system version from a Pi (mocked!)."""
|
|
type(system_thing).is_raspberrypi = mocker.PropertyMock(return_value=True)
|
|
|
|
mocker.patch(
|
|
"openflexure_microscope_server.things.system.OS_VERSION_FILE", version_file
|
|
)
|
|
assert system_thing.os_version == expected_result
|
|
|
|
|
|
def test_os_version_not_pi(system_thing, mocker):
|
|
"""Check reading the operating system when not on a Pi returns None."""
|
|
mocker.patch(
|
|
"openflexure_microscope_server.things.system.OS_VERSION_FILE", FAKE_OS_FILE
|
|
)
|
|
type(system_thing).is_raspberrypi = mocker.PropertyMock(return_value=False)
|
|
assert system_thing.os_version is None
|
|
|
|
|
|
def test_version_data(system_thing):
|
|
"""Check VersionData is returned.
|
|
|
|
The content of robust_version_strings() is tested elsewhere.
|
|
"""
|
|
data = system_thing.version_data
|
|
assert isinstance(data, VersionData)
|
|
assert data == robust_version_strings()
|
|
|
|
fake_data = VersionData(version="fake", version_source="fake")
|
|
system_thing._version_data = fake_data
|
|
data = system_thing.version_data
|
|
assert data == fake_data
|
|
|
|
|
|
def test_hostname(system_thing, mocker):
|
|
"""Check the hostname matches what socket.gethostname() returns."""
|
|
mock_gethostname = mocker.patch("socket.gethostname", return_value="foobar")
|
|
assert system_thing.hostname == "foobar"
|
|
mock_gethostname.assert_called_once_with()
|
|
|
|
|
|
def test_microscope_id(system_thing):
|
|
"""Check the microscope UUID is a valid UUID and doesn't change when read again."""
|
|
microscope_id = system_thing.microscope_id
|
|
assert isinstance(microscope_id, UUID)
|
|
assert microscope_id == system_thing.microscope_id
|
|
|
|
|
|
def test_thing_state(system_thing, mocker):
|
|
"""Check the thing state contains version data, hostname, and a UUID string."""
|
|
mocker.patch("socket.gethostname", return_value="foobar")
|
|
version_data = robust_version_strings()
|
|
state_dict = system_thing.thing_state
|
|
assert state_dict["hostname"] == "foobar"
|
|
# Check the UUID in the dictionary is a string
|
|
assert isinstance(state_dict["microscope-uuid"], str)
|
|
# Check uuid string can be converted to valid UUID
|
|
UUID(state_dict["microscope-uuid"])
|
|
assert state_dict["version"] == version_data.version
|
|
assert state_dict["version_source"] == version_data.version_source
|
|
|
|
|
|
class MockPiSystem(system.OpenFlexureSystem):
|
|
"""A OpenFlexureSystem Thing that always claims to be a Raspberry Pi."""
|
|
|
|
is_raspberrypi: bool = True
|
|
|
|
|
|
class MockNonPiSystem(system.OpenFlexureSystem):
|
|
"""A OpenFlexureSystem Thing that never claims to be a Raspberry Pi."""
|
|
|
|
is_raspberrypi: bool = False
|
|
|
|
|
|
def test_pi_shutdown(mocker):
|
|
"""Check that on a Pi will receive the correct shutdown command."""
|
|
# Check the shutdown command is as expected.
|
|
assert system.SHUTDOWN_CMD == ["sudo", "shutdown", "-h", "now"]
|
|
# Mock the shutdown command as we don't want to shutdown when running tests.
|
|
mocker.patch.object(
|
|
system, "SHUTDOWN_CMD", new=["python", "-c", "print('shutdown')"]
|
|
)
|
|
# Call shutdown on a MockPiSystem
|
|
system_thing = create_thing_without_server(MockPiSystem)
|
|
result = system_thing.shutdown()
|
|
|
|
# Check the result of the mock command was returned
|
|
assert result.output.strip() == "shutdown"
|
|
assert result.error.strip() == ""
|
|
|
|
|
|
def test_pi_reboot(mocker):
|
|
"""Check that on a Pi will receive the correct reboot command."""
|
|
# Check the reboot command is as expected.
|
|
assert system.REBOOT_CMD == ["sudo", "shutdown", "-r", "now"]
|
|
# Mock the reboot command as we don't want to reboot when running tests.
|
|
mocker.patch.object(system, "REBOOT_CMD", new=["python", "-c", "print('restart')"])
|
|
# Call reboot on a MockPiSystem
|
|
system_thing = create_thing_without_server(MockPiSystem)
|
|
result = system_thing.reboot()
|
|
|
|
# Check the result of the mock command was returned
|
|
assert result.output.strip() == "restart"
|
|
assert result.error.strip() == ""
|
|
|
|
|
|
def test_non_pi_shutdown(mocker):
|
|
"""Check that a server not on a pi runs os.kill when asked to shutdown.
|
|
|
|
It should do this instead of running the shutdown command in a subprocess.
|
|
"""
|
|
# Mock os.kill
|
|
mock_kill = mocker.patch("os.kill")
|
|
# Get this subprocess
|
|
pid = os.getpid()
|
|
system_thing = create_thing_without_server(MockNonPiSystem)
|
|
result = system_thing.shutdown()
|
|
|
|
# Check it tried to kill this process with SIGTERM
|
|
mock_kill.assert_called_once_with(pid, SIGTERM)
|
|
# Check output is an appropriate error message (that we have not shut down!)
|
|
assert result.output.strip() == ""
|
|
assert "but the server has not shutdown" in result.error
|
|
|
|
|
|
def test_non_pi_reboot():
|
|
"""Check that a server not on a pi refuses to restart."""
|
|
system_thing = create_thing_without_server(MockNonPiSystem)
|
|
result = system_thing.reboot()
|
|
|
|
# Check output is an appropriate error message
|
|
assert result.output.strip() == ""
|
|
assert result.error.strip() == "Restart is only available on Raspberry Pi."
|