mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-08-22 03:41:50 +00:00
491 lines
15 KiB
Python
491 lines
15 KiB
Python
|
#!/usr/bin/env python3
|
||
|
# pylint: disable=redefined-outer-name
|
||
|
# pylint: disable=unused-argument
|
||
|
# pylint: disable=missing-function-docstring
|
||
|
import json
|
||
|
import subprocess
|
||
|
import pytest
|
||
|
|
||
|
from selfprivacy_api.utils.block_devices import (
|
||
|
BlockDevice,
|
||
|
BlockDevices,
|
||
|
get_block_device,
|
||
|
resize_block_device,
|
||
|
)
|
||
|
from tests.common import read_json
|
||
|
|
||
|
SINGLE_LSBLK_OUTPUT = b"""
|
||
|
{
|
||
|
"blockdevices": [
|
||
|
{
|
||
|
"name": "sda1",
|
||
|
"path": "/dev/sda1",
|
||
|
"fsavail": "4614107136",
|
||
|
"fssize": "19814920192",
|
||
|
"fstype": "ext4",
|
||
|
"fsused": "14345314304",
|
||
|
"mountpoints": [
|
||
|
"/nix/store", "/"
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": "ec80c004-baec-4a2c-851d-0e1807135511",
|
||
|
"size": 20210236928,
|
||
|
"model": null,
|
||
|
"serial": null,
|
||
|
"type": "part"
|
||
|
}
|
||
|
]
|
||
|
}
|
||
|
"""
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def lsblk_singular_mock(mocker):
|
||
|
mock = mocker.patch(
|
||
|
"subprocess.check_output", autospec=True, return_value=SINGLE_LSBLK_OUTPUT
|
||
|
)
|
||
|
return mock
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def failed_check_output_mock(mocker):
|
||
|
mock = mocker.patch(
|
||
|
"subprocess.check_output",
|
||
|
autospec=True,
|
||
|
side_effect=subprocess.CalledProcessError(
|
||
|
returncode=1, cmd=["some", "command"]
|
||
|
),
|
||
|
)
|
||
|
return mock
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def only_root_in_userdata(mocker, datadir):
|
||
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "only_root.json")
|
||
|
assert read_json(datadir / "only_root.json")["volumes"][0]["device"] == "/dev/sda1"
|
||
|
assert (
|
||
|
read_json(datadir / "only_root.json")["volumes"][0]["mountPoint"]
|
||
|
== "/volumes/sda1"
|
||
|
)
|
||
|
assert read_json(datadir / "only_root.json")["volumes"][0]["filesystem"] == "ext4"
|
||
|
return datadir
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def no_devices_in_userdata(mocker, datadir):
|
||
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "no_devices.json")
|
||
|
assert read_json(datadir / "no_devices.json")["volumes"] == []
|
||
|
return datadir
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def undefined_devices_in_userdata(mocker, datadir):
|
||
|
mocker.patch("selfprivacy_api.utils.USERDATA_FILE", new=datadir / "undefined.json")
|
||
|
assert "volumes" not in read_json(datadir / "undefined.json")
|
||
|
return datadir
|
||
|
|
||
|
|
||
|
def test_create_block_device_object(lsblk_singular_mock, authorized_client):
|
||
|
output = get_block_device("sda1")
|
||
|
assert lsblk_singular_mock.call_count == 1
|
||
|
assert lsblk_singular_mock.call_args[0][0] == [
|
||
|
"lsblk",
|
||
|
"-J",
|
||
|
"-b",
|
||
|
"-o",
|
||
|
"NAME,PATH,FSAVAIL,FSSIZE,FSTYPE,FSUSED,MOUNTPOINTS,LABEL,UUID,SIZE,MODEL,SERIAL,TYPE",
|
||
|
"/dev/sda1",
|
||
|
]
|
||
|
assert output == json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]
|
||
|
|
||
|
|
||
|
def test_resize_block_device(lsblk_singular_mock, authorized_client):
|
||
|
result = resize_block_device("sdb")
|
||
|
assert result is True
|
||
|
assert lsblk_singular_mock.call_count == 1
|
||
|
assert lsblk_singular_mock.call_args[0][0] == [
|
||
|
"resize2fs",
|
||
|
"sdb",
|
||
|
]
|
||
|
|
||
|
|
||
|
def test_resize_block_device_failed(failed_check_output_mock, authorized_client):
|
||
|
result = resize_block_device("sdb")
|
||
|
assert result is False
|
||
|
assert failed_check_output_mock.call_count == 1
|
||
|
assert failed_check_output_mock.call_args[0][0] == [
|
||
|
"resize2fs",
|
||
|
"sdb",
|
||
|
]
|
||
|
|
||
|
|
||
|
VOLUME_LSBLK_OUTPUT = b"""
|
||
|
{
|
||
|
"blockdevices": [
|
||
|
{
|
||
|
"name": "sdb",
|
||
|
"path": "/dev/sdb",
|
||
|
"fsavail": "11888545792",
|
||
|
"fssize": "12573614080",
|
||
|
"fstype": "ext4",
|
||
|
"fsused": "24047616",
|
||
|
"mountpoints": [
|
||
|
"/volumes/sdb"
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": "fa9d0026-ee23-4047-b8b1-297ae16fa751",
|
||
|
"size": 12884901888,
|
||
|
"model": "Volume",
|
||
|
"serial": "21378102",
|
||
|
"type": "disk"
|
||
|
}
|
||
|
]
|
||
|
}
|
||
|
"""
|
||
|
|
||
|
|
||
|
def test_create_block_device(lsblk_singular_mock, authorized_client):
|
||
|
block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
|
||
|
assert block_device.name == "sdb"
|
||
|
assert block_device.path == "/dev/sdb"
|
||
|
assert block_device.fsavail == "11888545792"
|
||
|
assert block_device.fssize == "12573614080"
|
||
|
assert block_device.fstype == "ext4"
|
||
|
assert block_device.fsused == "24047616"
|
||
|
assert block_device.mountpoints == ["/volumes/sdb"]
|
||
|
assert block_device.label is None
|
||
|
assert block_device.uuid == "fa9d0026-ee23-4047-b8b1-297ae16fa751"
|
||
|
assert block_device.size == "12884901888"
|
||
|
assert block_device.model == "Volume"
|
||
|
assert block_device.serial == "21378102"
|
||
|
assert block_device.type == "disk"
|
||
|
assert block_device.locked is False
|
||
|
assert str(block_device) == "sdb"
|
||
|
assert (
|
||
|
repr(block_device)
|
||
|
== "<BlockDevice sdb of size 12884901888 mounted at ['/volumes/sdb']>"
|
||
|
)
|
||
|
assert hash(block_device) == hash("sdb")
|
||
|
|
||
|
|
||
|
def test_block_devices_equal(lsblk_singular_mock, authorized_client):
|
||
|
block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
block_device2 = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
|
||
|
assert block_device == block_device2
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def resize_block_mock(mocker):
|
||
|
mock = mocker.patch(
|
||
|
"selfprivacy_api.utils.block_devices.resize_block_device",
|
||
|
autospec=True,
|
||
|
return_value=True,
|
||
|
)
|
||
|
return mock
|
||
|
|
||
|
|
||
|
def test_call_resize_from_block_device(
|
||
|
lsblk_singular_mock, resize_block_mock, authorized_client
|
||
|
):
|
||
|
block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
block_device.resize()
|
||
|
assert resize_block_mock.call_count == 1
|
||
|
assert resize_block_mock.call_args[0][0] == "/dev/sdb"
|
||
|
assert lsblk_singular_mock.call_count == 0
|
||
|
|
||
|
|
||
|
def test_get_stats_from_block_device(lsblk_singular_mock, authorized_client):
|
||
|
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
stats = block_device.stats()
|
||
|
assert stats == {
|
||
|
"name": "sda1",
|
||
|
"path": "/dev/sda1",
|
||
|
"fsavail": "4614107136",
|
||
|
"fssize": "19814920192",
|
||
|
"fstype": "ext4",
|
||
|
"fsused": "14345314304",
|
||
|
"mountpoints": ["/nix/store", "/"],
|
||
|
"label": None,
|
||
|
"uuid": "ec80c004-baec-4a2c-851d-0e1807135511",
|
||
|
"size": "20210236928",
|
||
|
"model": None,
|
||
|
"serial": None,
|
||
|
"type": "part",
|
||
|
}
|
||
|
assert lsblk_singular_mock.call_count == 1
|
||
|
assert lsblk_singular_mock.call_args[0][0] == [
|
||
|
"lsblk",
|
||
|
"-J",
|
||
|
"-b",
|
||
|
"-o",
|
||
|
"NAME,PATH,FSAVAIL,FSSIZE,FSTYPE,FSUSED,MOUNTPOINTS,LABEL,UUID,SIZE,MODEL,SERIAL,TYPE",
|
||
|
"/dev/sda1",
|
||
|
]
|
||
|
|
||
|
|
||
|
def test_mount_block_device(
|
||
|
lsblk_singular_mock, only_root_in_userdata, authorized_client
|
||
|
):
|
||
|
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = block_device.mount()
|
||
|
assert result is False
|
||
|
volume = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = volume.mount()
|
||
|
assert result is True
|
||
|
assert (
|
||
|
read_json(only_root_in_userdata / "only_root.json")["volumes"][1]["device"]
|
||
|
== "/dev/sdb"
|
||
|
)
|
||
|
assert (
|
||
|
read_json(only_root_in_userdata / "only_root.json")["volumes"][1]["mountPoint"]
|
||
|
== "/volumes/sdb"
|
||
|
)
|
||
|
assert (
|
||
|
read_json(only_root_in_userdata / "only_root.json")["volumes"][1]["fsType"]
|
||
|
== "ext4"
|
||
|
)
|
||
|
|
||
|
|
||
|
def test_mount_block_device_when_undefined(
|
||
|
lsblk_singular_mock, undefined_devices_in_userdata, authorized_client
|
||
|
):
|
||
|
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = block_device.mount()
|
||
|
assert result is True
|
||
|
assert (
|
||
|
read_json(undefined_devices_in_userdata / "undefined.json")["volumes"][0][
|
||
|
"device"
|
||
|
]
|
||
|
== "/dev/sda1"
|
||
|
)
|
||
|
assert (
|
||
|
read_json(undefined_devices_in_userdata / "undefined.json")["volumes"][0][
|
||
|
"mountPoint"
|
||
|
]
|
||
|
== "/volumes/sda1"
|
||
|
)
|
||
|
assert (
|
||
|
read_json(undefined_devices_in_userdata / "undefined.json")["volumes"][0][
|
||
|
"fsType"
|
||
|
]
|
||
|
== "ext4"
|
||
|
)
|
||
|
|
||
|
|
||
|
def test_unmount_block_device(
|
||
|
lsblk_singular_mock, only_root_in_userdata, authorized_client
|
||
|
):
|
||
|
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = block_device.unmount()
|
||
|
assert result is True
|
||
|
volume = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = volume.unmount()
|
||
|
assert result is False
|
||
|
assert len(read_json(only_root_in_userdata / "only_root.json")["volumes"]) == 0
|
||
|
|
||
|
|
||
|
def test_unmount_block_device_when_undefined(
|
||
|
lsblk_singular_mock, undefined_devices_in_userdata, authorized_client
|
||
|
):
|
||
|
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
|
||
|
result = block_device.unmount()
|
||
|
assert result is False
|
||
|
assert (
|
||
|
len(read_json(undefined_devices_in_userdata / "undefined.json")["volumes"]) == 0
|
||
|
)
|
||
|
|
||
|
|
||
|
FULL_LSBLK_OUTPUT = b"""
|
||
|
{
|
||
|
"blockdevices": [
|
||
|
{
|
||
|
"name": "sda",
|
||
|
"path": "/dev/sda",
|
||
|
"fsavail": null,
|
||
|
"fssize": null,
|
||
|
"fstype": null,
|
||
|
"fsused": null,
|
||
|
"mountpoints": [
|
||
|
null
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": null,
|
||
|
"size": 20480786432,
|
||
|
"model": "QEMU HARDDISK",
|
||
|
"serial": "drive-scsi0-0-0-0",
|
||
|
"type": "disk",
|
||
|
"children": [
|
||
|
{
|
||
|
"name": "sda1",
|
||
|
"path": "/dev/sda1",
|
||
|
"fsavail": "4605702144",
|
||
|
"fssize": "19814920192",
|
||
|
"fstype": "ext4",
|
||
|
"fsused": "14353719296",
|
||
|
"mountpoints": [
|
||
|
"/nix/store", "/"
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": "ec80c004-baec-4a2c-851d-0e1807135511",
|
||
|
"size": 20210236928,
|
||
|
"model": null,
|
||
|
"serial": null,
|
||
|
"type": "part"
|
||
|
},{
|
||
|
"name": "sda14",
|
||
|
"path": "/dev/sda14",
|
||
|
"fsavail": null,
|
||
|
"fssize": null,
|
||
|
"fstype": null,
|
||
|
"fsused": null,
|
||
|
"mountpoints": [
|
||
|
null
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": null,
|
||
|
"size": 1048576,
|
||
|
"model": null,
|
||
|
"serial": null,
|
||
|
"type": "part"
|
||
|
},{
|
||
|
"name": "sda15",
|
||
|
"path": "/dev/sda15",
|
||
|
"fsavail": null,
|
||
|
"fssize": null,
|
||
|
"fstype": "vfat",
|
||
|
"fsused": null,
|
||
|
"mountpoints": [
|
||
|
null
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": "6B29-5BA7",
|
||
|
"size": 268435456,
|
||
|
"model": null,
|
||
|
"serial": null,
|
||
|
"type": "part"
|
||
|
}
|
||
|
]
|
||
|
},{
|
||
|
"name": "sdb",
|
||
|
"path": "/dev/sdb",
|
||
|
"fsavail": "11888545792",
|
||
|
"fssize": "12573614080",
|
||
|
"fstype": "ext4",
|
||
|
"fsused": "24047616",
|
||
|
"mountpoints": [
|
||
|
"/volumes/sdb"
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": "fa9d0026-ee23-4047-b8b1-297ae16fa751",
|
||
|
"size": 12884901888,
|
||
|
"model": "Volume",
|
||
|
"serial": "21378102",
|
||
|
"type": "disk"
|
||
|
},{
|
||
|
"name": "sr0",
|
||
|
"path": "/dev/sr0",
|
||
|
"fsavail": null,
|
||
|
"fssize": null,
|
||
|
"fstype": null,
|
||
|
"fsused": null,
|
||
|
"mountpoints": [
|
||
|
null
|
||
|
],
|
||
|
"label": null,
|
||
|
"uuid": null,
|
||
|
"size": 1073741312,
|
||
|
"model": "QEMU DVD-ROM",
|
||
|
"serial": "QM00003",
|
||
|
"type": "rom"
|
||
|
}
|
||
|
]
|
||
|
}
|
||
|
"""
|
||
|
|
||
|
|
||
|
@pytest.fixture
|
||
|
def lsblk_full_mock(mocker):
|
||
|
mock = mocker.patch(
|
||
|
"subprocess.check_output", autospec=True, return_value=FULL_LSBLK_OUTPUT
|
||
|
)
|
||
|
return mock
|
||
|
|
||
|
|
||
|
def test_get_block_devices(lsblk_full_mock, authorized_client):
|
||
|
block_devices = BlockDevices().get_block_devices()
|
||
|
assert len(block_devices) == 2
|
||
|
assert block_devices[0].name == "sda1"
|
||
|
assert block_devices[0].path == "/dev/sda1"
|
||
|
assert block_devices[0].fsavail == "4605702144"
|
||
|
assert block_devices[0].fssize == "19814920192"
|
||
|
assert block_devices[0].fstype == "ext4"
|
||
|
assert block_devices[0].fsused == "14353719296"
|
||
|
assert block_devices[0].mountpoints == ["/nix/store", "/"]
|
||
|
assert block_devices[0].label is None
|
||
|
assert block_devices[0].uuid == "ec80c004-baec-4a2c-851d-0e1807135511"
|
||
|
assert block_devices[0].size == "20210236928"
|
||
|
assert block_devices[0].model is None
|
||
|
assert block_devices[0].serial is None
|
||
|
assert block_devices[0].type == "part"
|
||
|
assert block_devices[1].name == "sdb"
|
||
|
assert block_devices[1].path == "/dev/sdb"
|
||
|
assert block_devices[1].fsavail == "11888545792"
|
||
|
assert block_devices[1].fssize == "12573614080"
|
||
|
assert block_devices[1].fstype == "ext4"
|
||
|
assert block_devices[1].fsused == "24047616"
|
||
|
assert block_devices[1].mountpoints == ["/volumes/sdb"]
|
||
|
assert block_devices[1].label is None
|
||
|
assert block_devices[1].uuid == "fa9d0026-ee23-4047-b8b1-297ae16fa751"
|
||
|
assert block_devices[1].size == "12884901888"
|
||
|
assert block_devices[1].model == "Volume"
|
||
|
assert block_devices[1].serial == "21378102"
|
||
|
assert block_devices[1].type == "disk"
|
||
|
|
||
|
|
||
|
def test_get_block_device(lsblk_full_mock, authorized_client):
|
||
|
block_device = BlockDevices().get_block_device("sda1")
|
||
|
assert block_device is not None
|
||
|
assert block_device.name == "sda1"
|
||
|
assert block_device.path == "/dev/sda1"
|
||
|
assert block_device.fsavail == "4605702144"
|
||
|
assert block_device.fssize == "19814920192"
|
||
|
assert block_device.fstype == "ext4"
|
||
|
assert block_device.fsused == "14353719296"
|
||
|
assert block_device.mountpoints == ["/nix/store", "/"]
|
||
|
assert block_device.label is None
|
||
|
assert block_device.uuid == "ec80c004-baec-4a2c-851d-0e1807135511"
|
||
|
assert block_device.size == "20210236928"
|
||
|
assert block_device.model is None
|
||
|
assert block_device.serial is None
|
||
|
assert block_device.type == "part"
|
||
|
|
||
|
|
||
|
def test_get_nonexistent_block_device(lsblk_full_mock, authorized_client):
|
||
|
block_device = BlockDevices().get_block_device("sda2")
|
||
|
assert block_device is None
|
||
|
|
||
|
|
||
|
def test_get_block_devices_by_mountpoint(lsblk_full_mock, authorized_client):
|
||
|
block_devices = BlockDevices().get_block_devices_by_mountpoint("/nix/store")
|
||
|
assert len(block_devices) == 1
|
||
|
assert block_devices[0].name == "sda1"
|
||
|
assert block_devices[0].path == "/dev/sda1"
|
||
|
assert block_devices[0].fsavail == "4605702144"
|
||
|
assert block_devices[0].fssize == "19814920192"
|
||
|
assert block_devices[0].fstype == "ext4"
|
||
|
assert block_devices[0].fsused == "14353719296"
|
||
|
assert block_devices[0].mountpoints == ["/nix/store", "/"]
|
||
|
assert block_devices[0].label is None
|
||
|
assert block_devices[0].uuid == "ec80c004-baec-4a2c-851d-0e1807135511"
|
||
|
assert block_devices[0].size == "20210236928"
|
||
|
assert block_devices[0].model is None
|
||
|
assert block_devices[0].serial is None
|
||
|
assert block_devices[0].type == "part"
|
||
|
|
||
|
|
||
|
def test_get_block_devices_by_mountpoint_no_match(lsblk_full_mock, authorized_client):
|
||
|
block_devices = BlockDevices().get_block_devices_by_mountpoint("/foo")
|
||
|
assert len(block_devices) == 0
|