Merge pull request #82 from benbalter/color

Allow disabling colorized output via CLI

bin/gman

@@ -14,9 +14,10 @@ class IsoCountryCodes
 end
 
 domain = ARGV[0]
+String.disable_colorization = true if ARGV[1] == "--no-color"
 
 if domain.nil?
-  puts "USAGE: gman <domain or email address>".red
+  puts "USAGE: gman <domain or email address> [--no-color]".red
   exit 1
 end
 

test/test_gman_bin.rb

@@ -53,4 +53,13 @@ def setup
     assert_match /DC/, @output
     assert_equal 0, @status.exitstatus
   end
+
+  should "allow you to disable colorization" do
+    output, status = test_bin("whitehouse.gov", "--no-color")
+    refute_match /\[0;32;49m/, output
+  end
+
+  should "color by default" do
+    assert_match /\[0;32;49m/, @output
+  end
 end