Handle nil intermediate object on deep fetch

1 files changed, 14 insertions, 0 deletions

diff --git a/spec/hashie/extensions/deep_fetch_spec.rb b/spec/hashie/extensions/deep_fetch_spec.rb
index 3459042..3fd4b1d 100644
--- a/spec/hashie/extensions/deep_fetch_spec.rb
+++ b/spec/hashie/extensions/deep_fetch_spec.rb
@@ -11,6 +11,7 @@ module Hashie
               { title: 'Call of the Wild' },
               { title: 'Moby Dick' }
             ],
+            shelves: nil,
             location: {
               address: '123 Library St.'
             }
@@ -62,6 +63,19 @@ module Hashie
                 )
               end
             end
+
+            context 'when the nested object is nil' do
+              it 'raises an UndefinedPathError' do
+                expect do
+                  instance.deep_fetch(:library, :shelves, :address)
+                end.to(
+                  raise_error(
+                    DeepFetch::UndefinedPathError,
+                    'Could not fetch path (library > shelves > address) at address'
+                  )
+                )
+              end
+            end
           end
         end
       end