temporaries 0.1.0 → 0.2.0

data/CHANGELOG

@@ -1,3 +1,7 @@
+== 0.2.0 2012-06-08
+
+ * Methods to set temporary method definitions.
+
 == 0.1.0 2012-06-06
 
  * Support minitest/unit and minitest/spec.

data/README.markdown

@@ -8,6 +8,7 @@ Set things temporarily and declaratively. Best way to test.
  * Set values of instance variables.
  * Set values of class variables.
  * Set values of globals.
+ * Set definitions of methods.
  * Set a temporary directory.
 
 Do it for the length of a test, or within a specific block in your
@@ -35,6 +36,7 @@ Here's the full list of methods you can use:
     use_instance_variable_value(object, :name, value)
     use_class_variable_value(object, :name, value)
     use_global_value(:name, value)
+    use_method_definition(module, :name, ->(*args){defintion})
 
 Sigils in the `name` are unnecessary in the last 3 (unlike
 `instance_variable_get`, for instance).
@@ -73,6 +75,10 @@ You may also set the temporary for the duration of a particular block:
       ...
     end
 
+    with_method_definition module, :name, ->(*args){definition} do
+      ...
+    end
+
     with_temporary_directory path do
       # `tmp' refers to the directory.
     end

data/lib/temporaries/values.rb

@@ -100,35 +100,35 @@ module Temporaries
         end
 
         mod.class_eval <<-EOS, __FILE__, __LINE__ + 1
-          def push_#{name}_value(#{signature}, value)
+          def push_#{name}(#{signature}, value)
             #{save}
             push_temporary(#{stack_key}, original_value)
             #{set}
           end
 
-          def pop_#{name}_value(#{signature})
+          def pop_#{name}(#{signature})
             value = pop_temporary(#{stack_key})
             #{restore}
           end
 
-          def with_#{name}_value(#{signature}, value)
-            push_#{name}_value(#{signature}, value)
+          def with_#{name}(#{signature}, value)
+            push_#{name}(#{signature}, value)
             begin
               yield
             ensure
-              pop_#{name}_value(#{signature})
+              pop_#{name}(#{signature})
             end
           end
         EOS
 
         mod::ClassMethods.module_eval <<-EOS, __FILE__, __LINE__ + 1
-          def use_#{name}_value(#{signature}, value)
+          def use_#{name}(#{signature}, value)
             temporaries_adapter.before do
-              push_#{name}_value(#{signature}, value)
+              push_#{name}(#{signature}, value)
             end
 
             temporaries_adapter.after do
-              pop_#{name}_value(#{signature})
+              pop_#{name}(#{signature})
             end
           end
         EOS
@@ -148,7 +148,7 @@ module Temporaries
       helpers.define(self)
     end
 
-    define_helpers_for :constant do
+    define_helpers_for :constant_value do
       signature 'mod, constant'
       exists 'mod.const_defined?(constant)'
       get 'mod.const_get(constant)'
@@ -159,13 +159,13 @@ module Temporaries
       remove 'mod.send :remove_const, constant'
     end
 
-    define_helpers_for :attribute do
+    define_helpers_for :attribute_value do
       signature 'object, attribute'
       get 'object.send(attribute)'
       set 'object.send("#{attribute}=", value)'
     end
 
-    define_helpers_for :hash do
+    define_helpers_for :hash_value do
       signature 'hash, key'
       exists 'hash.key?(key)'
       get 'hash[key]'
@@ -173,7 +173,7 @@ module Temporaries
       remove 'hash.delete(key)'
     end
 
-    define_helpers_for :instance_variable do
+    define_helpers_for :instance_variable_value do
       signature 'object, name'
       exists 'object.instance_variable_defined?("@#{name}")'
       get 'object.instance_variable_get "@#{name}"'
@@ -181,7 +181,7 @@ module Temporaries
       remove 'object.send :remove_instance_variable, "@#{name}"'
     end
 
-    define_helpers_for :class_variable do
+    define_helpers_for :class_variable_value do
       signature 'klass, name'
       exists 'klass.class_variable_defined?("@@#{name}")'
       get 'klass.send :class_variable_get, "@@#{name}"'
@@ -189,10 +189,18 @@ module Temporaries
       remove 'klass.send :remove_class_variable, "@@#{name}"'
     end
 
-    define_helpers_for :global do
+    define_helpers_for :global_value do
       signature 'name'
       get 'eval("$#{name}")'
       set 'eval("$#{name} = value")'
     end
+
+    define_helpers_for :method_definition do
+      signature 'klass, name'
+      exists 'klass.method_defined?(name) && (method = klass.instance_method(name)) && method.owner.equal?(klass)'
+      get 'method'
+      set 'klass.__send__(:define_method, name, value)'
+      remove 'klass.__send__(:remove_method, name)'
+    end
   end
 end

data/lib/temporaries/version.rb

@@ -1,5 +1,5 @@
 module Temporaries
-  VERSION = [0, 1, 0]
+  VERSION = [0, 2, 0]
 
   class << VERSION
     include Comparable

data/spec/unit/temporaries/values_spec.rb

@@ -531,6 +531,96 @@ describe Temporaries::Values do
       $variable.should == 2
     end
   end
+
+  describe "#push_method_definition and #pop_method_definition" do
+    before do
+      @klass = Class.new
+      @instance = @klass.new
+    end
+
+    describe "when the method already exists" do
+      before do
+        @klass.send(:define_method, :meth) { 2 }
+      end
+
+      it "should set the given module's method to the pushed definition until it is popped" do
+        @instance.meth.should == 2
+        @context.push_method_definition @klass, :meth, lambda{3}
+        @instance.meth.should == 3
+        @context.pop_method_definition @klass, :meth
+        @instance.meth.should == 2
+      end
+
+      it "should be nestable" do
+        @instance.meth.should == 2
+        @context.push_method_definition @klass, :meth, lambda{3}
+        @instance.meth.should == 3
+        @context.push_method_definition @klass, :meth, lambda{5}
+        @instance.meth.should == 5
+        @context.pop_method_definition @klass, :meth
+        @instance.meth.should == 3
+        @context.pop_method_definition @klass, :meth
+        @instance.meth.should == 2
+      end
+    end
+
+    describe "when the method does not already exist" do
+      it "should set the given module's method to the pushed definition, and remove it when popped" do
+        @klass.method_defined?(:meth).should be_false
+        @context.push_method_definition @klass, :meth, lambda{3}
+        @instance.meth.should == 3
+        @context.pop_method_definition @klass, :meth
+        @klass.method_defined?(:meth).should be_false
+      end
+    end
+  end
+
+  describe "#with_method_definition" do
+    before do
+      @klass = Class.new{def meth; 2; end}
+      @instance = @klass.new
+    end
+
+    it "should set the given module's method to the given definition only during the given block" do
+      @instance.meth.should == 2
+      block_run = false
+      @context.with_method_definition @klass, :meth, lambda{3} do
+        block_run = true
+        @instance.meth.should == 3
+      end
+      block_run.should be_true
+      @instance.meth.should == 2
+    end
+
+    it "should be nestable" do
+      @instance.meth.should == 2
+      blocks_run = []
+      @context.with_method_definition @klass, :meth, lambda{3} do
+        blocks_run << 1
+        @instance.meth.should == 3
+        @context.with_method_definition @klass, :meth, lambda{5} do
+          blocks_run << 2
+          @instance.meth.should == 5
+        end
+        @instance.meth.should == 3
+      end
+      blocks_run.should == [1, 2]
+      @instance.meth.should == 2
+    end
+
+    it "should handle nonlocal exits" do
+      exception_class = Class.new(Exception)
+      @instance.meth.should == 2
+      begin
+        @context.with_method_definition @klass, :meth, lambda{3} do
+          @instance.meth.should == 3
+          raise exception_class, 'boom'
+        end
+      rescue exception_class => e
+      end
+      @instance.meth.should == 2
+    end
+  end
 end
 
 describe Temporaries::Values do
@@ -680,4 +770,28 @@ describe Temporaries::Values do
       $variable.should == 2
     end
   end
+
+  describe ".use_method_definition" do
+    before do
+      @klass = klass = Class.new { def meth; 2; end }
+      @instance = @klass.new
+
+      @context_class = Class.new(TestContext) do
+        include Temporaries::Values
+        use_method_definition klass, :meth, lambda{3}
+      end
+      @context = @context_class.new
+    end
+
+    it "should set the given module's constant to the given value for the duration of the run" do
+      @instance.meth.should == 2
+      block_run = false
+      @context.run do
+        block_run = true
+        @instance.meth.should == 3
+      end
+      block_run.should be_true
+      @instance.meth.should == 2
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: temporaries
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-06-06 00:00:00.000000000 Z
+date: 2012-06-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ritual
@@ -107,7 +107,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2522047447244818612
+      hash: -3674253645093316834
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -116,7 +116,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: 2522047447244818612
+      hash: -3674253645093316834
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24