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"