mock_lobster 0.7 → 0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c64baa6dc25d834854c1e3ace6dd65810e9fd78b
-  data.tar.gz: 9894d875adf471c6be24fa0ec3898a02a013d7aa
+  metadata.gz: 95938acbaa892a320c315a450cc7667595087273
+  data.tar.gz: ce63347d18343ca105072d92bd931fba0a152a58
 SHA512:
-  metadata.gz: bbe8015a0b97aebc2b02065adc4b220a1f73db75824b18a2a0c9158eb94884ce8e7cd36592b2e4bf399c346b14247532153e5f0b731b7c639ea7da2f771e85da
-  data.tar.gz: b9a6eb72faf31252512002833e7e9123f132490d33ba7e6682e9cf197e94c1a29f2c27771d13b01cb5100bed23aeb00ab4aeae17c051bcac98c68b35939b106a
+  metadata.gz: 24854008be720906f525354bf8331d04542563f974fa598860a9a679d8efad0501a873da5471e00c2b76199deaa5178428a23b1df8a8a44a2b1e319964f083e3
+  data.tar.gz: 57027b10f26592974d826dd5012a63e85650ad2bc7dd58d632d8cbd67aea8cb20192c205613caa657cde8261bf6f1bf96344d07be6b7762add401ae52834d936

data/lib/mock_lobster.rb

@@ -5,14 +5,29 @@ class Object
     Mock.new(OpenStruct.new(method))
   end
 
-  def stub(method)
-    method.each_pair do |method_name, return_value|
-      self.send(:define_singleton_method, method_name) do
-        return return_value
+  def stubbify(hash)
+    index = 0
+    hash.each do |method_name, return_value|
+      if methods.include?(hash.to_a[index].first)
+        instance_variable_set("@copied_#{method_name}", method(hash.to_a[index].first))
+        index += 1
+        (@altered_methods ||= []) << method_name
       end
+      self.send(:define_singleton_method, method_name) { return_value }
     end
   end
 
+  def reset
+    modified_variables = instance_variables.select {|e| e.to_s if e.to_s.include?("copied")}
+    @altered_methods.each do |name|
+      self.send(:define_singleton_method, name) do
+        instance_variable_get("@copied_" + name.to_s).call
+      end
+      modified_variables.shift  #
+    end
+  end
+
+
   def expect(method, param)
     Expectation.new(method, param, self)
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mock_lobster
 version: !ruby/object:Gem::Version
-  version: '0.7'
+  version: '0.8'
 platform: ruby
 authors:
 - Lev Kravinsky
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-24 00:00:00.000000000 Z
+date: 2015-07-31 00:00:00.000000000 Z
 dependencies: []
 description: see github.com/levthedev/mock_lobster for documentation
 email: kravinskylev@gmail.com