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()