1384d79d2c
Bumping the payload version to two, and adding CPU, Memory, and Block Device information to the inventory. Added dependency on psutil for available memory and cpu_count. Requirements line copied from global requirements Change-Id: Ia39c85c91b1d60468667787a7978020084dc6c2a
165 lines
7.0 KiB
Python
165 lines
7.0 KiB
Python
"""
|
|
Copyright 2013 Rackspace, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
"""
|
|
|
|
import mock
|
|
from oslotest import base as test_base
|
|
import six
|
|
|
|
from ironic_python_agent import hardware
|
|
from ironic_python_agent import utils
|
|
|
|
if six.PY2:
|
|
OPEN_FUNCTION_NAME = '__builtin__.open'
|
|
else:
|
|
OPEN_FUNCTION_NAME = 'builtins.open'
|
|
|
|
|
|
class TestGenericHardwareManager(test_base.BaseTestCase):
|
|
def setUp(self):
|
|
super(TestGenericHardwareManager, self).setUp()
|
|
self.hardware = hardware.GenericHardwareManager()
|
|
|
|
@mock.patch('os.listdir')
|
|
@mock.patch('os.path.exists')
|
|
@mock.patch(OPEN_FUNCTION_NAME)
|
|
def test_list_network_interfaces(self,
|
|
mocked_open,
|
|
mocked_exists,
|
|
mocked_listdir):
|
|
mocked_listdir.return_value = ['lo', 'eth0']
|
|
mocked_exists.side_effect = [False, True]
|
|
mocked_open.return_value.__enter__ = lambda s: s
|
|
mocked_open.return_value.__exit__ = mock.Mock()
|
|
read_mock = mocked_open.return_value.read
|
|
read_mock.return_value = '00:0c:29:8c:11:b1\n'
|
|
interfaces = self.hardware.list_network_interfaces()
|
|
self.assertEqual(len(interfaces), 1)
|
|
self.assertEqual(interfaces[0].name, 'eth0')
|
|
self.assertEqual(interfaces[0].mac_address, '00:0c:29:8c:11:b1')
|
|
|
|
@mock.patch.object(utils, 'execute')
|
|
def test_get_os_install_device(self, mocked_execute):
|
|
mocked_execute.return_value = (
|
|
'RO RA SSZ BSZ StartSec Size Device\n'
|
|
'rw 256 512 4096 0 249578283616 /dev/sda\n'
|
|
'rw 256 512 4096 2048 8587837440 /dev/sda1\n'
|
|
'rw 256 512 4096 124967424 15728640 /dev/sda2\n'
|
|
'rw 256 512 4096 0 31016853504 /dev/sdb\n'
|
|
'rw 256 512 4096 0 249578283616 /dev/sdc\n', '')
|
|
|
|
self.assertEqual(self.hardware.get_os_install_device(), '/dev/sdb')
|
|
mocked_execute.assert_called_once_with('blockdev',
|
|
'--report',
|
|
check_exit_code=[0])
|
|
|
|
@mock.patch('psutil.cpu_count')
|
|
@mock.patch(OPEN_FUNCTION_NAME)
|
|
def test_get_cpus(self, mocked_open, mocked_cpucount):
|
|
mocked_open.return_value.__enter__ = lambda s: s
|
|
mocked_open.return_value.__exit__ = mock.Mock()
|
|
read_mock = mocked_open.return_value.read
|
|
read_mock.return_value = (
|
|
'processor : 0\n'
|
|
'vendor_id : GenuineIntel\n'
|
|
'cpu family : 6\n'
|
|
'model : 58\n'
|
|
'model name : Intel(R) Core(TM) i7-3720QM CPU @ 2.60GHz\n'
|
|
'stepping : 9\n'
|
|
'microcode : 0x15\n'
|
|
'cpu MHz : 2594.685\n'
|
|
'cache size : 6144 KB\n'
|
|
'fpu : yes\n'
|
|
'fpu_exception : yes\n'
|
|
'cpuid level : 13\n'
|
|
'wp : yes\n'
|
|
'flags : fpu vme de pse tsc msr pae mce cx8 apic sep '
|
|
'mtrr pge mca cmov pat pse36 clflush dts mmx fxsr sse sse2 ss '
|
|
'syscall nx rdtscp lm constant_tsc arch_perfmon pebs bts nopl '
|
|
'xtopology tsc_reliable nonstop_tsc aperfmperf eagerfpu pni '
|
|
'pclmulqdq ssse3 cx16 pcid sse4_1 sse4_2 x2apic popcnt aes xsave '
|
|
'avx f16c rdrand hypervisor lahf_lm ida arat epb xsaveopt pln pts '
|
|
'dtherm fsgsbase smep\n'
|
|
'bogomips : 5189.37\n'
|
|
'clflush size : 64\n'
|
|
'cache_alignment : 64\n'
|
|
'address sizes : 40 bits physical, 48 bits virtual\n'
|
|
'power management:\n'
|
|
'\n'
|
|
'processor : 1\n'
|
|
'vendor_id : GenuineIntel\n'
|
|
'cpu family : 6\n'
|
|
'model : 58\n'
|
|
'model name : Intel(R) Core(TM) i7-3720QM CPU @ 2.60GHz\n'
|
|
'stepping : 9\n'
|
|
'microcode : 0x15\n'
|
|
'cpu MHz : 2594.685\n'
|
|
'cache size : 6144 KB\n'
|
|
'fpu : yes\n'
|
|
'fpu_exception : yes\n'
|
|
'cpuid level : 13\n'
|
|
'wp : yes\n'
|
|
'flags : fpu vme de pse tsc msr pae mce cx8 apic sep '
|
|
'mtrr pge mca cmov pat pse36 clflush dts mmx fxsr sse sse2 ss '
|
|
'syscall nx rdtscp lm constant_tsc arch_perfmon pebs bts nopl '
|
|
'xtopology tsc_reliable nonstop_tsc aperfmperf eagerfpu pni '
|
|
'pclmulqdq ssse3 cx16 pcid sse4_1 sse4_2 x2apic popcnt aes xsave '
|
|
'avx f16c rdrand hypervisor lahf_lm ida arat epb xsaveopt pln pts '
|
|
'dtherm fsgsbase smep\n'
|
|
'bogomips : 5189.37\n'
|
|
'clflush size : 64\n'
|
|
'cache_alignment : 64\n'
|
|
'address sizes : 40 bits physical, 48 bits virtual\n'
|
|
'power management:\n'
|
|
)
|
|
|
|
mocked_cpucount.return_value = 2
|
|
|
|
cpus = self.hardware.get_cpus()
|
|
self.assertEqual(cpus.model_name,
|
|
'Intel(R) Core(TM) i7-3720QM CPU @ 2.60GHz')
|
|
self.assertEqual(cpus.frequency, '2594.685')
|
|
self.assertEqual(cpus.count, 2)
|
|
|
|
def test_list_hardware_info(self):
|
|
self.hardware.list_network_interfaces = mock.Mock()
|
|
self.hardware.list_network_interfaces.return_value = [
|
|
hardware.NetworkInterface('eth0', '00:0c:29:8c:11:b1'),
|
|
hardware.NetworkInterface('eth1', '00:0c:29:8c:11:b2'),
|
|
]
|
|
|
|
self.hardware.get_cpus = mock.Mock()
|
|
self.hardware.get_cpus.return_value = hardware.CPU(
|
|
'Awesome CPU x14 9001',
|
|
9001,
|
|
14)
|
|
|
|
self.hardware.get_memory = mock.Mock()
|
|
self.hardware.get_memory.return_value = hardware.Memory(1017012)
|
|
|
|
self.hardware.list_block_devices = mock.Mock()
|
|
self.hardware.list_block_devices.return_value = [
|
|
hardware.BlockDevice('/dev/sdj', 1073741824),
|
|
hardware.BlockDevice('/dev/hdaa', 65535),
|
|
]
|
|
|
|
hardware_info = self.hardware.list_hardware_info()
|
|
self.assertEqual(hardware_info['memory'], self.hardware.get_memory())
|
|
self.assertEqual(hardware_info['cpu'], self.hardware.get_cpus())
|
|
self.assertEqual(hardware_info['disks'],
|
|
self.hardware.list_block_devices())
|
|
self.assertEqual(hardware_info['interfaces'],
|
|
self.hardware.list_network_interfaces())
|