lib/nanoc/base/compilation/dependency_tracker.rb in nanoc-4.3.0 vs lib/nanoc/base/compilation/dependency_tracker.rb in nanoc-4.3.1

@@ -10,27 +10,32 @@
 
       def bounce(_obj)
       end
     end
 
+    include Nanoc::Int::ContractsSupport
+
     def initialize(dependency_store)
       @dependency_store = dependency_store
       @stack = []
     end
 
+    contract C::Or[Nanoc::Int::Item, Nanoc::Int::Layout] => C::Any
     def enter(obj)
       unless @stack.empty?
         Nanoc::Int::NotificationCenter.post(:dependency_created, @stack.last, obj)
         @dependency_store.record_dependency(@stack.last, obj)
       end
 
       @stack.push(obj)
     end
 
+    contract C::Or[Nanoc::Int::Item, Nanoc::Int::Layout] => C::Any
     def exit(_obj)
       @stack.pop
     end
 
+    contract C::Or[Nanoc::Int::Item, Nanoc::Int::Layout] => C::Any
     def bounce(obj)
       enter(obj)
       exit(obj)
     end
   end