Skip to content
This repository
tree: cf32baf915
Fetching contributors…

Cannot retrieve contributors at this time

file 190 lines (154 sloc) 5.882 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190
require 'abstract_unit'

class ConnectionTest < Test::Unit::TestCase
  ResponseCodeStub = Struct.new(:code)

  def setup
    @conn = ActiveResource::Connection.new('http://localhost')
    @matz = { :id => 1, :name => 'Matz' }
    @david = { :id => 2, :name => 'David' }
    @people = [ @matz, @david ].to_xml(:root => 'people')
    @people_single = [ @matz ].to_xml(:root => 'people-single-elements')
    @people_empty = [ ].to_xml(:root => 'people-empty-elements')
    @matz = @matz.to_xml(:root => 'person')
    @david = @david.to_xml(:root => 'person')
    @header = {'key' => 'value'}.freeze

    @default_request_headers = { 'Content-Type' => 'application/xml' }
    ActiveResource::HttpMock.respond_to do |mock|
      mock.get "/people/2.xml", @header, @david
      mock.get "/people.xml", {}, @people
      mock.get "/people_single_elements.xml", {}, @people_single
      mock.get "/people_empty_elements.xml", {}, @people_empty
      mock.get "/people/1.xml", {}, @matz
      mock.put "/people/1.xml", {}, nil, 204
      mock.put "/people/2.xml", {}, @header, 204
      mock.delete "/people/1.xml", {}, nil, 200
      mock.delete "/people/2.xml", @header, nil, 200
      mock.post "/people.xml", {}, nil, 201, 'Location' => '/people/5.xml'
      mock.post "/members.xml", {}, @header, 201, 'Location' => '/people/6.xml'
      mock.head "/people/1.xml", {}, nil, 200
    end
  end

  def test_handle_response
    # 2xx and 3xx are valid responses.
    [200, 299, 300, 399].each do |code|
      expected = ResponseCodeStub.new(code)
      assert_equal expected, handle_response(expected)
    end

    # 400 is a bad request (e.g. malformed URI or missing request parameter)
    assert_response_raises ActiveResource::BadRequest, 400

    # 401 is an unauthorized request
    assert_response_raises ActiveResource::UnauthorizedAccess, 401

    # 403 is a forbidden requst (and authorizing will not help)
    assert_response_raises ActiveResource::ForbiddenAccess, 403

    # 404 is a missing resource.
    assert_response_raises ActiveResource::ResourceNotFound, 404

    # 405 is a missing not allowed error
    assert_response_raises ActiveResource::MethodNotAllowed, 405

    # 409 is an optimistic locking error
    assert_response_raises ActiveResource::ResourceConflict, 409

    # 422 is a validation error
    assert_response_raises ActiveResource::ResourceInvalid, 422

    # 4xx are client errors.
    [402, 499].each do |code|
      assert_response_raises ActiveResource::ClientError, code
    end

    # 5xx are server errors.
    [500, 599].each do |code|
      assert_response_raises ActiveResource::ServerError, code
    end

    # Others are unknown.
    [199, 600].each do |code|
      assert_response_raises ActiveResource::ConnectionError, code
    end
  end

  ResponseHeaderStub = Struct.new(:code, :message, 'Allow')
  def test_should_return_allowed_methods_for_method_no_allowed_exception
    begin
      handle_response ResponseHeaderStub.new(405, "HTTP Failed...", "GET, POST")
    rescue ActiveResource::MethodNotAllowed => e
      assert_equal "Failed with 405 HTTP Failed...", e.message
      assert_equal [:get, :post], e.allowed_methods
    end
  end

  def test_initialize_raises_argument_error_on_missing_site
    assert_raise(ArgumentError) { ActiveResource::Connection.new(nil) }
  end

  def test_site_accessor_accepts_uri_or_string_argument
    site = URI.parse("http://localhost")

    assert_raise(URI::InvalidURIError) { @conn.site = nil }

    assert_nothing_raised { @conn.site = "http://localhost" }
    assert_equal site, @conn.site

    assert_nothing_raised { @conn.site = site }
    assert_equal site, @conn.site
  end

  def test_timeout_accessor
    @conn.timeout = 5
    assert_equal 5, @conn.timeout
  end

  def test_get
    matz = @conn.get("/people/1.xml")
    assert_equal "Matz", matz["name"]
  end

  def test_head
    response = @conn.head("/people/1.xml")
    assert response.body.blank?
    assert_equal 200, response.code
  end

  def test_get_with_header
    david = @conn.get("/people/2.xml", @header)
    assert_equal "David", david["name"]
  end

  def test_get_collection
    people = @conn.get("/people.xml")
    assert_equal "Matz", people[0]["name"]
    assert_equal "David", people[1]["name"]
  end
  
  def test_get_collection_single
    people = @conn.get("/people_single_elements.xml")
    assert_equal "Matz", people[0]["name"]
  end
  
  def test_get_collection_empty
    people = @conn.get("/people_empty_elements.xml")
    assert_equal [], people
  end

  def test_post
    response = @conn.post("/people.xml")
    assert_equal "/people/5.xml", response["Location"]
  end

  def test_post_with_header
    response = @conn.post("/members.xml", @header)
    assert_equal "/people/6.xml", response["Location"]
  end

  def test_put
    response = @conn.put("/people/1.xml")
    assert_equal 204, response.code
  end

  def test_put_with_header
    response = @conn.put("/people/2.xml", @header)
    assert_equal 204, response.code
  end

  def test_delete
    response = @conn.delete("/people/1.xml")
    assert_equal 200, response.code
  end

  def test_delete_with_header
    response = @conn.delete("/people/2.xml", @header)
    assert_equal 200, response.code
  end

  uses_mocha('test_timeout') do
    def test_timeout
      @http = mock('new Net::HTTP')
      @conn.expects(:http).returns(@http)
      @http.expects(:get).raises(Timeout::Error, 'execution expired')
      assert_raises(ActiveResource::TimeoutError) { @conn.get('/people_timeout.xml') }
    end
  end

  protected
    def assert_response_raises(klass, code)
      assert_raise(klass, "Expected response code #{code} to raise #{klass}") do
        handle_response ResponseCodeStub.new(code)
      end
    end

    def handle_response(response)
      @conn.send!(:handle_response, response)
    end
end
Something went wrong with that request. Please try again.