object_mock 0.1.1 → 0.1.2

data/lib/object_mock.rb

@@ -16,8 +16,11 @@ class Object
   def mock(methods)
     do_mock(methods)
     if block_given?
-      yield
-      unmock(*methods.keys)
+      begin
+        yield
+      ensure
+        unmock(*methods.keys)
+      end
     end
   end
   
@@ -36,8 +39,11 @@ class Object
     do_mock(methods, :instance, lambda {|m| instance_methods.collect{|n| n.to_sym}.include?(m.to_sym)}, self)
  
     if block
-      block.call
-      class_unmock(*methods.keys)
+      begin
+        block.call
+      ensure
+        class_unmock(*methods.keys)
+      end
     end
     
   end

data/object_mock.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name     = "object_mock"
-  s.version  = "0.1.1"
+  s.version  = "0.1.2"
   s.date     = "2010-04-19"
   s.summary  = "A simple, flexible and reversible mocking solution"
   s.email    = "mohammed.shalaby@espace.com.eg"

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: object_mock
 version: !ruby/object:Gem::Version 
-  hash: 25
+  hash: 31
   prerelease: 
   segments: 
   - 0
   - 1
-  - 1
-  version: 0.1.1
+  - 2
+  version: 0.1.2
 platform: ruby
 authors: 
 - Mohammed Shalaby