def test_get_devices_label():
c = Disk('disk', None, {}, [{}])
with mock.patch(
"datadog_checks.disk.disk.get_subprocess_output",
return_value=mock_blkid_output(),
__name__='get_subprocess_output',
):
labels = c._get_devices_label()
assert labels.get("/dev/mapper/vagrant--vg-root") == "label:DATA"
def test_get_devices_label_options():
"""
Verify that
- use_lsblk uses lsblk for labels,
- blkid_cache_file not null makes it use the blkid_cache and
- by default we use the blkid command.
"""
c_lsblk = Disk('disk', {}, [{'use_lsblk': True}])
c_blkid = Disk('disk', {}, [{}])
c_blkid_cache = Disk('disk', {}, [{'blkid_cache_file': 'filepath'}])
prefix_fun = "datadog_checks.disk.disk.Disk._get_devices_label_from_"
with mock.patch(prefix_fun + "lsblk", return_value="lsblk"), mock.patch(
prefix_fun + "blkid",
return_value="blkid"), mock.patch(prefix_fun + "blkid_cache",
return_value="blkid_cache"):
assert c_lsblk._get_devices_label() == "lsblk"
assert c_blkid._get_devices_label() == "blkid"
assert c_blkid_cache._get_devices_label() == "blkid_cache"
