def test_get_info(self, mock_exec_command, mock_dmidecode_parser_cls): mock_exec_command.return_value = 'blah' mparser = mock_dmidecode_parser_cls.return_value = mock.Mock() mparser.parse.return_value = {'key': 'value'} host = inspector.Host() rec = host.get_info() self.assertEqual(rec, {'key': 'value'})
def test_is_remote(self, get_ssh_client): get_ssh_client.return_value = mock.MagicMock() host = inspector.Host('test', 'user', 'pass') self.assertEqual(host.is_remote(), True)
def test_is_not_remote(self): host = inspector.Host() self.assertEqual(host.is_remote(), False)
def test_get_pci_devices(self, exec_command): host = inspector.Host() devs = host.get_pci_devices() exec_command.assert_called_once_with(['lspci', '-nnmm'])
def test_remote_exec_command(self, remote_command, get_ssh_client): mclient = get_ssh_client.return_value = mock.MagicMock() host = inspector.Host('mymachine', 'root', 'pass') host.exec_command('ls') inspector.remote_command.assert_called_once_with(mclient, 'ls')
def test_local_exec_command(self, local_command): host = inspector.Host() host.exec_command('ls') inspector.local_command.assert_called_once_with('ls')