def test_get_dut(self):
resmixer = ResourceFunctions(mock.MagicMock(), mock.MagicMock(), mock.MagicMock())
resmixer._logger = MockLogger()
mocked_dut = self._create_dut()
resmixer.duts = [mocked_dut]
self.assertEqual(resmixer.get_dut(1), mocked_dut)
with self.assertRaises(ValueError):
resmixer.get_dut(2)
def test_dut_release(self, mock_time):
mocked_dut = self._create_dut()
resmixer = ResourceFunctions(mock.MagicMock(), mock.MagicMock(), mock.MagicMock())
resmixer.duts = [mocked_dut]
resmixer._args = MockArgs()
resmixer._logger = MockLogger()
resmixer._resource_provider = mock.MagicMock()
resmixer._call_exception = mock.MagicMock()
resmixer.is_my_dut_index = mock.MagicMock(return_value=True)
resmixer.duts_release()
mocked_dut.close_dut.assert_called_once()
mocked_dut.close_connection.assert_called_once()
resmixer.duts_release()
self.assertEqual(len(resmixer.duts), 0)
def test_dut_release_exception(self, mock_time):
mocked_dut = self._create_dut()
mocked_dut.close_connection = mock.MagicMock(side_effect=[Exception])
resmixer = ResourceFunctions(mock.MagicMock(), mock.MagicMock(), mock.MagicMock())
resmixer.duts = [mocked_dut]
resmixer._args = MockArgs()
resmixer._logger = MockLogger()
resmixer._resource_provider = mock.MagicMock()
resmixer._resource_provider.allocator = mock.MagicMock()
type(resmixer._resource_provider.allocator).share_allocations = mock.PropertyMock(
return_value=False)
resmixer._resource_provider.allocator.release = mock.MagicMock()
resmixer._call_exception = mock.MagicMock()
resmixer.is_my_dut_index = mock.MagicMock(return_value=True)
resmixer.duts_release()
mocked_dut.close_dut.assert_called_once()
mocked_dut.close_connection.assert_called_once()
resmixer._resource_provider.allocator.release.assert_called_once()