Merge 4c27dc4 into 3e700c0

ninoseki · Oct 3, 2020 · 7f04d66 · 7f04d66
2 parents 3e700c0 + 4c27dc4
commit 7f04d66
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/rogue_one/cli.rb b/lib/rogue_one/cli.rb
@@ -5,6 +5,12 @@
 
 module RogueOne
   class CLI < Thor
+    class << self
+      def exit_on_failure?
+        true
+      end
+    end
+
     desc "report [DNS_SERVER]", "Show a report of a given DNS server"
     method_option :default_list, type: :string, default: "alexa", desc: "A default list of top 100 domains (Alexa or Fortune)"
     method_option :custom_list, type: :string, desc: "A path to a custom list of domains"

diff --git a/spec/cli_spec.rb b/spec/cli_spec.rb
@@ -40,4 +40,10 @@ def capture(stream)
       end
     end
   end
+
+  describe ".exit_on_failure?" do
+    it do
+      expect(described_class.exit_on_failure?).to eq(true)
+    end
+  end
 end