chore(spec): update to handle pre-run program exits

Author: devnote-dev

spec/command_spec.cr

@@ -17,6 +17,11 @@ private class TestArgsCommand < Cling::Command
     arguments.get "first"
     arguments.get "second"
   end
+
+  def on_error(ex : Exception)
+    # override default behaviour so that it works in specs
+    raise ex
+  end
 end
 
 private class TestOptionsCommand < Cling::Command
@@ -99,7 +104,9 @@ errors_command = TestErrorsCommand.new
 
 describe Cling::Command do
   it "executes the pre_run only" do
-    arguments_command.execute %w(--skip)
+    expect_raises Cling::ExitProgram do
+      arguments_command.execute %w(--skip)
+    end
   end
 
   it "fails on missing arguments" do

spec/helper_spec.cr

@@ -33,6 +33,11 @@ private class CrystalCommand < Cling::MainCommand
 
   def run(arguments : Cling::Arguments, options : Cling::Options) : Nil
   end
+
+  def on_error(ex : Exception)
+    # override default behaviour so that it works in specs
+    raise ex
+  end
 end
 
 command = CrystalCommand.new
@@ -43,7 +48,7 @@ describe Cling::MainCommand do
   it "prints the help message" do
     io = IO::Memory.new
     command.stdout = io
-    command.execute ""
+    command.execute "" rescue nil
 
     io.to_s.chomp.should eq <<-HELP
     Runs some Crystal commands

spec/main_spec.cr

@@ -25,6 +25,11 @@ private class GreetCommand < Cling::Command
       stdout.puts message
     end
   end
+
+  def on_error(ex : Exception)
+    # override default behaviour so that it works in specs
+    raise ex
+  end
 end
 
 command = GreetCommand.new
@@ -33,7 +38,7 @@ describe Cling do
   it "tests the help command" do
     io = IO::Memory.new
     command.stdout = io
-    command.execute ""
+    command.execute "" rescue nil
 
     io.to_s.should eq <<-HELP
     Greets a person
@@ -53,7 +58,7 @@ describe Cling do
   it "tests the main command" do
     io = IO::Memory.new
     command.stdout = io
-    command.execute %w(Dev)
+    command.execute %w(Dev) rescue nil
 
     io.to_s.should eq "Hello, Dev!\n"
   end