public function testSetIsMuni()
{
$this->lot->setIsMuni(true);
$this->assertTrue($this->lot->getIsMuni());
$this->lot->setIsMuni(0);
$this->assertFalse($this->lot->getIsMuni());
}
public function testLotOrderForMuni()
{
$positions = array(array('client_system_account_id' => 222, 'muni_id' => 78, 'security_id' => 22222), array('client_system_account_id' => 333, 'muni_id' => null, 'security_id' => 33333));
/** @var LotRepository $mockRepository */
$mockRepository = $this->getMockBuilder('Model\\WealthbotRebalancer\\Repository\\LotRepository')->disableOriginalConstructor()->setMethods(array('getPositionsByPortfolio', 'getLastPositionLots'))->getMock();
$mockRepository->expects($this->any())->method('getPositionsByPortfolio')->will($this->returnValue($positions));
$mockRepository->expects($this->any())->method('getLastPositionLots')->will($this->returnCallback(function (Portfolio $portfolio, $securityId, $clientSystemAccountId, $isMuni = false) {
$lotCollection = new LotCollection();
$lot = new Lot();
$lot->setIsMuni($isMuni);
$lotCollection->add($lot);
return $lotCollection;
}));
$mockRepository->expects($this->any())->method('findLotsByAccountAndSecurity')->will($this->returnCallback(function (Portfolio $portfolio, $securityId, $clientSystemAccountId, $isMuni = false) {
$lotCollection = new LotCollection();
$lot = new Lot();
$lot->setIsMuni($isMuni);
$lotCollection->add($lot);
return $lotCollection;
}));
$lotCollection = $mockRepository->findLotsBySubclass(new Portfolio(), new Subclass(), new Account());
$lot1 = $lotCollection->first();
$this->assertFalse($lot1->getIsMuni());
$lot2 = $lotCollection->next();
$this->assertTrue($lot2->getIsMuni());
$lot3 = $lotCollection->next();
$this->assertFalse($lot3->getIsMuni());
//-------------------------------------------------------------------------------------------------------------/
$account = new Account();
$client = new Client();
$client->setPortfolio(new Portfolio());
$account->setClient($client);
$lotCollection = $mockRepository->findLotsByAccountAndSecurity($account, new Security());
$lot1 = $lotCollection->first();
$this->assertFalse($lot1->getIsMuni());
$lot2 = $lotCollection->next();
$this->assertTrue($lot2->getIsMuni());
$lot3 = $lotCollection->next();
$this->assertFalse($lot3->getIsMuni());
}
