riak-client 0.9.8 → 1.0.0.beta

data/lib/riak/client/curb_backend.rb

@@ -1,89 +0,0 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-
-require 'thread'
-require 'riak/client/http_backend'
-require 'riak/client/pump'
-require 'riak/failed_request'
-
-module Riak
-  class Client
-    # An HTTP backend for Riak::Client that uses the 'curb' library/gem.
-    # Conforms to the Riak::Client::HTTPBackend interface.
-    class CurbBackend < HTTPBackend
-      def self.configured?
-        begin
-          require 'curb'
-          true
-        rescue LoadError
-          false
-        end
-      end
-
-      private
-      def perform(method, uri, headers, expect, data=nil, &block)
-        # Setup
-        curl.headers = RequestHeaders.new(headers).to_a
-        curl.url = uri.to_s
-        response_headers.initialize_http_header(nil)
-        if block_given?
-          curl.on_body(&Pump.new(block))
-        else
-          curl.on_body
-        end
-        # Perform
-        case method
-        when :post
-          data = data.read if data.respond_to?(:read)
-          curl.http_post(data)
-        when :put
-          # Hacks around limitations in curb's PUT semantics
-          _headers, curl.headers = curl.headers, {}
-          curl.put_data = data
-          curl.headers = RequestHeaders.new(curl.headers).to_a + _headers
-          curl.http("PUT")
-        else
-          curl.send("http_#{method}")
-        end
-
-        # Verify
-        if valid_response?(expect, curl.response_code)
-          result = { :headers => response_headers.to_hash, :code => curl.response_code.to_i }
-          if return_body?(method, curl.response_code, block_given?)
-            result[:body] = curl.body_str
-          end
-          result
-        else
-          raise HTTPFailedRequest.new(method, expect, curl.response_code, response_headers.to_hash, curl.body_str)
-        end
-      end
-
-      def curl
-        Thread.current[:curl_easy_handle] ||= Curl::Easy.new.tap do |c|
-          configure_ssl(c) if @client.ssl_enabled?
-
-          c.follow_location = false
-          c.on_header do |header_line|
-            response_headers.parse(header_line)
-            header_line.size
-          end
-        end
-      end
-
-      def configure_ssl(curl)
-        curl.ssl_verify_peer = @client.ssl_options[:verify_mode] == "peer"
-      end
-    end
-  end
-end

data/spec/riak/curb_backend_spec.rb

@@ -1,76 +0,0 @@
-# Copyright 2010 Sean Cribbs, Sonian Inc., and Basho Technologies, Inc.
-#
-#    Licensed under the Apache License, Version 2.0 (the "License");
-#    you may not use this file except in compliance with the License.
-#    You may obtain a copy of the License at
-#
-#        http://www.apache.org/licenses/LICENSE-2.0
-#
-#    Unless required by applicable law or agreed to in writing, software
-#    distributed under the License is distributed on an "AS IS" BASIS,
-#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#    See the License for the specific language governing permissions and
-#    limitations under the License.
-require File.expand_path("../spec_helper", File.dirname(__FILE__))
-
-begin
-  require 'curb'
-rescue LoadError
-  warn "Skipping CurbBackend specs, curb library not found."
-else
-  $mock_server = DrbMockServer
-  $mock_server.maybe_start
-
-  describe Riak::Client::CurbBackend do
-    def setup_http_mock(method, uri, options={})
-      method = method.to_s.upcase
-      uri = URI.parse(uri)
-      path = uri.path || "/"
-      query = uri.query || ""
-      status = options[:status] ? Array(options[:status]).first.to_i : 200
-      body = options[:body] || []
-      headers = options[:headers] || {}
-      headers['Content-Type'] ||= "text/plain"
-      @_mock_set = [status, headers, method, path, query, body]
-      $mock_server.expect(*@_mock_set)
-    end
-
-    before :each do
-      @client = Riak::Client.new(:http_port => $mock_server.port, :http_backend => :Curb) # Point to our mock
-      @backend = @client.http
-      @_mock_set = false
-    end
-
-    after :each do
-      if @_mock_set
-        $mock_server.satisfied.should be_true("Expected #{@_mock_set.inspect}, failed")
-      end
-      Thread.current[:curl_easy_handle] = nil
-    end
-
-    it_should_behave_like "HTTP backend"
-
-    it "should split long headers into 8KB chunks" do
-      setup_http_mock(:put, @backend.path("/riak/","foo").to_s, :body => "ok")
-      lambda do
-        @backend.put(200, "/riak/", "foo", "body",{"Long-Header" => (["12345678"*10]*100).join(", ") })
-        headers = @backend.send(:curl).headers
-        headers.should be_kind_of(Array)
-        headers.select {|h| h =~ /^Long-Header:/ }.should have(2).items
-        headers.select {|h| h =~ /^Long-Header:/ }.should be_all {|h| h.size < 8192 }
-      end.should_not raise_error
-    end
-
-    it "should support IO objects as the request body" do
-      file = File.open(File.expand_path("../../fixtures/cat.jpg", __FILE__))
-      lambda do
-        setup_http_mock(:put, @backend.path("/riak/","foo").to_s, :body => "ok")
-        @backend.put(200, "/riak/", "foo", file,{})
-      end.should_not raise_error
-      lambda do
-        setup_http_mock(:post, @backend.path("/riak/","foo").to_s, :body => "ok")
-        @backend.post(200, "/riak/", "foo", file, {})
-      end.should_not raise_error
-    end
-  end
-end