Set provided test input via a function

Author: joeyates

spec/support/highline_test_helpers.rb

@@ -4,9 +4,13 @@
 
 module HighLineTestHelpers
   def prepare_highline
-    @input = instance_double(IO, eof?: false, gets: "q\n")
+    @input = instance_double(IO, eof?: false, gets: nil)
     @output = StringIO.new
     Imap::Backup::Setup.highline = HighLine.new(@input, @output)
     [@input, @output]
   end
+
+  def set_highline_input(*lines)
+    allow(@input).to receive(:gets).and_return(*lines.map { |line| "#{line}\n" })
+  end
 end

spec/unit/setup/folder_chooser_spec.rb

@@ -27,6 +27,7 @@ module Imap::Backup
         Translator.new.setup
         allow(Kernel).to receive(:system)
         allow(Logger.logger).to receive(:warn)
+        set_highline_input("q\n")
       end
 
       describe "display" do
@@ -51,7 +52,11 @@ module Imap::Backup
         end
 
         describe "display" do
-          before { subject.run }
+          before do
+            set_highline_input("q\n")
+
+            subject.run
+          end
 
           it "shows folders which are being backed up" do
             expect(output.string).to include("+ my_folder")
@@ -64,7 +69,7 @@ module Imap::Backup
 
         context "when adding folders" do
           before do
-            allow(input).to receive(:gets).and_return("2\n", "q\n")
+            set_highline_input("2\n", "q\n")
 
             subject.run
           end
@@ -77,7 +82,7 @@ module Imap::Backup
 
         context "when removing folders" do
           before do
-            allow(input).to receive(:gets).and_return("1\n", "q\n")
+            set_highline_input("1\n", "q\n")
 
             subject.run
           end

spec/unit/setup/global_options/download_strategy_chooser_spec.rb

@@ -22,6 +22,7 @@ module Imap::Backup
       Translator.new.setup
       allow(Kernel).to receive(:system)
       allow(Kernel).to receive(:puts)
+      set_highline_input("q\n")
     end
 
     it "clears the screen" do
@@ -37,15 +38,15 @@ module Imap::Backup
     end
 
     it "accepts choices" do
-      allow(input).to receive(:gets).and_return("2\n", "q\n")
+      set_highline_input("2\n", "q\n")
 
       subject.run
 
       expect(config).to have_received(:download_strategy=).with("delay_metadata")
     end
 
     it "shows help" do
-      allow(input).to receive(:gets).and_return("help\n", "q\n")
+      set_highline_input("help\n", "q\n")
 
       subject.run
 
@@ -62,7 +63,7 @@ module Imap::Backup
       end
 
       it "marks the entry as current" do
-        allow(input).to receive(:gets) { "q\n" }
+        set_highline_input("q\n")
 
         subject.run
 

spec/unit/setup/global_options_spec.rb

@@ -27,6 +27,7 @@ module Imap::Backup
       allow(Kernel).to receive(:system)
       allow(Setup::GlobalOptions::DownloadStrategyChooser).
         to receive(:new) { download_strategy_chooser }
+      set_highline_input("q\n")
     end
 
     it "clears the screen" do
@@ -58,7 +59,7 @@ module Imap::Backup
     end
 
     it "accepts choices" do
-      allow(input).to receive(:gets).and_return("1\n", "q\n")
+      set_highline_input("1\n", "q\n")
 
       subject.run
 

spec/unit/setup_spec.rb

@@ -53,9 +53,8 @@ module Imap::Backup
       before do
         Translator.new.setup
         allow(Logger).to receive(:setup_logging)
-        allow(input).to receive(:eof?) { false }
-        allow(input).to receive(:gets) { "q\n" }
         allow(Kernel).to receive(:system)
+        set_highline_input("q\n")
       end
 
       describe "main menu" do
@@ -73,7 +72,7 @@ module Imap::Backup
           let(:config_modified) { true }
 
           before do
-            allow(input).to receive(:gets) { "exit\n" }
+            set_highline_input("exit\n")
             subject.run
           end
 
@@ -129,7 +128,7 @@ module Imap::Backup
         let(:global_options) { instance_double(Setup::GlobalOptions, run: nil) }
 
         before do
-          allow(input).to receive(:gets).and_return("3\n", "q\n")
+          set_highline_input("3\n", "q\n")
           allow(Setup::GlobalOptions).
             to receive(:new).with(config: config) { global_options }
         end
@@ -148,7 +147,7 @@ module Imap::Backup
         let(:config_modified) { true }
 
         before do
-          allow(input).to receive(:gets).and_return("1\n", "exit\n")
+          set_highline_input("1\n", "exit\n")
           allow(Setup::Asker).to receive(:email).
             with(no_args) { "new@example.com" }
           allow(Setup::Account).to receive(:new).
@@ -179,8 +178,8 @@ module Imap::Backup
         end
 
         before do
-          allow(input).to receive(:gets).and_return("add\n", "exit\n")
           allow(config.accounts).to receive(:<<)
+          set_highline_input("add\n", "exit\n")
           allow(Setup::Asker).to receive(:email).
             with(no_args) { added_email }
           allow(Setup::Account).to receive(:new).
@@ -244,7 +243,7 @@ module Imap::Backup
         let(:config_modified) { true }
 
         before do
-          allow(input).to receive(:gets) { "save\n" }
+          set_highline_input("save\n")
         end
 
         it "exits" do
@@ -263,7 +262,7 @@ module Imap::Backup
         let(:config_modified) { true }
 
         before do
-          allow(input).to receive(:gets) { "exit\n" }
+          set_highline_input("exit\n")
         end
 
         it "exits" do