health_check 1.1.2 → 1.2.0

data/test/test_with_railsapp

@@ -3,56 +3,35 @@
 # Any failure causes exit
 set -e
 
-rehash=''
-if [[ -s "$HOME/.rvm/scripts/rvm" ]] ; then
-  echo "Detected user installed rvm"
-elif [[ -s "/usr/local/rvm/scripts/rvm" ]] ; then
-  echo "Detected root installed rvm"
-elif [[ -d "$HOME/.rbenv" ]] ; then
-  echo "Detected rbenv ..."
-  rehash='rbenv rehash'
-else
-  printf "Note: Neither rvm nor rbenv was not found.\n"
-fi
-
-echo "Checking required commands exist:"
-for cmd in bash gem egrep ls tail kill find cpio ruby gem rake bundle
-do
-    echo -n "  "
-    which $cmd || ( echo "Aborting test: Missing $cmd command!" && exit 2 )
-done
-if ! bundle check ; then
-    echo "Test aborted (missing required gems)"
-    exit 2
-fi
+case "$1" in
+redo)
+    . test/init_variables
+    cd $railsapp
+    echo Dropping database ...
+    $rake db:drop 
+    echo Removing migrations ...
+    rm -f db/migrate/* db/schema.rb
+    case `ruby -e 'puts JRUBY_VERSION' 2> /dev/null` in
+	[0-9]*)
+        echo 'Jruby requires the database to be created before the server is started: running rake db:migrate'
+        $rake db:migrate 
+	;;
+    esac
+    actual_rails_version=`$rails -v`
+    ;;
+*)
+    . test/setup_railsapp
+    ;;
+esac
 
-base_dir=`pwd`
+cd $railsapp
+date > $custom_file
 
 testurl="$base_dir/test/testurl"
 fake_smtp_server="$base_dir/test/fake_smtp_server"
 
-test_dir=`pwd`/tmp
-rm -rf tmp/Gemfile* tmp/railsapp tmp/bin tmp/gems
-mkdir -p tmp/gems
-
-[ -d lib/health_check ] || exec echo test MUST be executed in the base of the health_check gem/clone of git repository
-
-echo Installing health_check as a gem into tmp/gems
-env GEM_HOME=$test_dir/gems rake install
-
-export GEM_PATH="$test_dir/gems:`gem environment gempath`"
-echo Set GEM_PATH=$GEM_PATH
-
-echo Gems in tmp/gems:
-ls tmp/gems
-
-echo GEM LIST
-gem list
-
-echo Environment:
-env | egrep 'TRAVIS|RAILS|RUBY|_ENV|GEM|BUNDLE'
-
 server_pid=''
+fake_smtp_pid=''
 
 finish()
 {
@@ -60,24 +39,31 @@ finish()
     echo ========================================================
     echo TEST ${1:-FAILED}
     echo ========================================================
-    echo Result of: ls -lR $test_dir/railsapp/log $test_dir/railsapp/db
-    ls -lR $test_dir/railsapp/log $test_dir/railsapp/db
+    echo Result of: ls -lR $railsapp/log $railsapp/db
+    ls -lR $railsapp/log $railsapp/db
 
-    if [ -s $test_dir/railsapp/log/test.log ]
+    if [ -s $railsapp/log/test.log ]
     then
         echo ========================================================
-        echo Last 300 lines of test log
-        tail -300 $test_dir/railsapp/log/test.log
+        echo Last 50 lines of test log
+        tail -50 $railsapp/log/test.log
     fi
 
+    case "$fake_smtp_pid" in
+        [0-9]*)
+            echo ========================================================
+            echo "Killing fake smtp server [pid: $fake_smtp_pid]"
+            kill -9 $fake_smtp_pid
+            ;;
+    esac
     case "$server_pid" in
         [0-9]*)
             echo ========================================================
             echo "Killing rails server [pid: $server_pid]"
             kill -9 $server_pid 
-            wait
             ;;
     esac
+    wait
     trap "" 0
     echo ========================================================
     echo TEST ${1:-FAILED}
@@ -87,126 +73,24 @@ finish()
 
 trap "finish FAILED 1" 0
 
-rails='rails'
-if [ -x bin/rails ]
-then
-    rails="`pwd`/bin/rails"
-    echo "Using binstub: $rails for rails command"
-fi
-rake='rake'
-if [ -x bin/rake ]
-then
-    rake="`pwd`/bin/rake"
-    echo "Using binstub: $rake for rake command"
-fi
-
-cd $test_dir
-
-actual_rails_version=`$rails -v`
-
-case `ruby -e 'puts JRUBY_VERSION' 2> /dev/null` in
-    [0-9]*)
-        db=jdbcsqlite3
-        # Appears to need a bit extra time
-        ;;
-    *)
-        db=sqlite3
-        ;;
-esac
-
-echo "Creating $actual_rails_version app in $test_dir/railsapp using adapter $db"
-case "$actual_rails_version" in
-    *' '[12].*)
-        $rails railsapp -d $db
-        ;;
-    *' '[34].*)
-        $rails new railsapp -d $db
-        ;;
-    *)
-        echo "Unknown rails version"
-        ;;
-esac
-
-cd railsapp
-
-echo "Changed current directory to railsapp root: `pwd`"
-
-echo "Fixing rdoc require in Rakefile if needed"
-ruby -p -i.bak -e '$_.gsub!(/rake.rdoctask/, "rdoc/task")' Rakefile
-
-echo "Configuring mailer to point to fake_smtp_server port 3555"
-cat >> config/environment.rb <<'!EOF!'
-
-ActionMailer::Base.delivery_method = :smtp
-ActionMailer::Base.smtp_settings = { :address => "localhost", :port => 3555 }
-
-!EOF!
-
-echo "Last ten lines of config/environment.rb:"
-tail -10 config/environment.rb
-echo 
-
-if [ -s Gemfile ]
-then
-    echo Adding health_check as gem to Gemfile...
-    echo "gem 'health_check', :path => '$base_dir'" >> Gemfile
-
-    case "$BUNDLE_GEMFILE" in
-        ?*)
-            echo Unsetting BUNDLE_GEMFILE '(so Gemfile for rails application will be used)'
-            unset BUNDLE_GEMFILE
-            ;;
-    esac
-    echo
-    echo Gemfile contents:
-    cat Gemfile
-    echo
-    echo running bundle install --binstubs
-    bundle install --binstubs
-    echo "Using binstubs in `pwd`/bin for rails and rake commands"
-    rails="`pwd`/bin/rails"
-    rake="`pwd`/bin/rake"
-    # Fix for rvm, otherwise bundle run from rails create fails
-    export PATH="`pwd`/bin:$PATH"
-else
-    dest=`pwd`/vendor/plugins/health_check
-    echo Installing health_check as plugin in $dest ...
-    mkdir -p $dest
-    (
-	cd $base_dir
-
-        find . | egrep -v '/tmp|/\.git|\.gem$' | cpio -pdl $dest
-        
-	cd $dest
-
-        echo Files installed as plugin:
-        find . -type f
-        echo
-    )
-fi
-
-echo Setting RAILS_ENV=test RACK_ENV=test
-export RAILS_ENV=test RACK_ENV=test
-
-case $db in
-    jdbcsqlite3)
-        echo
-        echo 'Jruby requires the dastabase to be created before the server is started: running rake db:migrate'
-        $rake db:migrate 
-        echo
-        ;;
-esac
-
 port=3456
-echo Starting server on port $port ...
 
 if [ -x script/server ]
 then
-    bundle exec ./script/server webrick -p $port &
+    if [ -f Gemfile ]
+    then
+	echo Starting server on port $port using bundle exec ./script/server ...
+	bundle exec ./script/server webrick -p $port &
+    else
+	echo Starting server on port $port using ./script/server ...
+	./script/server webrick -p $port &
+    fi
 else
+    echo Starting server on port $port using $rails s ...
     $rails s webrick -p $port &
 fi
 server_pid=$!
+sleep 17
 
 echo STATIC-FILE > public/static.txt
 
@@ -215,80 +99,119 @@ host=http://localhost:${port}
 # get a static file
 echo
 echo 'TESTING can get a static file ...'
-$testurl ${host}/static.txt STATIC-FILE
+$testurl ${host}/static.txt 200 text/plain STATIC-FILE
 echo
 
 echo 'TESTING health_check/migration should pass with no database migrations ...'
 mkdir -p db/migrate
 ls db/migrate
-$testurl ${host}/health_check/migration success
+$testurl ${host}/health_check/migration 200 text/plain $success
 echo
 
 echo 'TESTING health_check/migration should fail without initial database migration ...'
 cp $base_dir/test/migrate/nine/* db/migrate
 ls db/migrate
-$testurl ${host}/health_check/migration failed
+$testurl ${host}/health_check/migration 550 text/plain failed
 echo
 
 echo 'TESTING health_check/database should pass without initial database migration (since it ignores the difference) ...'
-$testurl ${host}/health_check/database success
+$testurl ${host}/health_check/database 200 text/plain $success
 echo
 
 echo 'TESTING health_check/site should pass ...'
-$testurl ${host}/health_check/site success
+$testurl ${host}/health_check/site 200 text/plain $success
 echo
 
 echo 'TESTING health_check/migration should pass after initial database migration ...'
 $rake db:migrate 
-$testurl ${host}/health_check/migration success
+$testurl ${host}/health_check/migration 200 text/plain $success
 echo
 
 echo 'TESTING health_check/migration should fail without all migrations ...'
 cp $base_dir/test/migrate/twelve/* db/migrate
 ls db/migrate
-$testurl ${host}/health_check/migration failed
+$testurl ${host}/health_check/migration 550 text/plain failed
 echo
 
 echo 'TESTING health_check/migration should pass after both database migrations ...'
 $rake db:migrate 
-$testurl ${host}/health_check/migration success
+$testurl ${host}/health_check/migration 200 text/plain $success
 echo
 
 echo 'TESTING health_check/migration should pass after both database migrations ...'
 $rake db:migrate 
-$testurl ${host}/health_check/migration success
+$testurl ${host}/health_check/migration 200 text/plain $success
 echo
 
 echo 'TESTING health_check/email should fail without smtp available ...'
-$testurl ${host}/health_check/email failed
+$testurl ${host}/health_check/email 550 text/plain failed
 echo
 
 echo 'TESTING health_check/email should pass with smtp available ...'
 $fake_smtp_server &
+fake_smtp_pid=$!
 sleep 5
-$testurl ${host}/health_check/email success
+$testurl ${host}/health_check/email 200 text/plain $success
 echo
 
 echo 'TESTING health_check (all) should fail without smtp available ...'
-$testurl ${host}/health_check failed
+$testurl ${host}/health_check 550 text/plain failed
 echo
 
 echo 'TESTING health_check/all should fail without smtp available ...'
-$testurl ${host}/health_check failed
+$testurl ${host}/health_check 550 text/plain failed
 echo
 
+kill -9 $fake_smtp_pid || echo fake_smtp_server had finished as expected
 echo 'TESTING health_check (all) should pass with smtp available ...'
 $fake_smtp_server &
+fake_smtp_pid=$!
 sleep 5
-$testurl ${host}/health_check success
+$testurl ${host}/health_check 200 text/plain $success
 echo
 
+kill -9 $fake_smtp_pid || echo fake_smtp_server had finished as expected
 echo 'TESTING health_check/all should pass with smtp available ...'
 $fake_smtp_server &
+fake_smtp_pid=$!
 sleep 5
-$testurl ${host}/health_check/all success
+$testurl ${host}/health_check/all 200 text/plain $success
+echo
+
+echo 'TESTING health_check/custom should pass ...'
+$testurl ${host}/health_check/custom 200 text/plain $success
+echo
+
+echo 'TESTING health_check/custom.json should pass ...'
+$testurl ${host}/health_check/custom.json 200 application/json '"healthy":true'
+$testurl ${host}/health_check/custom.json 200 application/json "\"message\":\"$success\""
+echo
+
+echo 'TESTING health_check/custom.xml should pass ...'
+$testurl ${host}/health_check/custom.xml 200 application/xml '<healthy type="boolean">true</healthy>'
+$testurl ${host}/health_check/custom.xml 200 application/xml "<message>$success</message>"
+echo
+
+echo 'TESTING health_check/custom should fail when custom returns string ...'
+rm -f $custom_file
+$testurl ${host}/health_check/custom 550 text/plain failed
+echo
+
+echo 'TESTING health_check (all) should fail when custom check fails ...'
+$testurl ${host}/health_check 550 text/plain "$custom_file is missing!"
+echo
+
+echo 'TESTING health_check.json (all) should fail when custom check fails ...'
+$testurl ${host}/health_check.json 555 application/json '"healthy":false'
+$testurl ${host}/health_check.json 555 application/json "$custom_file is missing!"
+echo
+
+echo 'TESTING health_check.xml (all) should fail when custom check fails ...'
+$testurl ${host}/health_check.xml 555 application/xml '<healthy type="boolean">false</healthy>'
 echo
 
+date > $custom_file
+
 finish PASSED 0
 exit 0
 

data/test/testurl

@@ -3,31 +3,58 @@ require 'net/http'
 require 'uri'
 
 def open(url)
-  Net::HTTP.get(URI.parse(url))
+  Net::HTTP.get_response(URI.parse(url))
 end
 
-page_content = open(ARGV[0]) rescue nil
-unless page_content
+response = open(ARGV[0]) rescue nil
+unless response
   i=0
   print "waiting.."
-  while i < 120 and not page_content
+  while i < 120 and not response
     print "."
     STDOUT.flush
     i += 1
     sleep(1)
-    page_content = open(ARGV[0]) rescue nil
+    response = open(ARGV[0]) rescue nil
   end
-  puts "\n  #{page_content ? 'got url content' : 'timed out'} after waiting #{i} seconds"
+  unless response
+    puts "\nFAIL: timed out after waiting #{i} seconds"
+    exit 9
+  end
+  puts "\n  got url content after waiting #{i} seconds"
 end
 
-puts page_content
+page_content = response.body
+
+puts "  response code: #{response.code} #{response.message}"
+puts "  content_type: #{response.content_type}"
+puts "  body: #{page_content}"
+
 if ARGV[1]
-  if page_content.to_s.include? ARGV[1]
-    puts "PASS (found #{ARGV[1]})"
-    exit 0
+  if response.code == ARGV[1]
+    puts "PASS (response code was #{ARGV[1]})"
+  else
+    puts "FAIL (response code was #{response.code}, expected #{ARGV[1]})" 
+    exit 1
+  end
+end
+
+if ARGV[2]
+  if response.content_type == ARGV[2]
+    puts "PASS (content type was #{response.content_type})"
   else
-    puts "FAIL (could not find #{ARGV[1]})" 
+    puts "FAIL (content type was #{response.content_type}, expected #{ARGV[2]})" 
     exit 2
   end
 end
-exit(page_content.nil? ? 1 : 0)
+
+if ARGV[3]
+  if page_content.to_s.include? ARGV[3]
+    puts "PASS (found #{ARGV[3]})"
+  else
+    puts "FAIL (could not find #{ARGV[3]})" 
+    exit 3
+  end
+end
+
+exit 0