def test_private_execute_command(self): cmixer = Commands(mock.MagicMock(), mock.MagicMock(), mock.MagicMock(), mock.MagicMock(), mock.MagicMock()) cmixer._logger = MockLogger() cmixer.get_time = time.time mocked_dut = mock.MagicMock() mocked_dut.execute_command = mock.MagicMock() response = MockResponse(["line1"], retcode=0) mocked_dut.execute_command.return_value = response cmixer._parse_response = mock.MagicMock(return_value={"test": "parsed"}) self.assertEqual(cmixer._execute_command(mocked_dut, "test_command"), response)
def test_execute_command_execute_failures(self): cmixer = Commands(mock.MagicMock(), mock.MagicMock(), mock.MagicMock(), mock.MagicMock(), mock.MagicMock()) cmixer._logger = MockLogger() cmixer.get_time = time.time mocked_dut = mock.MagicMock() mocked_dut.execute_command = mock.MagicMock() mocked_dut.execute_command.side_effect = [TestStepFail, TestStepError, TestStepTimeout] with self.assertRaises(TestStepFail): cmixer._execute_command(mocked_dut, "test_command") with self.assertRaises(TestStepError): cmixer._execute_command(mocked_dut, "test_command") with self.assertRaises(TestStepTimeout): cmixer._execute_command(mocked_dut, "test_command")