diff options
| -rw-r--r-- | lib/custodian.rb | 1 | ||||
| -rwxr-xr-x | lib/custodian/protocol-tests/https.rb | 210 |
2 files changed, 1 insertions, 210 deletions
diff --git a/lib/custodian.rb b/lib/custodian.rb index 79fca72..6e95f38 100644 --- a/lib/custodian.rb +++ b/lib/custodian.rb @@ -149,6 +149,7 @@ class Custodian # Given a test-type "foo" we'll attempt to instantiate a class called FOOTest. # test = hash['test_type'] + test = "http" if ( test == "https" ) clazz = test.upcase clazz = "#{clazz}Test" diff --git a/lib/custodian/protocol-tests/https.rb b/lib/custodian/protocol-tests/https.rb deleted file mode 100755 index 72dd42c..0000000 --- a/lib/custodian/protocol-tests/https.rb +++ /dev/null @@ -1,210 +0,0 @@ -#!/usr/bin/ruby1.8 - -require 'net/http' -require 'net/https' -require 'uri' - - - -class HTTPSTest - - # - # Data passed from the JSON hash. - # - attr_reader :test_data - - # - # The HTTP status, the HTTP response body, and the error text - # we return on failure. - # - attr_reader :status, :body, :error - - - - # - # Save the data away. - # - def initialize( data ) - @test_data = data - @error = nil - - # - # Ensure we have an URL - # - if ( @test_data["target_host"].nil? ) - @error = "Missing URL for the test." - raise ArgumentError, @error - end - - # - # Ensure we have a port - # - if ( @test_data["test_port"].nil? ) - @error = "Missing port for the test." - raise ArgumentError, @error - end - - end - - - # - # Run the test. - # - # Return "true" on success - # - # Return "false" on failure. - # - # If the test fails the details should be retrieved from "error()". - # - def run_test - - # - # Reset state from previous test. - # - @error = nil - - # - # Do the fetch, if this success then we'll have the - # @status + @text setup - # - if ( getURL(@test_data["target_host"], @test_data["timeout"].to_i ) ) - - # - # Do we need to test for a HTTP status code? - # - if ( @test_data["http_status"] ) - puts "Testing for HTTP status code: #{@test_data['http_status']}" if ( @test_data['verbose'] ) - - if ( @status != @test_data['http_status'].to_i) - @error = "#{@error} status code was #{@status} not #{@test_data['http_status']}" - end - end - - # - # Do we need to search for text in the body of the reply? - # - if ( @test_data['http_text'] ) - puts "Testing for text in the response: '#{@test_data['http_text']}'" if ( @test_data['verbose'] ) - - if (! @body.match(/#{@test_data['http_text']}/i) ) - @error = "#{@error} The response did not contain our expected text '#{test_data['http_text']}'" - end - end - - return true if ( @error.nil? ) - - return false - end - - return false - end - - - # - # Return the error text for why this test failed. - # - def error - return @error - end - - - # - # Retrieve a HTTP page from the web. - # - # NOTE: This came from sentinel. - def getURL (uri_str, timeout) - - timeout( timeout ) do - begin - uri_str = 'http://' + uri_str unless uri_str.match(/^http/) - url = URI.parse(uri_str) - http = Net::HTTP.new(url.host, url.port) - http.open_timeout = timeout - http.read_timeout = timeout - - if (url.scheme == "https") - http.use_ssl = true - http.verify_mode = OpenSSL::SSL::VERIFY_NONE - end - - response = nil - - # - # Ensure we have a trailing "/" - # - if ( url.path.empty? ) - url.path = "/" - end - - if nil == url.query - response = http.start { http.get(url.path) } - else - response = http.start { http.get("#{url.path}?#{url.query}") } - end - - case response - when Net::HTTPRedirection - then - newURL = response['location'].match(/^http/)? - response['Location']:uri_str+response['Location'] - return( getURL(newURL, timeout) ) - else - @status = response.code.to_i - @body = response.body - end - - return true - rescue Errno::EHOSTUNREACH => ex - @error = "no route to host" - return false - rescue Timeout::Error => ex - @error = "time out reached" - return false - rescue Errno::ECONNREFUSED => ex - @error = "Connection refused" - return false - rescue Timeout::Error => e - @error = "TIMEOUT: #{e}" - return false - rescue => ex - @error = ex - return false - end - end - end - -end - - - -# -# Sample test, for testing. -# -if __FILE__ == $0 then - - # - # Sample data. - # - test = { - "target_host" => "http://collector2.sh.bytemark.co.uk/", - "test_type" => "http", - "verbose" => 1, - "timeout" => 10, - "test_port" => 80, - "test_alert" => "Collector is unavailable", - "http_status" => "200" - } - - - # - # Run the test. - # - http = HTTPSTest.new( test ) - if ( http.run_test ) - puts "TEST OK" - else - puts "TEST FAILED" - puts http.error() - end - -end |