lazy deps

lib/active_dry_deps/container.rb

@@ -5,15 +5,29 @@ class Container < Hash
 
     def resolve(container_key)
       unless key?(container_key)
-        self[container_key] = Object.const_get(container_key)
+        self[container_key] = Entry.new(value: Object.const_get(container_key))
       end
 
-      self[container_key]
+      self[container_key]&.value
     end
 
-    def register(container_key)
-      self[container_key.to_s] = yield
+    def register(container_key, &block)
+      self[container_key.to_s] = Entry.new(proc: block)
     end
 
+    Entry =
+      Struct.new(:input) do
+        def initialize(input = {})
+          super
+          @value = input[:value] if input.key?(:value)
+        end
+
+        def value
+          return @value if defined? @value
+
+          @value = input.fetch(:proc).call
+        end
+      end
+
   end
 end

lib/active_dry_deps/stub.rb

@@ -4,8 +4,8 @@ module ActiveDryDeps
 
   module StubDeps
 
-    def stub(key, proxy_object, &block)
-      self::CONTAINER.stub(key, proxy_object, &block)
+    def stub(key, &block)
+      self::CONTAINER.stub(key, &block)
     end
 
     def unstub(*keys)
@@ -20,16 +20,8 @@ def self.extended(container)
       const_set(:CONTAINER_ORIG, container.dup)
     end
 
-    def stub(key, proxy_object)
-      if block_given?
-        begin
-          self[key] = proxy_object
-        ensure
-          unstub(key)
-        end
-      else
-        self[key] = proxy_object
-      end
+    def stub(key, &block)
+      self[key] = ActiveDryDeps::Container::Entry.new(proc: block)
     end
 
     def unstub(*unstub_keys)

spec/active_dry_deps_spec.rb

@@ -36,8 +36,8 @@ def expect_call_orig
       expect(CreateOrder.call).to eq %w[CreateDeparture CreateDeparture job-performed message-ok email-sent-hello]
     end
 
-    it 'direct stub with `Deps.stub`' do
-      Deps.stub('CreateDeparture', double(call: '1'))
+    it 'stubs dependency at the container level' do
+      Deps.stub('CreateDeparture') { double(call: '1') }
 
       expect(CreateOrder.call).to eq %w[1 1 job-performed message-ok email-sent-hello]
 
@@ -46,10 +46,15 @@ def expect_call_orig
       expect_call_orig
     end
 
-    it 'direct stub with `Deps.sub` with block' do
-      Deps.stub('CreateDeparture', double(call: '1')) do
-        expect(CreateOrder.call).to eq %w[1 1 job-performed message-ok email-sent-hello]
-      end
+    it 'raises exception when calls a stub block' do
+      expect {
+        Deps.stub('CreateDeparture') { raise StandardError, 'Something went wrong' }
+      }.not_to raise_error
+
+      expect { CreateOrder.call }
+        .to raise_error(StandardError, 'Something went wrong')
+
+      Deps.unstub
 
       expect_call_orig
     end