lib/henry/container.rb in henry-container-0.0.54 vs lib/henry/container.rb in henry-container-0.0.55

@@ -131,21 +131,22 @@
     end
 
     # Retrun the custom extended_context fo the given task.
     # @note default_extended_context will be used for undefined keys.
     #
+    # @param [String] task_name the target Task name.
     # @return [Hash] the Task custom extended context.
-    def extended_context
+    def extended_context(task_name)
       (self.params[task_name] || {})['extended_context']
     end
     
     # Returns the custom execution_context for the given task.
     # @note default_execution_context will be used for undefined keys.
     #
     # @param [String] task_name the target Task name.
     # @return [Hash] the Task custom execution_context.
-    def task_extended_context
-      self.default_extended_context.merge(self.extended_context[task_name] || {})
+    def task_extended_context(task_name)
+      self.default_extended_context.merge(self.extended_context(task_name) || {})
     end
 
 
     # Returns de task hints (tasks to be executed) or an empty array if they are not defined.
     #