Class: TestbeatRestRequest

Inherits:
Object
  • Object
show all
Defined in:
lib/rspec/spec_helper.rb

Instance Method Summary collapse

Constructor Details

#initialize(node, testbeat) ⇒ TestbeatRestRequest

Returns a new instance of TestbeatRestRequest.



403
404
405
406
407
408
 
# File 'lib/rspec/spec_helper.rb', line 403

def initialize(node, testbeat)
  #@headers = {}
  @timeout = 10
  @node = node
  @testbeat = testbeat
end

Instance Method Details

#run ⇒ Object

Initiate the request and return Net::HTTPResponse object, supporting response [:responseHeaderName], .body (string), .code (int), .msg (string)



412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
 
# File 'lib/rspec/spec_helper.rb', line 412

def run
  reuse_id = @testbeat.context_block_id
  previous = $_testbeat_rest_reuse[reuse_id]
  if previous
    @response = previous
    @testbeat.logger.info{ "Request reused within #{reuse_id} responded #{@response.code} #{@response.message}" }
    return @response
  end
  # If there's no built in auth support in Net::HTTP we can check for 401 here and re-run the request with auth header
  Net::HTTP.start(@node.host,
    @testbeat.port,
    :use_ssl => !@testbeat.unencrypted?,
    :verify_mode => OpenSSL::SSL::VERIFY_NONE,  # Ideally verify should be enabled for non-labs hosts (anything with a FQDN including dots)
    :open_timeout => @timeout,
    :read_timeout => @timeout
    ) do |http|

    if not HTTP_VERBS.has_key?(@testbeat.method)
      raise "Testbeat can't find HTTP verb #{@testbeat.method}"
    end

    req = HTTP_VERBS[@testbeat.method].new(@testbeat.resource)
    if @testbeat.headers?
      @testbeat.headers.each {|name, value| req[name] = value }
    end
    if @testbeat.form?
      if not req.methods.include? :set_form_data
        raise "Testbeat can't set form data for HTTP verb #{@testbeat.method}"
      end
      req.set_form_data(@testbeat.form)
    end
    if @testbeat.body?
      if not req.request_body_permitted?
        raise "Testbeat can't set body for HTTP verb #{@testbeat.method}"
      end
      req.body = @testbeat.body
    end

    @response = http.request(req) # Net::HTTPResponse object

    # The redirect must not be authenticated. Consider copying 401 handling to above redirect handling.
    # Follows a single redirect, no recursive redirects (a feature in a test framework).
    if (@response.code == "301" or @response.code == "302") and @testbeat.redirect
      @testbeat.logger.info{ "Redirecting #{@response.code} to #{@response['location']}" }
      redirectTo = URI.parse(@response['location'])
      redirectToPath = [redirectTo.path,redirectTo.query].join('?')
      #reqRedirect = req.new(redirectTo.path, req.to_hash()) # new(path, initheader = nil)
      reqRedirect = HTTP_VERBS[@testbeat.method].new(redirectToPath) # new(path, initheader = nil)
      if @testbeat.headers?
        @testbeat.headers.each {|name, value| reqRedirect[name] = value }
      end
      reqRedirect.body = req.body
      @response = http.request(reqRedirect)
      req = reqRedirect # Needed by auth support below.
      @testbeat.logger.info{ "Redirected #{@response.code}" }
    end

    if @response.code == "401" and @testbeat.user and not @testbeat.unauthenticated?
      u = @testbeat.user
      @testbeat.logger.info{ "Authenticating to #{@testbeat.resource} with #{u[:username]}:#{u[:password]}" }
      req.basic_auth u[:username], u[:password]
      @response = http.request(req)
    end

    @testbeat.logger.info{ "Request #{@testbeat.resource} responded #{@response.code} #{@response.message}" }
    $_testbeat_rest_reuse[reuse_id] = @response
    return @response
  end
end