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