SelfPrivacy
/
selfprivacy-rest-api
7
2
Fork 1
Code Issues 15 Pull Requests 3 Projects Releases 1 Wiki Activity

Add jobs mocking to tests explicitly

pull/13/head
Inex Code 2022-08-18 01:38:38 +04:00
parent e5584e0e1c
commit 19168dfdaf
3 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/conftest.py
View File

@ -17,7 +17,10 @@ def tokens_file(mocker, shared_datadir):
@pytest.fixture @pytest.fixture
def jobs_file(mocker, shared_datadir): def jobs_file(mocker, shared_datadir):
"""Mock tokens file.""" """Mock tokens file."""
mock = mocker.patch("selfprivacy_api.utils.JOBS_FILE", shared_datadir / "jobs.json") mock = mocker.patch(
"selfprivacy_api.utils.JOBS_FILE",
shared_datadir / "jobs.json"
)
return mock return mock

32
tests/test_block_device_utils.py
View File

@ -85,7 +85,7 @@ def undefined_devices_in_userdata(mocker, datadir):
return datadir return datadir
def test_create_block_device_object(lsblk_singular_mock): def test_create_block_device_object(lsblk_singular_mock, authorized_client):
output = get_block_device("sda1") output = get_block_device("sda1")
assert lsblk_singular_mock.call_count == 1 assert lsblk_singular_mock.call_count == 1
assert lsblk_singular_mock.call_args[0][0] == [ assert lsblk_singular_mock.call_args[0][0] == [
@ -99,7 +99,7 @@ def test_create_block_device_object(lsblk_singular_mock):
assert output == json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0] assert output == json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]
def test_resize_block_device(lsblk_singular_mock): def test_resize_block_device(lsblk_singular_mock, authorized_client):
result = resize_block_device("sdb") result = resize_block_device("sdb")
assert result is True assert result is True
assert lsblk_singular_mock.call_count == 1 assert lsblk_singular_mock.call_count == 1
@ -109,7 +109,7 @@ def test_resize_block_device(lsblk_singular_mock):
] ]
def test_resize_block_device_failed(failed_check_output_mock): def test_resize_block_device_failed(failed_check_output_mock, authorized_client):
result = resize_block_device("sdb") result = resize_block_device("sdb")
assert result is False assert result is False
assert failed_check_output_mock.call_count == 1 assert failed_check_output_mock.call_count == 1
@ -144,7 +144,7 @@ VOLUME_LSBLK_OUTPUT = b"""
""" """
def test_create_block_device(lsblk_singular_mock): def test_create_block_device(lsblk_singular_mock, authorized_client):
block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
assert block_device.name == "sdb" assert block_device.name == "sdb"
@ -169,7 +169,7 @@ def test_create_block_device(lsblk_singular_mock):
assert hash(block_device) == hash("sdb") assert hash(block_device) == hash("sdb")
def test_block_devices_equal(lsblk_singular_mock): def test_block_devices_equal(lsblk_singular_mock, authorized_client):
block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
block_device2 = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0]) block_device2 = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
@ -186,7 +186,7 @@ def resize_block_mock(mocker):
return mock return mock
def test_call_resize_from_block_device(lsblk_singular_mock, resize_block_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 = BlockDevice(json.loads(VOLUME_LSBLK_OUTPUT)["blockdevices"][0])
block_device.resize() block_device.resize()
assert resize_block_mock.call_count == 1 assert resize_block_mock.call_count == 1
@ -194,7 +194,7 @@ def test_call_resize_from_block_device(lsblk_singular_mock, resize_block_mock):
assert lsblk_singular_mock.call_count == 0 assert lsblk_singular_mock.call_count == 0
def test_get_stats_from_block_device(lsblk_singular_mock): def test_get_stats_from_block_device(lsblk_singular_mock, authorized_client):
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
stats = block_device.stats() stats = block_device.stats()
assert stats == { assert stats == {
@ -223,7 +223,7 @@ def test_get_stats_from_block_device(lsblk_singular_mock):
] ]
def test_mount_block_device(lsblk_singular_mock, only_root_in_userdata): def test_mount_block_device(lsblk_singular_mock, only_root_in_userdata, authorized_client):
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
result = block_device.mount() result = block_device.mount()
assert result is False assert result is False
@ -245,7 +245,7 @@ def test_mount_block_device(lsblk_singular_mock, only_root_in_userdata):
def test_mount_block_device_when_undefined( def test_mount_block_device_when_undefined(
lsblk_singular_mock, undefined_devices_in_userdata lsblk_singular_mock, undefined_devices_in_userdata, authorized_client
): ):
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
result = block_device.mount() result = block_device.mount()
@ -270,7 +270,7 @@ def test_mount_block_device_when_undefined(
) )
def test_unmount_block_device(lsblk_singular_mock, only_root_in_userdata): def test_unmount_block_device(lsblk_singular_mock, only_root_in_userdata, authorized_client):
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
result = block_device.unmount() result = block_device.unmount()
assert result is True assert result is True
@ -281,7 +281,7 @@ def test_unmount_block_device(lsblk_singular_mock, only_root_in_userdata):
def test_unmount_block_device_when_undefined( def test_unmount_block_device_when_undefined(
lsblk_singular_mock, undefined_devices_in_userdata lsblk_singular_mock, undefined_devices_in_userdata, authorized_client
): ):
block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0]) block_device = BlockDevice(json.loads(SINGLE_LSBLK_OUTPUT)["blockdevices"][0])
result = block_device.unmount() result = block_device.unmount()
@ -407,7 +407,7 @@ def lsblk_full_mock(mocker):
return mock return mock
def test_get_block_devices(lsblk_full_mock): def test_get_block_devices(lsblk_full_mock, authorized_client):
block_devices = BlockDevices().get_block_devices() block_devices = BlockDevices().get_block_devices()
assert len(block_devices) == 2 assert len(block_devices) == 2
assert block_devices[0].name == "sda1" assert block_devices[0].name == "sda1"
@ -438,7 +438,7 @@ def test_get_block_devices(lsblk_full_mock):
assert block_devices[1].type == "disk" assert block_devices[1].type == "disk"
def test_get_block_device(lsblk_full_mock): def test_get_block_device(lsblk_full_mock, authorized_client):
block_device = BlockDevices().get_block_device("sda1") block_device = BlockDevices().get_block_device("sda1")
assert block_device is not None assert block_device is not None
assert block_device.name == "sda1" assert block_device.name == "sda1"
@ -456,12 +456,12 @@ def test_get_block_device(lsblk_full_mock):
assert block_device.type == "part" assert block_device.type == "part"
def test_get_nonexistent_block_device(lsblk_full_mock): def test_get_nonexistent_block_device(lsblk_full_mock, authorized_client):
block_device = BlockDevices().get_block_device("sda2") block_device = BlockDevices().get_block_device("sda2")
assert block_device is None assert block_device is None
def test_get_block_devices_by_mountpoint(lsblk_full_mock): def test_get_block_devices_by_mountpoint(lsblk_full_mock, authorized_client):
block_devices = BlockDevices().get_block_devices_by_mountpoint("/nix/store") block_devices = BlockDevices().get_block_devices_by_mountpoint("/nix/store")
assert len(block_devices) == 1 assert len(block_devices) == 1
assert block_devices[0].name == "sda1" assert block_devices[0].name == "sda1"
@ -479,6 +479,6 @@ def test_get_block_devices_by_mountpoint(lsblk_full_mock):
assert block_devices[0].type == "part" assert block_devices[0].type == "part"
def test_get_block_devices_by_mountpoint_no_match(lsblk_full_mock): def test_get_block_devices_by_mountpoint_no_match(lsblk_full_mock, authorized_client):
block_devices = BlockDevices().get_block_devices_by_mountpoint("/foo") block_devices = BlockDevices().get_block_devices_by_mountpoint("/foo")
assert len(block_devices) == 0 assert len(block_devices) == 0

2
tests/test_jobs.py
View File

@ -7,7 +7,7 @@ from selfprivacy_api.utils import WriteUserData, ReadUserData
from selfprivacy_api.jobs import Jobs, JobStatus from selfprivacy_api.jobs import Jobs, JobStatus
def test_jobs(jobs_file, shared_datadir): def test_jobs(authorized_client, jobs_file, shared_datadir):
jobs = Jobs() jobs = Jobs()
assert jobs.get_jobs() == [] assert jobs.get_jobs() == []