def test_get_disk_key_1(self, virtual):
"""
virtual is True
key is found
return value of key is expected
"""
disk_map = dict(path='/dev/vdb', foo='bar')
ret = dg.Matcher('foo', 'bar')._get_disk_key(disk_map)
assert ret == disk_map.get('foo')
def test_get_disk_key_3(self, virtual_mock):
"""
virtual is False
key is found
retrun value of key is expected
"""
virtual_mock.return_value = False
disk_map = dict(path='/dev/vdb', foo='bar')
ret = dg.Matcher('foo', 'bar')._get_disk_key(disk_map)
assert ret is disk_map.get('foo')
def test_get_disk_key_2(self, virtual_mock):
"""
virtual is True
key is not found
retrun False is expected
"""
virtual_mock.return_value = True
disk_map = dict(path='/dev/vdb')
ret = dg.Matcher('bar', 'foo')._get_disk_key(disk_map)
assert ret is ''
def test_get_disk_key_4(self, virtual_mock):
"""
virtual is False
key is not found
expect raise Exception
"""
virtual_mock.return_value = False
disk_map = dict(path='/dev/vdb')
with pytest.raises(Exception):
dg.Matcher('bar', 'foo')._get_disk_key(disk_map)
pytest.fail("No disk_key found for foo or None")
def test_virtual_1(self):
""" all hosts are physical
"""
dg.__grains__ = {'virtual': 'pysical'}
obj = dg.Matcher(None, None)
obj.virtual is False
def test_virtual(self):
""" Test if virtual
"""
dg.__grains__ = {'virtual': 'kvm'}
obj = dg.Matcher(None, None)
obj.virtual is True
