ClassType: namespace is passed to methods immediately

Author: dg

src/PhpGenerator/ClassType.php

@@ -83,7 +83,7 @@ public static function from($from)
 		}
 		foreach ($from->getMethods() as $method) {
 			if ($method->getDeclaringClass() == $from) { // intentionally ==
-				$class->methods[$method->getName()] = Method::from($method);
+				$class->methods[$method->getName()] = Method::from($method)->setNamespace($class->namespace);
 			}
 		}
 		return $class;
@@ -121,10 +121,6 @@ public function __toString()
 			return array_map(array($namespace, 'unresolveName'), $arr);
 		};
 
-		foreach ($this->methods as $method) {
-			$method->setNamespace($this->namespace);
-		}
-
 		return Strings::normalize(
 			($this->documents ? str_replace("\n", "\n * ", "/**\n" . implode("\n", $this->documents)) . "\n */\n" : '')
 			. ($this->abstract ? 'abstract ' : '')
@@ -456,7 +452,7 @@ public function setMethods(array $methods)
 			if (!$v instanceof Method) {
 				throw new Nette\InvalidArgumentException('Argument must be Nette\PhpGenerator\Method[].');
 			}
-			$this->methods[$v->getName()] = $v;
+			$this->methods[$v->getName()] = $v->setNamespace($this->namespace);
 		}
 		return $this;
 	}
@@ -495,7 +491,7 @@ public function addMethod($name)
 		} else {
 			$method->setVisibility('public');
 		}
-		return $this->methods[$name] = $method;
+		return $this->methods[$name] = $method->setNamespace($this->namespace);
 	}
 
 }