injected definitions copy resolvedDependencyBlock

AliSoftware · Nov 25, 2015 · ff4f365 · ff4f365
1 parent 75bbfd3
commit ff4f365
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dip/Dip/Definition.swift b/Dip/Dip/Definition.swift
@@ -97,6 +97,8 @@ public final class DefinitionOf<T, F>: Definition {
       fatalError("You can not change resolveDependencies block after it was set.")
     }
     self.resolveDependenciesBlock = block
+    self.injectedDefinition?.resolveDependenciesBlock = { block($0, $1 as! T) }
+    self.injectedWeakDefinition?.resolveDependenciesBlock = { block($0, $1 as! T) }
     return self
   }