minitest-stub-const 0.5 → 0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3c368abae48ce6602249a2ab7e84151f0c0791bc
-  data.tar.gz: 604721ca05230092a7d3908d062c3643c6a9f41c
+  metadata.gz: 509b3a1937e6fe91ce920a6025654d83f8dead26
+  data.tar.gz: 12fe651bf614e29b271edc788e878b0e59069cf8
 SHA512:
-  metadata.gz: f6096aee46275edcd51e6675c35c80ac223ca6d6062dd4fd91304a6e736a189a2d103fb84d7a61de9414a1fe1c51a6957129e92c2a05477ad3599c58248c6eba
-  data.tar.gz: 1b1e4d2376d0bbb965ff4b1b0b7809775a975070f81cec2ed350aa0893ad18c023580a5c150aa4227fac449d572e9376096be0fd0f85f6aabebc6c7a26eedc33
+  metadata.gz: 4b0bb35ed1ce85a8e89eaf76d3116b7e71d79108d2d55830dc1908fff0f770b1e9e9c3c36600997612653400f17a14b9c41a897122b42eb556bdfdbd1d6fab3d
+  data.tar.gz: 947a70ac83373271262508d208c201622963850766d7fd379472cf751ad00e0584a9d5c5178a0bc862fc8beec100692190cc3dfad319628e7f7964979fd30856

data/README.md

@@ -3,7 +3,7 @@
 [![Build Status](https://travis-ci.org/adammck/minitest-stub-const.svg)](https://travis-ci.org/adammck/minitest-stub-const)
 
 Stub constants for the duration of a block in MiniTest.  
-Like RSpec's [stub_const] [rspec], but boring and non-magical.
+Similar to RSpec's [stub_const] [rspec].
 
 
 ## Example
@@ -51,6 +51,7 @@ describe ThingAdder do
       SomeLib.stub_const(:Thing, mock) do
         adder.add_thing
       end
+
       assert mock.verify
     end
   end

data/lib/minitest/stub_const.rb

@@ -8,24 +8,54 @@ class Object
   #   module Foo
   #     BAR = :original
   #   end
-  #   
+  #
   #   Foo.stub_const(:BAR, :stubbed) do
   #     Foo::BAR
   #   end
   #   # => :stubbed
-  #   
+  #
   #   Foo::BAR
   #   # => :original
-  def stub_const(name, val, &block)
-    defined = const_defined?(name)
-    orig = const_get(name) if defined
-    silence_warnings { const_set(name, val) }
+  def stub_const(name, val = nil, &block)
+    stub_consts(name => val, &block)
+  end
+
+  # Same as stub_const except it supports a Hash of +name+ to +value+ pairs
+  # of the constants that will be stubbed for the duration of the +block+.
+  #
+  # Example:
+  #
+  #   module Foo
+  #     BAR = :original1
+  #     BAZ = :original2
+  #   end
+  #
+  #   Foo.stub_consts(BAR: :stubble, BAZ: :stubby) do
+  #     [Foo::BAR, Foo::BAZ]
+  #   end
+  #   # => [:stubble, :stubby]
+  #
+  #   [Foo::BAR, Foo::BAZ]
+  #   # => [:original1, :original2]
+  def stub_consts(consts, &block)
+    original_values = {}
+    consts.each_pair do |name, val|
+      if const_defined?(name)
+        original_value = const_get(name)
+        original_values[name] = original_value
+      end
+      silence_warnings { const_set(name, val) }
+    end
+
     yield
+
   ensure
-    if defined
-      silence_warnings { const_set(name, orig) }
-    else
-      remove_const(name)
+    consts.keys.each do |name|
+      if original_values.key?(name)
+        silence_warnings { const_set(name, original_values[name]) }
+      else
+        remove_const(name)
+      end
     end
   end
 

data/test/test_stub_const.rb

@@ -8,6 +8,12 @@ module A
       :old
     end
   end
+
+  module C
+    def self.who
+      :you
+    end
+  end
 end
 
 describe 'Object' do
@@ -65,4 +71,46 @@ describe 'Object' do
       refute defined?(A::X)
     end
   end
+
+  describe '#stub_consts' do
+    before do
+      @mock = MiniTest::Mock.new
+      @mock.expect(:what, :new)
+
+      @mock2 = MiniTest::Mock.new
+      @mock2.expect(:who, :me)
+
+      @mock3 = MiniTest::Mock.new
+      @mock3.expect(:where, :there)
+    end
+
+    def run_stub_block
+      A.stub_consts(B: @mock, C: @mock2, D: @mock3) do
+        assert_equal :new, A::B.what
+        assert_equal :me, A::C.who
+        assert_equal :there, A::D.where
+      end
+    end
+
+    it 'replaces a set of constants for the duration of a block' do
+      run_stub_block
+
+      @mock.verify
+      @mock2.verify
+      @mock3.verify
+    end
+
+    it 'restores previous values' do
+      run_stub_block
+
+      assert_equal :old, A::B.what
+      assert_equal :you, A::C.who
+    end
+
+    it 'removes constants that were undefined prior to the block' do
+      run_stub_block
+
+      refute A.const_defined?(:D)
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: minitest-stub-const
 version: !ruby/object:Gem::Version
-  version: '0.5'
+  version: '0.6'
 platform: ruby
 authors:
 - Adam Mckaig
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-09-17 00:00:00.000000000 Z
+date: 2016-11-29 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -42,7 +42,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.3
+rubygems_version: 2.0.14.1
 signing_key: 
 specification_version: 4
 summary: Stub constants for the duration of a block in MiniTest