/** * 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; }
/** * class_statement_3 * method_modifiers function is_reference T_STRING '(' parameter_list ')' method_body */ protected function class_statement_3($params) { if (is_array($params[0])) { $methodModifiers = $params[0]; } else { $methodModifiers = array($params[0]); } $isReference = $params[2] ? true : false; $methodName = $params[3]->getValue(); $parameterList = $params[5]; $methodBody = $params[7] ? $params[7] : null; $method = new Stagehand_PHP_Class_Method($methodName); $method->setCode($methodBody); $lex = $this->getParser()->lex; $docComment = $lex->getLatestDocComment(); if ($docComment) { $method->setDocComment($docComment, true); } if ($isReference) { $method->setReference(); } if ($parameterList) { foreach ($parameterList as $argument) { $method->addArgument($argument); } } $this->_setModifiers($method, $methodModifiers); $this->addCurrentMethod($method); return parent::execute(__FUNCTION__, $params); }
/** * @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 ); }