Developer
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from media_library_viewer_api.services.monitoring_actions import poll_machine_snapshot
|
|
|
|
|
|
def test_poll_machine_snapshot_records_disk_lookup():
|
|
store = MagicMock()
|
|
machine = {
|
|
"id": "local",
|
|
"name": "This machine",
|
|
"mode": "local",
|
|
"media_root": "/srv/media",
|
|
}
|
|
|
|
with (
|
|
patch(
|
|
"media_library_viewer_api.services.monitoring_actions.build_machine_client",
|
|
return_value=object(),
|
|
) as build_client,
|
|
patch(
|
|
"media_library_viewer_api.services.monitoring_actions.disk_space",
|
|
return_value={"mount": "/srv/media", "used_pct": "12.5%"},
|
|
) as disk_fn,
|
|
):
|
|
result = poll_machine_snapshot(machine, store, metrics_limit=123, request_id="poll:test")
|
|
|
|
assert result["request_id"] == "poll:test"
|
|
assert result["disk_mount"] == "/srv/media"
|
|
assert result["actions"] == ["disk lookup"]
|
|
build_client.assert_called_once_with(machine, store)
|
|
disk_fn.assert_called_once_with(build_client.return_value, "/srv/media")
|
|
assert store.record_machine_action.call_count == 1
|
|
recorded_action = store.record_machine_action.call_args
|
|
assert recorded_action.args[1] == "disk lookup for /srv/media"
|
|
assert recorded_action.kwargs["request_id"] == "poll:test"
|
|
assert recorded_action.args[2] == "ok"
|