def test_disk_get_specified_filesystems(self):
"""Tests `Disk._get_specified_filesystems`."""
# This minimal `_disk_dict` contains everything this method touches.
self.disk_instance_mock._disk_dict = { # pylint: disable=protected-access
'/very/good/mountpoint': {
'device_path': '/dev/sda1'
},
'/mounted/here/too': {
'device_path': '/dev/sda1'
},
'/less/good/mountpoint': {
'device_path': '/dev/sda2'
},
'/a/samba/share': {
'device_path': '//server.local/cool_share'
}
}
with self.subTest('Get all filesystems with mount points.'):
# pylint: disable=protected-access
self.assertDictEqual(
Disk._get_specified_filesystems(
self.disk_instance_mock,
self.disk_instance_mock._disk_dict # recall dicts are iterables of their keys.
),
self.disk_instance_mock._disk_dict
)
# pylint: enable=protected-access
with self.subTest('Get only `/dev/sda1` filesystems.'):
self.assertDictEqual(
Disk._get_specified_filesystems( # pylint: disable=protected-access
self.disk_instance_mock,
('/dev/sda1',)
),
{
'/very/good/mountpoint': {
'device_path': '/dev/sda1'
}
}
)
def test_disk_get_specified_filesystems(self):
"""Tests `Disk._get_specified_filesystems`."""
# This minimal `_disk_dict` contains everything this method touches.
self.disk_instance_mock._disk_dict = { # pylint: disable=protected-access
'/very/good/mountpoint': {
'device_path': '/dev/sda1'
},
'/mounted/here/too': {
'device_path': '/dev/sda1'
},
'/less/good/mountpoint': {
'device_path': '/dev/sda2'
},
'/a/samba/share': {
'device_path': '//server.local/cool_share'
}
}
with self.subTest('Get all filesystems with mount points.'):
# pylint: disable=protected-access
self.assertDictEqual(
Disk._get_specified_filesystems(
self.disk_instance_mock,
self.disk_instance_mock.
_disk_dict # recall dicts are iterables of their keys.
),
self.disk_instance_mock._disk_dict)
# pylint: enable=protected-access
with self.subTest('Get only `/dev/sda1` filesystems.'):
result_disk_dict = Disk._get_specified_filesystems( # pylint: disable=protected-access
self.disk_instance_mock, ('/dev/sda1', ))
# With Python < 3.6, dict ordering isn't guaranteed,
# so we don't know which disk will be selected.
self.assertEqual(len(result_disk_dict), 1)
# As long as `device_path` is also correct, this passes.
self.assertEqual(
result_disk_dict[list(
result_disk_dict.keys())[0]]['device_path'], '/dev/sda1')
