pauldix-typhoeus 0.0.8 → 0.0.10

data/lib/typhoeus/remote.rb

@@ -6,19 +6,64 @@ module Typhoeus
   end
     
   module ClassMethods
+    def mock(method, args = {})
+      @remote_mocks ||= {}
+      @remote_mocks[method] ||= {}
+      args[:code]    ||= 200
+      args[:body]    ||= ""
+      args[:headers] ||= ""
+      args[:time]    ||= 0
+      url = args.delete(:url)
+      url ||= :catch_all
+      @remote_mocks[method][url] = args
+    end
+    
+    def get_mock(method, url, options)
+      return nil unless @remote_mocks
+      if @remote_mocks.has_key? method
+        if @remote_mocks[method].has_key? url
+          get_mock_and_run_handlers(@remote_mocks[method][url], options)
+        elsif @remote_mocks[method].has_key? :catch_all
+          get_mock_and_run_handlers(@remote_mocks[method][:catch_all], options)
+        else
+          nil
+        end
+      else
+        nil
+      end
+    end
+    
+    def get_mock_and_run_handlers(response_args, options)
+      response = Response.new(response_args[:code], response_args[:headers], response_args[:body], response_args[:time])
+      if response.code >= 200 && response.code < 300 && options.has_key?(:on_success)
+        response = options[:on_success].call(response)
+      elsif options.has_key?(:on_failure)
+        response = options[:on_failure].call(response)
+      end
+      response
+    end
+    
     def get(url, options = {})
+      mock_object = get_mock(:get, url, options)
+      return mock_object if mock_object
       remote_proxy_object(url, :get, options)
     end
     
     def post(url, options = {}, &block)
+      mock_object = get_mock(:post, url, options)
+      return mock_object if mock_object
       remote_proxy_object(url, :post, options)
     end
 
     def put(url, options = {}, &block)
+      mock_object = get_mock(:put, url, options)
+      return mock_object if mock_object
       remote_proxy_object(url, :put, options)
     end
     
     def delete(url, options = {}, &block)
+      mock_object = get_mock(:delete, url, options)
+      return mock_object if mock_object
       remote_proxy_object(url, :delete, options)
     end
     

data/lib/typhoeus/remote_proxy_object.rb

@@ -16,7 +16,7 @@ module Typhoeus
     def method_missing(sym, *args, &block)
       unless @proxied_object
         if @cache && @cache_key
-          @proxied_object = @cache.get(@cache_key)
+          @proxied_object = @cache.get(@cache_key) rescue nil
         end
         
         unless @proxied_object

data/lib/typhoeus.rb

@@ -12,7 +12,7 @@ require 'typhoeus/remote_proxy_object'
 require 'typhoeus/response'
 
 module Typhoeus
-  VERSION = "0.0.8"
+  VERSION = "0.0.10"
   
   def self.easy_object_pool
     @easy_objects ||= []

data/spec/typhoeus/remote_spec.rb

@@ -14,6 +14,111 @@ describe Typhoeus do
   after(:all) do
     stop_method_server(@pid)
   end
+  
+  describe "mocking" do
+    it "should mock out GET" do
+      @klass.mock(:get)
+      response = @klass.get("http://mock_url")
+      response.code.should == 200
+      response.body.should == ""
+      response.headers.should == ""
+      response.time.should == 0
+    end
+    
+    it "should mock out PUT" do
+      @klass.mock(:put)
+      response = @klass.put("http://mock_url")
+      response.code.should == 200
+      response.body.should == ""
+      response.headers.should == ""
+      response.time.should == 0
+    end
+
+    it "should mock out POST" do
+      @klass.mock(:post)
+      response = @klass.post("http://mock_url")
+      response.code.should == 200
+      response.body.should == ""
+      response.headers.should == ""
+      response.time.should == 0
+    end
+    
+    it "should mock out DELETE" do
+      @klass.mock(:delete)
+      response = @klass.delete("http://mock_url")
+      response.code.should == 200
+      response.body.should == ""
+      response.headers.should == ""
+      response.time.should == 0
+    end
+    
+    it "should take an optional url" do
+      @klass.mock(:get, :url => "http://stuff")
+      response = @klass.get("http://stuff")
+      response.code.should == 200
+      response.body.should == ""
+      response.headers.should == ""
+      response.time.should == 0
+      
+      @klass.get("http://localhost:234234234").code.should == 0
+    end
+    
+    describe "remote methods" do
+      it "should work for defined remote methods" do
+        @klass.instance_eval do
+          define_remote_method :do_stuff, :base_uri => "http://localhost:3001", :on_success => lambda {|r| r.body.should == "hi"; :great_success}
+        end
+        @klass.mock(:get, :url => "http://localhost:3001", :body => "hi")
+        @klass.do_stuff.should == :great_success
+      end
+      
+      it "should call the on failure handler for remote methods" do
+        @klass.instance_eval do
+          define_remote_method :do_stuff, :base_uri => "http://localhost:3001", :on_failure => lambda {|r| r.body.should == "hi"; :fail}
+        end
+        @klass.mock(:get, :url => "http://localhost:3001", :body => "hi", :code => 500)
+        @klass.do_stuff.should == :fail
+      end
+    end
+    
+    describe "response hash" do
+      it "should use provided code" do
+        @klass.mock(:get, :url => "http://localhost/whatever", :code => 301)
+        response = @klass.get("http://localhost/whatever")
+        response.code.should == 301
+        response.body.should == ""
+        response.headers.should == ""
+        response.time.should == 0
+      end
+
+      it "should use provided body" do
+        @klass.mock(:get, :url => "http://localhost/whatever", :body => "hey paul")
+        response = @klass.get("http://localhost/whatever")
+        response.code.should == 200
+        response.body.should == "hey paul"
+        response.headers.should == ""
+        response.time.should == 0
+      end
+
+      it "should use provided headers" do
+        @klass.mock(:get, :url => "http://localhost/whatever", :headers => "whooo, headers!")
+        response = @klass.get("http://localhost/whatever")
+        response.code.should == 200
+        response.body.should == ""
+        response.headers.should == "whooo, headers!"
+        response.time.should == 0
+      end
+      
+      it "should use provided time" do
+        @klass.mock(:get, :url => "http://localhost/whatever", :time => 123)
+        response = @klass.get("http://localhost/whatever")
+        response.code.should == 200
+        response.body.should == ""
+        response.headers.should == ""
+        response.time.should == 123
+      end
+    end
+  end
 
   describe "get" do
     it "should add a get method" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: pauldix-typhoeus
 version: !ruby/object:Gem::Version 
-  version: 0.0.8
+  version: 0.0.10
 platform: ruby
 authors: 
 - Paul Dix