public function setUp()
{
$subclassMock1 = $this->getMock('Wealthbot\\AdminBundle\\Entity\\Subclass', array('getId'));
$subclassMock1->expects($this->any())->method('getId')->will($this->returnValue(1));
$subclassMock1->setName('Subclass1');
$subclassMock1->setExpectedPerformance(0.5);
$subclassMock2 = $this->getMock('Wealthbot\\AdminBundle\\Entity\\Subclass', array('getId'));
$subclassMock2->expects($this->any())->method('getId')->will($this->returnValue(2));
$subclassMock2->setName('Subclass2');
$subclassMock2->setExpectedPerformance(0.7);
$entity1 = new CeModelEntity();
$security1 = new Security();
$security1->setExpenseRatio(0.6);
$securityAssignment1 = new SecurityAssignment();
$securityAssignment1->setSecurity($security1);
$securityAssignment1->setSubclass($subclassMock1);
$entity1->setSubclass($subclassMock1);
$entity1->setPercent(40);
$entity1->setSecurityAssignment($securityAssignment1);
$entity2 = new CeModelEntity();
$security2 = new Security();
$security2->setExpenseRatio(0.3);
$securityAssignment2 = new SecurityAssignment();
$securityAssignment2->setSecurity($security2);
$securityAssignment2->setSubclass($subclassMock2);
$entity2->setSubclass($subclassMock2);
$entity2->setPercent(60);
$entity2->setSecurityAssignment($securityAssignment2);
$entity3 = new CeModelEntity();
$security3 = new Security();
$security3->setExpenseRatio(0.7);
$securityAssignment3 = new SecurityAssignment();
$securityAssignment3->setSecurity($security3);
$securityAssignment3->setSubclass($subclassMock1);
$entity3->setSubclass($subclassMock1);
$entity3->setPercent(100);
$entity3->setSecurityAssignment($securityAssignment3);
$entity3->setIsQualified(true);
$model = new CeModel('Model');
$model->addModelEntity($entity1);
$model->addModelEntity($entity2);
$model->addModelEntity($entity3);
$model->setLowMarketReturn(0.6);
$model->setGenerousMarketReturn(1.3);
$model->setForecast(0);
$this->portfolioInformation = new PortfolioInformation();
$this->portfolioInformation->setModel($model);
}
public function testBuildGroupModelEntities()
{
$entity1 = new CeModelEntity();
$entity1->setIsQualified(false);
$entity2 = new CeModelEntity();
$entity2->setIsQualified(true);
$entity3 = new CeModelEntity();
$entity3->setIsQualified(true);
$this->model->addModelEntity($entity1);
$this->model->addModelEntity($entity2);
$this->model->addModelEntity($entity3);
$this->model->buildGroupModelEntities();
$this->assertCount(2, $this->model->getQualifiedModelEntities(), 'Invalid count of qualified model entities.');
$this->assertCount(1, $this->model->getNonQualifiedModelEntities(), 'Invalid count of non qualified model entities.');
$this->model->removeModelEntity($entity1);
$this->model->removeModelEntity($entity2);
$this->assertCount(1, $this->model->getQualifiedModelEntities(), 'Invalid count of qualified model entities after model was removed.');
$this->assertCount(0, $this->model->getNonQualifiedModelEntities(), 'Invalid count of non qualified model entities after model was removed.');
}
private function getCeModelObjects()
{
$adminModel1 = new CeModel();
$adminModel1->setName('Admin model 1');
$adminModel1->setSlug('admin_model_1');
$adminModel1->setRiskRating(1);
$adminModel1->setType(CeModel::TYPE_STRATEGY);
$adminModel2 = new CeModel();
$adminModel2->setName('Admin model 2');
$adminModel2->setSlug('admin_model_2');
$adminModel2->setRiskRating(2);
$adminModel2->setType(CeModel::TYPE_STRATEGY);
$model1 = $this->getMock('Wealthbot\\AdminBundle\\Entity\\CeModel', array('getId'));
$model1->expects($this->any())->method('getId')->will($this->returnValue(31));
$model1->setName('Model1');
$model1->setSlug('model1');
$model1->setOwnerId(1);
$model1->setRiskRating(5);
$model1->setType(CeModel::TYPE_STRATEGY);
$child1 = new CeModel();
$child1->setName('Child model 1');
$child1->setSlug('child_model_1');
$child1->setOwnerId(1);
$child1->setParentId(31);
$model2 = new CeModel();
$model2->setName('Model2');
$model2->setSlug('model2');
$model2->setOwnerId(1);
$model2->setRiskRating(7);
$model3 = new CeModel();
$model3->setName('Model3');
$model3->setSlug('model3');
$model3->setOwnerId(6);
$model3->setRiskRating(2);
$objects = array();
$objects[] = $adminModel1;
$objects[] = $adminModel2;
$objects[] = $model1;
$objects[] = $child1;
$objects[] = $model2;
$objects[] = $model3;
return $objects;
}
/**
* Get isAssumptionLocked
*
* @return boolean
*/
public function getIsAssumptionLocked()
{
return parent::getIsAssumptionLocked();
}
