def test_get_disk_key_3(self):
"""
virtual is False
key is found
retrun value of key is expected
"""
disk_map = Device(path='/dev/vdb', sys_api={'foo': 'bar'})
ret = drive_selection.Matcher('foo', 'bar')._get_disk_key(disk_map)
assert ret is disk_map.sys_api.get('foo')
def test_get_disk_key_4(self):
"""
virtual is False
key is not found
expect raise Exception
"""
disk_map = Device(path='/dev/vdb')
with pytest.raises(Exception):
drive_selection.Matcher('bar', 'foo')._get_disk_key(disk_map)
pytest.fail("No disk_key found for foo or None")
