http_spec.rb

# Copyright (c) 2009-2012 VMware, Inc.

require File.dirname(__FILE__) + '/../spec_helper'

require 'posix/spawn'
require 'yajl'

describe "http messages" do

  def http(method, args=nil)
    uri = URI.parse(@http_uri)
    req = Net::HTTP::Post.new(uri.request_uri)
    req.basic_auth(@user, @pass)

    hash = {:reply_to => 'rspec', :method => method}
    hash['arguments'] = args if args
    req.body = Yajl::Encoder.encode(hash)

    Net::HTTP.start(uri.host, uri.port) do |http|
      response = http.request(req)
      msg = Yajl::Parser.parse(response.body)
      if msg.has_key?('value')
        yield msg['value']
      else
        yield msg
      end
    end
  end

  def http_up?
    uri = URI.parse(@http_uri)
    req = Net::HTTP::Post.new(uri.request_uri)
    req.basic_auth(@user, @pass)

    Net::HTTP.start(uri.host, uri.port) do |http|
      response = http.request(req)
      return response.is_a?(Net::HTTPSuccess)
    end
  rescue Errno::ECONNREFUSED
    false
  end

  before(:all) do
    @user = "http"
    @pass = @user.reverse
    @port = get_free_port
    @http_uri = "http://#{@user}:#{@pass}@localhost:#{@port}/agent"
    @agent_id = "rspec_agent"

    puts "starting http agent"
    agent = File.expand_path("../../../bin/agent", __FILE__)
    @basedir = File.expand_path("../../../tmp", __FILE__)
    FileUtils.mkdir_p(@basedir) unless Dir.exist?(@basedir)
    command = "ruby #{agent} -n #{@http_uri} -a #{@agent_id} -h 1 -b #{@basedir} -l ERROR"
    @agent_pid = POSIX::Spawn::spawn(command)

    counter = 0
    while !http_up?
      counter += 1
      # wait max 10 seconds for the agent to start
      raise "unable to connect to agent" if counter > 100
      sleep 0.1
    end
  end

  it "should respond to state message" do
    http('state') do |msg|
      msg.should have_key('deployment')
      msg.should have_key('networks')
      msg.should have_key('resource_pool')
      msg.should have_key('agent_id')
      msg.should have_key('vm')
      msg.should have_key('job_state')
    end
  end

  it "should respond to ping message" do
    http('ping') do |msg|
      msg.should == 'pong'
    end
  end

  it "should respond noop message" do
    http('noop') do |msg|
      msg.should == 'nope'
    end
  end

  it "should respond to start message" do
    http('start') do |msg|
      msg.should == 'started'
    end
  end

  it "should respond to drain message" do
    http('drain', ['shutdown', 'bar']) do |msg|
      msg.should == 0
    end
  end

  it "should respond to stop message" do
    http('stop') do |msg|
      msg.should == 'stopped'
    end
  end

  it "should respond to apply message" do
    task = nil
    http('apply', []) do |msg|
      task = msg
      task['state'].should == "running"
      task['agent_task_id'].should_not be_nil
    end

    while task['state'] == "running"
      sleep 0.5
      http('get_task', [ task['agent_task_id'] ]) do |msg|
        task = msg
        unless task['state']
          msg.should have_key('exception')
          break
        end
      end
    end
  end

  after(:all) do
    if @agent_pid
      puts "stopping agent"
      Process.kill(:TERM, @agent_pid)
    else
      raise "unable to stop agent, you need to clean up by hand"
    end
    FileUtils.rm_rf(@basedir)
  end
end