fog--fog/tests/rackspace/helper.rb

LINKS_FORMAT = [{
  'href' => String,
  'rel' => String
}]

module Shindo
  class Tests

    unless Fog.mocking?
      Fog.timeout = 2000
      Fog::Logger.warning "Setting default fog timeout to #{Fog.timeout} seconds"
    end

    def given_a_load_balancer_service(&block)
      @service = Fog::Rackspace::LoadBalancers.new
      instance_eval(&block)
    end
    
    def given_a_load_balancer(&block)
        @lb = @service.load_balancers.create({
            :name => ('fog' + Time.now.to_i.to_s),
            :protocol => 'HTTP',
            :port => 80,
            :virtual_ips => [{ :type => 'PUBLIC'}],
            :nodes => [{ :address => '1.1.1.1', :port => 80, :condition => 'ENABLED'}]
          })
        @lb.wait_for { ready? }
      begin
        instance_eval(&block)
      ensure
        @lb.wait_for { ready? }
        @lb.destroy
      end
    end
  
   def wait_for_request(description = "waiting", &block)
     return if Fog.mocking?
     tests(description) do
       Fog.wait_for &block
     end
   end

   def wait_for_server_deletion(server)
     return if Fog.mocking?
     begin
       @instance.wait_for { state = 'DELETED' }
     rescue Fog::Compute::RackspaceV2::NotFound => e
       # do nothing
     end
   end

    def wait_for_server_state(service, server_id, state, error_states=nil)
      current_state = nil
      until current_state == state
        current_state = service.get_server(server_id).body['server']['status']
        if error_states
          error_states = Array(error_states)           
          raise "ERROR! Server should have transitioned to '#{state}' not '#{current_state}'" if error_states.include?(current_state)
        end
        sleep 10 unless Fog.mocking?
      end
      sleep 30 unless Fog.mocking?
    end

    def rackspace_test_image_id(service) 
      # I chose to use the first Ubuntu because it will work with the smallest flavor and it doesn't require a license
      @image_id ||= Fog.credentials[:rackspace_image_id] || service.images.find {|img| img.name =~ /Ubuntu/ }.id
    end

    def rackspace_test_flavor_id(service)
      @flavor_id ||= Fog.credentials[:rackspace_flavor_id] || service.flavors.first.id
    end

    # After a server has been successfully deleted they are still being reported as attached to a cloud network
    # causing delete calls to fail. This method attempts to address that.
    def delete_test_network(network)
      return false if Fog.mocking? || network.nil?
      attempt = 0
      begin
        network.destroy
      rescue Fog::Compute::RackspaceV2::ServiceError => e
        if attempt == 3
           Fog::Logger.warning "Unable to delete #{network.label}"
          return false
        end
         Fog::Logger.warning "Network #{network.label} Delete Fail Attempt #{attempt}- #{e.inspect}"
        attempt += 1
        sleep 60
        retry
      end
      return true
    end
  end
end