Add cookie argument to check_haproxy. Fixes #15

Author: benprew

Diff for: check_haproxy.rb

@@ -60,6 +60,10 @@
     options.http_error_critical = true
   end
 
+  opts.on('--cookie [COOKIE]', 'Login/Session cookie') do |v|
+    options.cookie = v
+  end
+
   opts.on( '-h', '--help', 'Display this screen' ) do
     puts opts
     exit 3
@@ -96,14 +100,15 @@ def open_options(options)
   open_opts = {
     :http_basic_authentication => [options.user, options.password]
   }
+  open_opts['Cookie'] = options.cookie if options.cookie
 
   # allows https with invalid certificate on ruby 1.9
   # src: http://snippets.aktagon.com/snippets/370-hack-for-using-openuri-with-ssl
   if options.insecure_ssl && RUBY_VERSION =~ /1\.9/
     open_opts[:ssl_verify_mode] = OpenSSL::SSL::VERIFY_NONE
   end
 
-  open_opts
+  return open_opts
 end
 
 def haproxy_response(options)
@@ -119,7 +124,7 @@ def haproxy_response(options)
   end
 
   begin
-    open(options.url, open_options(options))
+    return URI.open(options.url, open_options(options))
   rescue OpenURI::HTTPError => e
     puts "ERROR: #{e.message}"
     options.http_error_critical ? exit(CRITICAL) : exit(UNKNOWN)

Diff for: test/test_check_haproxy.sh

@@ -1,5 +1,7 @@
 #!/bin/bash
 
+set -e
+
 echo -n "testing nothing smokes..."
 for f in test/haproxy/*; do
     ./check_haproxy.rb -u "$f" |head -n1
@@ -23,3 +25,5 @@ echo "OK"
 echo -n "testing crit limit..."
 ./check_haproxy.rb -u "test/haproxy/fedoraproject_org.csv;" -w 1 -c2 -p fedmsg-raw-zmq-outbound-backend |grep 'CRIT.*too many sessions' > /dev/null
 echo "OK"
+
+echo "SUCCESS! ALL TESTS PASSED"