def test_reset_duts(self):
bench = Bench()
bench.logger = mock.MagicMock()
bench.args = mock.MagicMock()
bench._resources._args = bench.args
type(bench.args).my_duts = mock.PropertyMock(return_value=False)
type(bench.args).pause_when_external_dut = mock.MagicMock(
return_value=False)
bench._resources.init(mock.MagicMock())
mock_dut = mock.MagicMock()
dutconf = {"reset.return_value": True, "initCLI.return_value": True}
mock_dut.configure_mock(**dutconf)
mock_dutrange = range(2)
mock_duts = [mock_dut, mock_dut]
bench.duts = mock_duts
# TODO: This mocking does not work somehow
with mock.patch.object(bench.resource_configuration,
"get_dut_range",
return_value=mock_dutrange):
with mock.patch.object(bench, "is_my_dut_index",
return_value=True):
for method in ["hard", "soft", None]:
bench.args.reset = method
bench.reset_dut()
mock_dut.reset.assert_called_with(method)
self.assertEqual(mock_dut.reset.call_count, 2)
mock_dut.reset.reset_mock()
def test_reset_duts(self):
bench = Bench()
mock_dut = mock.MagicMock()
dutconf = {"reset.return_value": True, "initCLI.return_value": True}
mock_dut.configure_mock(**dutconf)
mock_dutrange = range(2)
mock_duts = [mock_dut, mock_dut]
bench.duts = mock_duts
with mock.patch.object(bench,
"get_dut_range",
return_value=mock_dutrange):
with mock.patch.object(bench, "is_my_dut", return_value=True):
for method in ["hard", "soft", None]:
bench.args.reset = method
bench.reset_dut()
mock_dut.reset.assert_called_with(method)
self.assertEquals(mock_dut.reset.call_count, 2)
mock_dut.reset.reset_mock()
