integration

#!/usr/bin/env ruby
#
@tests = 0

# Use ./integration REMOTE to use remote gem sources.
#
@remote = ARGV.include? 'REMOTE'

def remote?
  @remote
end

def bundle_options
  @remote ? '' : '--local'
end

def gem_options
  @remote ? '--remote' : '--local'
end

def each_type
  yield :full, 20
  yield :live, 0
end

def ask_server query
  `curl -S 'localhost:8080/books?query=#{query}&ids=20'`
end
def server query, match
  @tests += 1
  if ask_server(query).chomp.include?(match)
    puts "Yay, server returned #{match} for #{query}!"
  else
    fail "#{query} result does not include #{match}"
  end
end
def client query, match, port = 3210
  @tests += 1
  if `curl -S 'localhost:#{port}/search/full?query=#{query}&ids=20'`.chomp.include?(match)
    puts "Yay, client returned #{match} for #{query}!"
  else
    fail "#{query} result does not include #{match}"
  end
end

def install_all_in_one
  puts "Installing the Sinatra/Unicorn client/server."
  system "picky generate all_in_one sandbox/server"
  system "cd sandbox/server; bundle install #{bundle_options}"
  system "cd sandbox/server; bundle exec rake index"
  system "cd sandbox/server; bundle exec unicorn -c unicorn.rb &"

  sleep 8 # loading
end

def install_sinatra_server
  puts "Installing the Sinatra/Unicorn server."
  system "picky generate sinatra_server sandbox/server"
  system "cd sandbox/server; bundle install #{bundle_options}"
  system "cd sandbox/server; bundle exec rake index"
  system "cd sandbox/server; bundle exec unicorn -c unicorn.rb &"

  sleep 8 # loading
end

def install_classic_server
  puts "Installing the Picky/Unicorn server."
  system "picky generate classic_server sandbox/server"
  system "cd sandbox/server; bundle install #{bundle_options}"
  system "cd sandbox/server; bundle exec rake index"
  system "cd sandbox/server; bundle exec rake start &"

  sleep 8 # loading
end

def run_search_requests_on_server
  puts
  puts "Running search requests on the server"
  server "alan", "author"
  server "ALAN", "title"
  server "pinchn~", "pynchon"
  server "sp", "title"
  server "sp", "author"
  server "title:lyterature~+2002", "literature"
  puts
  puts
end

def install_sinatra_client
  puts "Installing the client."
  system "picky generate sinatra_client sandbox/client"
  system "cd sandbox/client; bundle install #{bundle_options}"
  system "cd sandbox/client; unicorn -p 3210 &"

  sleep 5 # loading
end

def run_search_requests_on_client port = 3210
  puts
  puts "Running search requests on the client directly"
  client "alan", "author", port
  client "ALAN", "title", port
  client "pinchn~", "pynchon", port
  client "sp", "title", port
  client "sp", "author", port
  client "title:lyterature~+2002", "literature", port
  client "alan", "Alan Turing", port
  puts
  puts
end

def open_client_in_browser port = 3210
  system "open -a 'Google Chrome' http://localhost:#{port}/?q=alan"
  sleep 1
  system "open -a 'Google Chrome' http://localhost:#{port}/?q=a%2A%20b"
  sleep 1
  system "open -a 'Google Chrome' http://localhost:#{port}/configure"
  sleep 5
end

def run_server_integration_tests
  puts "Now the server integration specs"
  if system('cd sandbox/server; bundle exec rake spec')
    puts "Yay! Integration specs ran through!"
  else
    raise "Integration specs failed!"
  end
  sleep 1
end

def run_server_rake_task name
  unless system("cd sandbox/server; rake #{name} --trace")
    raise "Integration specs failed!"
  end
end

def run_server_rake_tasks
  puts "Now the server rake tasks"
  run_server_rake_task '-T'
  run_server_rake_task 'analyze'
  run_server_rake_task 'check'
  run_server_rake_task 'routes'
  run_server_rake_task 'todo'
  run_server_rake_task 'try[test]'
end

def run_server_speed_tests
  puts "Now for a speed test"
  queries = ['alan', 'pinchn~', 'sp', 'title:lyterature~+2002'].cycle
  100.times do
    ask_server queries.next
  end
end

def kill_server
  puts "Killing server (if there)"
  system "ps ax | grep 'unicorn -c unicorn.rb' | awk '{print $1}'| xargs kill"
end

def kill_client
  puts "Killing client (if there)"
  system "ps ax | grep 'unicorn -p 3210' | awk '{print $1}'| xargs kill"
end

def delete_server_directory
  system "rm -r sandbox/server"
end

def delete_client_directory
  system "rm -r sandbox/client"
end

# Runs a whole integration.
#
begin
  puts "Please run using Ruby 1.9" if RUBY_VERSION < '1.9.0'

  system "gem install bundler #{ gem_options }"
  system "gem install unicorn #{ gem_options }"
  system "gem install rspec #{ gem_options }"

  system "./install #{ 'REMOTE' if remote? }"

  # All in one.
  #
  install_all_in_one
  run_server_rake_tasks

  run_search_requests_on_client 8080 # Since it is just one server.
  open_client_in_browser 8080

  kill_server
  delete_server_directory

  # Sinatra Client.
  #
  install_sinatra_client

  # Sinatra Server.
  #
  install_sinatra_server
  run_search_requests_on_server
  run_server_integration_tests
  run_server_speed_tests
  run_server_rake_tasks

  run_search_requests_on_client
  open_client_in_browser

  kill_server
  delete_server_directory

  # Classic Server.
  #
  install_classic_server
  run_search_requests_on_server
  run_server_integration_tests
  run_server_speed_tests
  run_server_rake_tasks

  run_search_requests_on_client
  open_client_in_browser

  puts
  puts "SUCCESS! #{@tests} tests run without a hitch :)"
  puts
rescue Exception => e
  puts e.message
  puts e.backtrace
ensure
  puts "Press enter to kill (Break and have mercy with Ctrl-C)"
  gets

  kill_server
  kill_client

  delete_server_directory
  delete_client_directory
end