def test__get_nm_address_already_detected(self, parse_mock, dump_mock, raw_mock, unlink_mock): internal_info = self.node.driver_internal_info internal_info['intel_nm_channel'] = '0x0B' internal_info['intel_nm_address'] = '0x0A' self.node.driver_internal_info = internal_info self.node.save() with task_manager.acquire(self.context, self.node.uuid, shared=False) as task: ret = nm_vendor._get_nm_address(task) self.assertEqual(('0x0B', '0x0A'), ret) self.assertFalse(parse_mock.called) self.assertFalse(dump_mock.called) self.assertFalse(raw_mock.called) self.assertFalse(unlink_mock.called)
def test__get_nm_address_detected(self, parse_mock, dump_mock, raw_mock, unlink_mock): parse_mock.return_value = ('0x0A', '0x0B') with task_manager.acquire(self.context, self.node.uuid, shared=False) as task: ret = nm_vendor._get_nm_address(task) self.assertEqual(('0x0B', '0x0A'), ret) self.node.refresh() internal_info = self.node.driver_internal_info self.assertEqual('0x0A', internal_info['intel_nm_address']) self.assertEqual('0x0B', internal_info['intel_nm_channel']) parse_mock.assert_called_once_with(self.temp_filename) dump_mock.assert_called_once_with(task, self.temp_filename) unlink_mock.assert_called_once_with(self.temp_filename) raw_mock.assert_called_once_with(task, mock.ANY)