Fixed unit tests

1 files changed, 7 insertions, 0 deletions

diff --git a/cmd2/cmd2.py b/cmd2/cmd2.py
index 5aeb4fdf..72784d4c 100644
--- a/cmd2/cmd2.py
+++ b/cmd2/cmd2.py
@@ -2109,10 +2109,13 @@ class Cmd(cmd.Cmd):
         :param saved_state: contains information needed to restore state data
         """
         if saved_state.redirecting:
+            delete_name = None
             # If we redirected output to the clipboard
             if statement.output and not statement.output_to:
                 self.stdout.seek(0)
                 write_to_paste_buffer(self.stdout.read())
+                if statement.output == constants.REDIRECTION_APPEND:
+                    delete_name = self.stdout.name
 
             try:
                 # Close the file or pipe that stdout was redirected to
@@ -2120,6 +2123,10 @@ class Cmd(cmd.Cmd):
             except BrokenPipeError:
                 pass
 
+            # If we appended to the clipboard, make sure to delete the tempfile
+            if delete_name is not None:
+                os.unlink(delete_name)
+
             # Restore the stdout values
             self.stdout = saved_state.saved_self_stdout
             sys.stdout = saved_state.saved_sys_stdout