From 57939654e955a1b0b64baa3c3ee670566dd26e6b Mon Sep 17 00:00:00 2001 From: Russell Haering Date: Tue, 28 Jan 2014 11:25:00 -0800 Subject: [PATCH] add test of list_hardware_info --- teeth_agent/tests/hardware.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/teeth_agent/tests/hardware.py b/teeth_agent/tests/hardware.py index 180a564c7..6435b683d 100644 --- a/teeth_agent/tests/hardware.py +++ b/teeth_agent/tests/hardware.py @@ -53,3 +53,17 @@ class TestGenericHardwareManager(unittest.TestCase): self.assertEqual(self.hardware.get_os_install_device(), '/dev/sdb') self.hardware._cmd.assert_called_once_with('blockdev') blockdev.assert_called_once_with('--report') + + def test_list_hardwre_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'), + ] + + hardware_info = self.hardware.list_hardware_info() + self.assertEqual(len(hardware_info), 2) + self.assertEqual(hardware_info[0].type, 'mac_address') + self.assertEqual(hardware_info[1].type, 'mac_address') + self.assertEqual(hardware_info[0].id, '00:0c:29:8c:11:b1') + self.assertEqual(hardware_info[1].id, '00:0c:29:8c:11:b2')