def test_execute_cleanup(self, _): with testutils.create_tempfile() as tmp: cleanup = mock.Mock() command = execcmd.BaseCommand(tmp, cleanup=cleanup) command.execute() cleanup.assert_called_once_with()
def test_args(self, _): class FakeCommand(execcmd.BaseCommand): command = mock.sentinel.command with testutils.create_tempfile() as tmp: fake_command = FakeCommand(tmp) self.assertEqual([mock.sentinel.command, tmp], fake_command.args) fake_command = execcmd.BaseCommand(tmp) self.assertEqual([tmp], fake_command.args)
def test_execute_normal_command(self, mock_get_os_utils): mock_osutils = mock_get_os_utils() with testutils.create_tempfile() as tmp: command = execcmd.BaseCommand(tmp) command.execute() mock_osutils.execute_process.assert_called_once_with( [command._target_path], shell=command.shell) # test __call__ API. mock_osutils.execute_process.reset_mock() command() mock_osutils.execute_process.assert_called_once_with( [command._target_path], shell=command.shell)