refactor: DRY up command output retrieval logic in run_spec.rb

Author: kaorukobo

spec/unit/run_spec.rb

@@ -155,28 +155,22 @@
 
   context "with uuid option" do
     it "runs command successfully with logging without uuid set globally" do
-      output = StringIO.new
-      command = TTY::Command.new(output: output, uuid: false)
+      lines = retrieve_log_lines { |output|
+        command = TTY::Command.new(output: output, uuid: false)
+        command.run(:echo, "hello")
+      }
 
-      command.run(:echo, "hello")
-      output.rewind
-
-      lines = output.readlines
-      lines.last.gsub!(/\d+\.\d+/, "x")
       expect(lines).to eq(
         generic_colored_log_lines(prefix: nil)
       )
     end
 
     it "runs command successfully with logging without uuid set locally" do
-      output = StringIO.new
-      command = TTY::Command.new(output: output)
-
-      command.run(:echo, "hello", uuid: false)
-      output.rewind
+      lines = retrieve_log_lines { |output|
+        command = TTY::Command.new(output: output)
+        command.run(:echo, "hello", uuid: false)
+      }
 
-      lines = output.readlines
-      lines.last.gsub!(/\d+\.\d+/, "x")
       expect(lines).to eq(
         generic_colored_log_lines(prefix: nil)
       )
@@ -185,51 +179,56 @@
 
   context "with tag option" do
     it "prints the tag set globally" do
-      output = StringIO.new
       tag = "task"
-      command = TTY::Command.new(output: output, tag: tag)
 
-      command.run(:echo, "hello")
+      lines = retrieve_log_lines { |output|
+        command = TTY::Command.new(output: output, tag: tag)
+        command.run(:echo, "hello")
+      }
 
-      output.rewind
-      lines = output.readlines
-      lines.last.gsub!(/\d+\.\d+/, "x")
       expect(lines).to eq(
         generic_colored_log_lines(prefix: tag)
       )
     end
 
     it "prints the tag set locally" do
-      output = StringIO.new
       tag = "task"
-      command = TTY::Command.new(output: output)
 
-      command.run(:echo, "hello", tag: tag)
+      lines = retrieve_log_lines { |output|
+        command = TTY::Command.new(output: output)
+        command.run(:echo, "hello", tag: tag)
+      }
 
-      output.rewind
-      lines = output.readlines
-      lines.last.gsub!(/\d+\.\d+/, "x")
       expect(lines).to eq(
         generic_colored_log_lines(prefix: tag)
       )
     end
 
     it "prints the tag even if uuid is set to false" do
-      output = StringIO.new
       tag = "task"
-      command = TTY::Command.new(output: output, tag: tag, uuid: false)
 
-      command.run(:echo, "hello")
+      lines = retrieve_log_lines { |output|
+        command = TTY::Command.new(output: output, tag: tag, uuid: false)
+        command.run(:echo, "hello")
+      }
 
-      output.rewind
-      lines = output.readlines
-      lines.last.gsub!(/\d+\.\d+/, "x")
       expect(lines).to eq(
         generic_colored_log_lines(prefix: tag)
       )
     end
   end
 
+  # Retrieves log lines from the output produced within the given block.
+  # Also replaces the execution time portion in the output with `x`.
+  def retrieve_log_lines
+    output = StringIO.new
+    yield(output)
+    output.rewind
+    lines = output.readlines
+    lines.last.gsub!(/\d+\.\d+/, "x")
+    lines
+  end
+
   # Generates the expected log lines in colored mode, with/without `[prefix]`
   def generic_colored_log_lines(prefix: nil)
     if prefix