injected definitions copy resolvedDependencyBlock

ilyapuchka · ilyapuchka · commit ff4f36511f80 · 2015-11-25T22:46:01.000+01:00
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
   }
   

Original file line number	Diff line number	Diff line change
`@@ -97,6 +97,8 @@ public final class DefinitionOf<T, F>: Definition {`
`97`	`97`	`fatalError("You can not change resolveDependencies block after it was set.")`
`98`	`98`	`}`
`99`	`99`	`self.resolveDependenciesBlock = block`
	`100`	`+ self.injectedDefinition?.resolveDependenciesBlock = { block($0, $1 as! T) }`
	`101`	`+ self.injectedWeakDefinition?.resolveDependenciesBlock = { block($0, $1 as! T) }`
`100`	`102`	`return self`
`101`	`103`	`}`
`102`	`104`