/** * A function for test of code generator. * * @param string $name class name */ function define_class_for_code_genration_test() { $className = 'ExampleForCodeGenerationTest'; $class = new Stagehand_PHP_Class($className); $publicProperty = new Stagehand_PHP_Class_Property('a'); $protectedProperty = new Stagehand_PHP_Class_Property('b', 100); $privateProperty = new Stagehand_PHP_Class_Property('c', array(1, 3, 5)); $staticProperty = new Stagehand_PHP_Class_Property('d', 'static'); $protectedProperty->defineProtected(); $privateProperty->definePrivate(); $staticProperty->defineStatic(); $argumentA = new Stagehand_PHP_Class_Method_Argument('a'); $argumentB = new Stagehand_PHP_Class_Method_Argument('b'); $argumentC = new Stagehand_PHP_Class_Method_Argument('c'); $argumentD = new Stagehand_PHP_Class_Method_Argument('d'); $argumentE = new Stagehand_PHP_Class_Method_Argument('e'); $argumentF = new Stagehand_PHP_Class_Method_Argument('f'); $argumentB->setTypeHinting('array'); $argumentC->setRequirement(false); $argumentD->setRequirement(false); $argumentD->setValue('d'); $argumentE->setRequirement(false); $argumentE->setValue(array(2, 4, 6)); $argumentF->setReference(); $publicMethod = new Stagehand_PHP_Class_Method('foo'); $protectedMethod = new Stagehand_PHP_Class_Method('bar'); $privateMethod = new Stagehand_PHP_Class_Method('baz'); $staticMethod = new Stagehand_PHP_Class_Method('qux'); $abstractMethod = new Stagehand_PHP_Class_Method('quux'); $abstractStaticMethod = new Stagehand_PHP_Class_Method('corge'); $finalMethod = new Stagehand_PHP_Class_Method('grault'); $finalStaticMethod = new Stagehand_PHP_Class_Method('garply'); $referenceMethod = new Stagehand_PHP_Class_Method('waldo'); $protectedMethod->defineProtected(); $protectedMethod->addArgument($argumentA); $protectedMethod->addArgument($argumentB); $protectedMethod->setCode('return true;'); $privateMethod->definePrivate(); $privateMethod->addArgument($argumentC); $privateMethod->addArgument($argumentD); $privateMethod->setCode('$c += 1; return $d;'); $staticMethod->defineStatic(); $staticMethod->addArgument($argumentE); $abstractMethod->defineAbstract(); $abstractStaticMethod->defineAbstract(); $abstractStaticMethod->defineStatic(); $finalMethod->defineFinal(); $finalMethod->addArgument($argumentF); $finalStaticMethod->defineFinal(); $finalStaticMethod->defineStatic(); $referenceMethod->setReference(); $constant1 = new Stagehand_PHP_Class_Constant('A'); $constant2 = new Stagehand_PHP_Class_Constant('B', 10); $constant3 = new Stagehand_PHP_Class_Constant('C', 'text constant'); $class->addProperty($publicProperty); $class->addProperty($protectedProperty); $class->addProperty($privateProperty); $class->addProperty($staticProperty); $class->addMethod($publicMethod); $class->addMethod($protectedMethod); $class->addMethod($privateMethod); $class->addMethod($staticMethod); $class->addMethod($abstractMethod); $class->addMethod($abstractStaticMethod); $class->addMethod($finalMethod); $class->addMethod($finalStaticMethod); $class->addMethod($referenceMethod); $class->addConstant($constant1); $class->addConstant($constant2); $class->addConstant($constant3); return $class; }
/** * @test */ public function renderMethod() { $name = 'getFoo'; $method = new Stagehand_PHP_Class_Method($name); $method->setCode('$a = 0; return 1;'); $this->assertEquals($method->render(), <<<EOF public function getFoo() { \$a = 0; return 1; } EOF ); $foo = new Stagehand_PHP_Class_Method_Argument('foo'); $bar = new Stagehand_PHP_Class_Method_Argument('bar'); $bar->setRequirement(false); $bar->setValue(10); $method->addArgument($foo); $method->addArgument($bar); $this->assertEquals($method->render(), <<<EOF public function getFoo(\$foo, \$bar = 10) { \$a = 0; return 1; } EOF ); $method->defineStatic(); $this->assertEquals($method->render(), <<<EOF public static function getFoo(\$foo, \$bar = 10) { \$a = 0; return 1; } EOF ); $method->defineAbstract(); $this->assertEquals($method->render(), <<<EOF abstract public static function getFoo(\$foo, \$bar = 10); EOF ); $method->defineFinal(); $this->assertEquals($method->render(), <<<EOF final public static function getFoo(\$foo, \$bar = 10) { \$a = 0; return 1; } EOF ); $method->setReference(); $this->assertEquals($method->render(), <<<EOF final public static function &getFoo(\$foo, \$bar = 10) { \$a = 0; return 1; } EOF ); }