imperium 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: '0843f656e4b5c634136bdcddd59a8a81528ca45b'
-  data.tar.gz: 85309c76a17789e23090a36c8d076c119755d1c9
+  metadata.gz: f59bdee1b6ac9ba26ac29376201e2287dc2a0d40
+  data.tar.gz: 607de30419fe22615ff7449c70e27f8757520278
 SHA512:
-  metadata.gz: 1e2acf912515f5ebd12282d4f7f6f13a164b386741a0578c8a2180d21f8828552c1ed6c480396ce057486fa7fb8d56e5315be11fc08b306eeaa0c8317e516c7f
-  data.tar.gz: 2538b80f19905a4b360252681615fce837b4f28e5c4a9e59c3b42914454e2e303756fe67818aaf1882c9dc82d2812b2a3cf0e5d513599a75e113b6b33d80efe2
+  metadata.gz: f14950628947bcf8e279b07d6ad39d3cf812347d5899962fbfef983c03a461434ceca309eb17973c0d3e9917d31252cd44ae762774d0f99cf828c982b542c45e
+  data.tar.gz: 99ce04ef3f3cf342bcf51b0566acfee85b6f7daa97b99e902bffe04670377de1b536c22bac5217cba571cc6329920574fabd8396673acc7567b77aa138d6e63c

data/lib/imperium/error.rb

@@ -1,5 +1,18 @@
 
 module Imperium
   class Error < StandardError; end
+
   class InvalidConsistencySpecification < Error; end
+
+  # Generic Network timeout error
+  class TimeoutError < Error; end
+
+  # Raised when opening the TCP socket times out
+  class ConnectTimeout < TimeoutError; end
+
+  # Raised when sending the request body took too long
+  class SendTimeout < TimeoutError; end
+
+  # Raised when the remote server took too long to respond.
+  class ReceiveTimeout < TimeoutError; end
 end

data/lib/imperium/http_client.rb

@@ -13,19 +13,25 @@ module Imperium
     end
 
     def delete(path)
-      url = config.url.join(path)
-      @driver.delete(url)
+      wrapping_timeouts do
+        url = config.url.join(path)
+        @driver.delete(url)
+      end
     end
 
     def get(path, query: {})
-      url = config.url.join(path)
-      url.query_values = query
-      @driver.get(url, header: build_request_headers)
+      wrapping_timeouts do
+        url = config.url.join(path)
+        url.query_values = query
+        @driver.get(url, header: build_request_headers)
+      end
     end
 
     def put(path, value)
-      url = config.url.join(path)
-      @driver.put(url, body: value)
+      wrapping_timeouts do
+        url = config.url.join(path)
+        @driver.put(url, body: value)
+      end
     end
 
     private
@@ -37,5 +43,17 @@ module Imperium
         {}
       end
     end
+
+    # We're doing this wrap and re-raise dance to give a more consistent set of
+    # exceptions that can come from us.
+    def wrapping_timeouts
+      yield
+    rescue HTTPClient::ConnectTimeout => ex
+      raise Imperium::ConnectTimeout, ex.message
+    rescue HTTPClient::SendTimeout => ex
+      raise Imperium::SendTimeout, ex.message
+    rescue HTTPClient::ReceiveTimeout => ex
+      raise Imperium::ReceiveTimeout, ex.message
+    end
   end
 end

data/lib/imperium/testing.rb

@@ -0,0 +1,37 @@
+require 'imperium'
+
+module Imperium
+  # A collection of functions to build responses for use in tests needing mock
+  # responses.
+  module Testing
+    class Client < Imperium::Client
+      def initialize
+      end
+      public :hashify_options
+    end
+    private_constant :Client
+
+    MockResponse = Struct.new(:content, :status, :headers)
+
+    @client = Client.new
+    def self.kv_get_response(body: '[]', status: 200, headers: {}, prefix: '', options: [])
+      expanded_options = @client.hashify_options(options)
+      string_body = if String === body
+                      body
+                    else
+                      body.map { |obj|
+                        obj['Value'] = Base64.encode64(obj['Value']) if obj['Value']
+                        obj[:Value] = Base64.encode64(obj[:Value]) if obj[:Value]
+                        obj
+                      }.to_json
+                    end
+
+      response = MockResponse.new(string_body, status, headers)
+      KVGETResponse.new(response, prefix: prefix, options: expanded_options)
+    end
+
+    def self.kv_not_found_response(headers: {}, options: [])
+      kv_get_response(body: '', status: 404, headers: headers, options: options)
+    end
+  end
+end

data/lib/imperium/version.rb

@@ -1,3 +1,3 @@
 module Imperium
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: imperium
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Tyler Pickett
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-03-29 00:00:00.000000000 Z
+date: 2017-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: addressable
@@ -165,6 +165,7 @@ files:
 - lib/imperium/kv_get_response.rb
 - lib/imperium/kv_pair.rb
 - lib/imperium/response.rb
+- lib/imperium/testing.rb
 - lib/imperium/version.rb
 homepage: https://github.com/instructure/imperium
 licenses: