191 lines
8.0 KiB
Python
191 lines
8.0 KiB
Python
# Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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.
|
|
|
|
"""Test class for IPMI Module."""
|
|
|
|
import subprocess
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from proliantutils.ilo import ipmi
|
|
from proliantutils.tests.ilo import ipmi_sample_outputs as constants
|
|
|
|
|
|
class IloIpmiTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
super(IloIpmiTestCase, self).setUp()
|
|
self.info = {'address': "x.x.x.x",
|
|
'username': "admin",
|
|
'password': "Admin"}
|
|
|
|
def test_get_ilo_version_valid_1(self):
|
|
in_vals = '2.03'
|
|
expected = 2.03
|
|
actual = ipmi.get_ilo_version(in_vals)
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_get_ilo_version_valid_2(self):
|
|
in_vals = '2.3'
|
|
expected = 2.30
|
|
actual = ipmi.get_ilo_version(in_vals)
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_get_ilo_version_invalid(self):
|
|
in_vals = 'x.y'
|
|
expected = None
|
|
actual = ipmi.get_ilo_version(in_vals)
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_get_ilo_version_appended_spaces(self):
|
|
in_vals = ' 2.50 '
|
|
expected = 2.5
|
|
actual = ipmi.get_ilo_version(in_vals)
|
|
self.assertEqual(actual, expected)
|
|
|
|
def test_get_ilo_version_none(self):
|
|
in_vals = None
|
|
expected = None
|
|
actual = ipmi.get_ilo_version(in_vals)
|
|
self.assertEqual(actual, expected)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_lt_suggested(self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.LESSER_THAN_MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.return_value = constants.NIC_FRU_OUT
|
|
parse_mock.return_value = "1Gb"
|
|
expected_out = "1Gb"
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertEqual(expected_out, actual_out)
|
|
ipmi_mock.assert_called_once_with(mock.ANY, "fru print 0x0")
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_lt_suggested_none(self,
|
|
ipmi_mock,
|
|
parse_mock):
|
|
fw_rev = constants.LESSER_THAN_MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.return_value = constants.NIC_FRU_OUT
|
|
parse_mock.return_value = None
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertIsNone(actual_out)
|
|
self.assertEqual(ipmi_mock.call_count, 126)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_lt_suggested_out_of_range_check(
|
|
self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.LESSER_THAN_MIN_SUGGESTED_FW_STR
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
format_call_args = map(lambda x: x[0][1],
|
|
ipmi_mock.call_args_list)
|
|
self.assertNotIn("fru print 0x7d", format_call_args)
|
|
self.assertEqual(actual_out, None)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_lt_suggested_in_range_check(
|
|
self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.LESSER_THAN_MIN_SUGGESTED_FW_STR
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
format_call_args = map(lambda x: x[0][1],
|
|
ipmi_mock.call_args_list)
|
|
self.assertIn("fru print 0xef", format_call_args)
|
|
self.assertEqual(actual_out, None)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_suggested(self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.return_value = constants.NIC_FRU_OUT_ALL
|
|
parse_mock.return_value = "1Gb"
|
|
expected_out = "1Gb"
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertEqual(expected_out, actual_out)
|
|
ipmi_mock.assert_called_once_with(mock.ANY, "fru print")
|
|
self.assertEqual(ipmi_mock.call_count, 1)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_suggested_none(self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.return_value = constants.NIC_FRU_OUT_ALL
|
|
parse_mock.return_value = None
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertIsNone(actual_out)
|
|
self.assertEqual(ipmi_mock.call_count, 1)
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_gt_suggested(self, ipmi_mock, parse_mock):
|
|
fw_rev = constants.GREATER_THAN_MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.return_value = constants.NIC_FRU_OUT_ALL
|
|
parse_mock.return_value = "1Gb"
|
|
expected_out = "1Gb"
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertEqual(expected_out, actual_out)
|
|
ipmi_mock.assert_called_once_with(mock.ANY, "fru print")
|
|
|
|
@mock.patch.object(ipmi, '_parse_ipmi_nic_capacity')
|
|
@mock.patch.object(ipmi, '_exec_ipmitool')
|
|
def test_get_nic_capacity_fw_lt_suggested_loop_N_times(self,
|
|
ipmi_mock,
|
|
parse_mock):
|
|
fw_rev = constants.LESSER_THAN_MIN_SUGGESTED_FW_STR
|
|
ipmi_mock.side_effect = ["Device not present", "Device not present",
|
|
"Device not present", "Device not present",
|
|
"Device not present", "Device not present",
|
|
"Device not present", "Device not present",
|
|
constants.NIC_FRU_OUT]
|
|
parse_mock.return_value = "1Gb"
|
|
expected_out = "1Gb"
|
|
actual_out = ipmi.get_nic_capacity(self.info, fw_rev)
|
|
self.assertEqual(ipmi_mock.call_count, 9)
|
|
self.assertEqual(expected_out, actual_out)
|
|
|
|
@mock.patch.object(subprocess, 'Popen')
|
|
def test__exec_ipmitool(self, popen_mock):
|
|
pro_obj = mock.MagicMock()
|
|
popen_mock.return_value = pro_obj
|
|
pro_obj.communicate.return_value = constants.NIC_FRU_OUT_TUPLE
|
|
expected_output = constants.NIC_FRU_OUT
|
|
cmd = "fru print 0x64"
|
|
actual_out = ipmi._exec_ipmitool(self.info, cmd)
|
|
self.assertEqual(expected_output, actual_out)
|
|
|
|
@mock.patch.object(subprocess, 'Popen')
|
|
def test__exec_ipmitool_none(self, popen_mock):
|
|
popen_mock.side_effect = Exception
|
|
cmd = "fru print 0x2"
|
|
actual_out = ipmi._exec_ipmitool(self.info, cmd)
|
|
self.assertIsNone(actual_out)
|
|
|
|
def test__parse_ipmi_nic_capacity(self):
|
|
exec_output = constants.NIC_FRU_OUT
|
|
expected_output = "1Gb"
|
|
actual_out = ipmi._parse_ipmi_nic_capacity(exec_output)
|
|
self.assertEqual(expected_output, actual_out)
|
|
|
|
def test__parse_ipmi_nic_capacity_no_port_details(self):
|
|
exec_output = constants.NIC_FRU_OUT_NO_PORT_DETAILS
|
|
actual_out = ipmi._parse_ipmi_nic_capacity(exec_output)
|
|
self.assertIsNone(actual_out)
|
|
|
|
def test__parse_ipmi_nic_capacity_no_product_name(self):
|
|
exec_output = constants.NIC_FRU_OUT_NO_PRODUCT_NAME
|
|
actual_out = ipmi._parse_ipmi_nic_capacity(exec_output)
|
|
self.assertIsNone(actual_out)
|