public function addTestIdMethod(ClassGenerator $generator, State $state)
{
$method = new MethodGenerator('testGetSetId');
$class = $state->getEntityModel()->getClassName();
$code = <<<EOF
\$object = new {$class}();
\$object->setId(123);
\$this->assertEquals(123, \$object->getId());
EOF;
$method->setBody($code);
$generator->addMethodFromGenerator($method);
}
protected function buildEntityManager(ClassGenerator $generator)
{
$setter = $this->getSetter('entityManager', 'EntityManager');
$getter = $this->getLazyGetter('entityManager', 'EntityManager', '$this->getServiceLocator()->get(\'entity_manager\')');
$generator->addMethodFromGenerator($setter);
$generator->addMethodFromGenerator($getter);
}
/**
* @param ClassGenerator $generator
* @param State $state
*/
public function addGetEntityManager(ClassGenerator $generator, State $state)
{
$doc = "@return \\PHPUnit_Framework_MockObject_MockObject|EntityManager";
$body = <<<EOF
return \$this->getMockBuilder('Doctrine\\ORM\\EntityManager')
->disableOriginalConstructor()
->getMock();
EOF;
$method = new MethodGenerator('getEntityManager', [], MethodGenerator::FLAG_PUBLIC, $body, $doc);
$generator->addMethodFromGenerator($method);
}
public function buildEditAction(ClassGenerator $generator, State $state)
{
$service = 'get' . $state->getServiceModel()->getClassName();
$name = lcfirst($state->getEntityModel()->getClassName());
$method = new MethodGenerator('editAction');
$method->setDocBlock(new DocBlockGenerator('Show one entity'));
$method->setBody(<<<EOF
\$id = \$this->params()->fromRoute('id');
\${$name} = \$this->{$service}()->loadById(\$id);
/** @var Form \$form */
\$form = \$this->getServiceLocator()->get('{$state->getFormFactoryModel()->getServiceName()}');
\$form->bind(\${$name});
if (\$this->getRequest()->isPost()) {
\$form->setData(\$this->params()->fromPost());
if (\$form->isValid()) {
\$this->{$service}()->save(\${$name});
\$this->getEntityManager()->flush();
\$this->flashMessenger()->addSuccessMessage('Saved');
\$this->redirect()->toRoute('home');
}
}
return array(
'form' => \$form
);
EOF
);
$generator->addMethodFromGenerator($method);
}
