added tests

Author: dg

tests/PhpGenerator/ClassType.interface.phpt

@@ -18,6 +18,8 @@ $interface
 	->addExtend('ITwo')
 	->addComment('Description of interface');
 
+Assert::same(['IOne', 'ITwo'], $interface->getExtends());
+
 $interface->addMethod('getForm');
 
 Assert::matchFile(__DIR__ . '/ClassType.interface.expect', (string) $interface);

tests/PhpGenerator/ClassType.phpt

@@ -13,13 +13,19 @@ require __DIR__ . '/../bootstrap.php';
 
 
 $class = new ClassType('Example');
+
+Assert::false($class->isFinal());
+Assert::false($class->isAbstract());
+Assert::same([], $class->getExtends());
+Assert::same([], $class->getTraits());
+
 $class
 	->setAbstract(TRUE)
 	->setFinal(TRUE)
 	->setExtends('ParentClass')
 	->addImplement('IExample')
 	->addImplement('IOne')
-	->addTrait('ObjectTrait')
+	->setTraits(['ObjectTrait'])
 	->addTrait('AnotherTrait', ['sayHello as protected'])
 	->addComment("Description of class.\nThis is example\n")
 	->addComment('@property-read Nette\Forms\Form $form')
@@ -28,6 +34,13 @@ $class
 
 Assert::same(['ROLE' => 'admin', 'ACTIVE' => FALSE], $class->getConsts());
 
+Assert::true($class->isFinal());
+Assert::true($class->isAbstract());
+Assert::same('ParentClass', $class->getExtends());
+Assert::same(['ObjectTrait', 'AnotherTrait'], $class->getTraits());
+Assert::count(2, $class->getConstants());
+Assert::type(Nette\PhpGenerator\Constant::class, $class->getConstants()['ROLE']);
+
 $class->addConstant('FORCE_ARRAY', new PhpLiteral('Nette\Utils\Json::FORCE_ARRAY'))
 	->setVisibility('private')
 	->addComment('Commented');
@@ -43,6 +56,8 @@ $p = $class->addProperty('sections', ['first' => TRUE])
 	->setStatic(TRUE);
 
 Assert::same($p, $class->getProperty('sections'));
+Assert::true($p->isStatic());
+Assert::null($p->getVisibility());
 
 $m = $class->addMethod('getHandle')
 	->addComment('Returns file handle.')
@@ -51,14 +66,27 @@ $m = $class->addMethod('getHandle')
 	->setBody('return $this->?;', ['handle']);
 
 Assert::same($m, $class->getMethod('getHandle'));
-
-$class->addMethod('getSections')
+Assert::true($m->isFinal());
+Assert::false($m->isStatic());
+Assert::false($m->isAbstract());
+Assert::false($m->getReturnReference());
+Assert::same('public', $m->getVisibility());
+Assert::same('return $this->handle;', $m->getBody());
+
+$m = $class->addMethod('getSections')
 	->setStatic(TRUE)
 	->setVisibility('protected')
 	->setReturnReference(TRUE)
 	->addBody('$mode = ?;', [123])
-	->addBody('return self::$sections;')
-	->addParameter('mode', new PhpLiteral('self::ORDER'));
+	->addBody('return self::$sections;');
+$m->addParameter('mode', new PhpLiteral('self::ORDER'));
+
+Assert::false($m->isFinal());
+Assert::true($m->isStatic());
+Assert::true($m->getReturnReference());
+Assert::false($m->getReturnNullable());
+Assert::null($m->getReturnType());
+Assert::same('protected', $m->getVisibility());
 
 $method = $class->addMethod('show')
 	->setAbstract(TRUE);
@@ -87,3 +115,9 @@ $parameters = $method->getParameters();
 Assert::count(2, $parameters);
 $method->setParameters(array_values($parameters));
 Assert::same($parameters, $method->getParameters());
+
+
+Assert::exception(function () {
+	$class = new ClassType;
+	$class->addMethod('method')->setVisibility('unknown');
+}, Nette\InvalidArgumentException::class, 'Argument must be public|protected|private.');

tests/PhpGenerator/Closure.phpt

@@ -24,6 +24,24 @@ Assert::match(
 }', (string) $function);
 
 
+$uses = $function->getUses();
+Assert::count(2, $uses);
+Assert::type(Nette\PhpGenerator\Parameter::class, $uses[0]);
+Assert::type(Nette\PhpGenerator\Parameter::class, $uses[1]);
+
+$uses = $function->setUses([$uses[0]]);
+
+Assert::match(
+'function &($a, $b) use ($this) {
+	return $a + $b;
+}', (string) $function);
+
+Assert::exception(function () {
+	$function = new Closure;
+	$function->setUses([123]);
+}, Nette\InvalidArgumentException::class);
+
+
 $closure = function (stdClass $a, $b = NULL) {};
 $function = Closure::from($closure);
 Assert::match(

tests/PhpGenerator/PhpNamespace.phpt

@@ -33,6 +33,7 @@ Assert::same('\A', $namespace->unresolveName('A'));
 Assert::same('A', $namespace->unresolveName('foo\A'));
 
 $namespace->addUse('Bar\C');
+Assert::same(['C' => 'Bar\\C'], $namespace->getUses());
 
 Assert::same('\Bar', $namespace->unresolveName('Bar'));
 Assert::same('C', $namespace->unresolveName('\bar\C'));
@@ -50,6 +51,9 @@ Assert::same($namespace, $classA->getNamespace());
 $interfaceB = $namespace->addInterface('B');
 Assert::same($namespace, $interfaceB->getNamespace());
 
+Assert::count(2, $namespace->getClasses());
+Assert::type(Nette\PhpGenerator\ClassType::class, $namespace->getClasses()['A']);
+
 Assert::exception(function () use ($namespace) {
 	$traitC = $namespace->addTrait('C');
 	Assert::same($namespace, $traitC->getNamespace());