public function testAdd()
{
$newObj = new Base();
$this->arrayCollection->add($newObj);
$this->assertEquals($newObj, $this->arrayCollection->get(2));
$newObj->setId(65);
$this->arrayCollection->add($newObj);
$this->assertEquals($newObj, $this->arrayCollection->get(65));
$this->arrayCollection->add($newObj, 500);
$this->assertEquals($newObj, $this->arrayCollection->get(500));
}
public function getTradeDataCollectionForJob(Job $job)
{
$sql = "SELECT\n rq.id as id,\n s.id as security_id,\n sca.id as account_id,\n sca.account_number as account_number,\n st.name as security_type,\n rq.status as action,\n p.quantity as position_quantity,\n s.symbol as symbol,\n SUM(rq.quantity) as quantity,\n SUM(rq.amount) as amount,\n IF(p.quantity = SUM(rq.quantity) OR (rq.status = '" . QueueItem::STATUS_SELL . "' AND rq.lot_id IS NULL), 'AS', 'S') as quantity_type\n FROM " . $this->table . " rq\n LEFT JOIN " . self::TABLE_SECURITY . " s on s.id = rq.security_id\n LEFT JOIN " . self::TABLE_SYSTEM_ACCOUNT . " sca ON sca.id = rq.system_client_account_id\n LEFT JOIN " . self::TABLE_SECURITY_TYPE . " st ON st.id = s.security_type_id\n LEFT JOIN " . self::TABLE_LOT . " l ON l.id = rq.lot_id\n LEFT JOIN " . self::TABLE_POSITION . " p ON p.id = l.position_id\n LEFT JOIN " . self::TABLE_REBALANCER_ACTION . " ra On ra.id = rq.rebalancer_action_id\n WHERE ra.job_id = :jobId\n GROUP BY sca.id, s.id\n ORDER BY rq.id";
$paramaters = array('jobId' => $job->getId());
$results = $this->db->query($sql, $paramaters);
$tradeDataCollection = new ArrayCollection();
foreach ($results as $result) {
$tradeData = new TradeData();
$tradeData->loadFromArray($result);
if ($tradeData->getAction() === TradeData::ACTION_SELL) {
$vsps = $this->findVSPForTradeData($tradeData);
$tradeData->setVsps($vsps);
}
$tradeData->setJobId($job->getId());
$tradeDataCollection->add($tradeData);
}
return $tradeDataCollection;
}
/**
* @param array $data
* @return ArrayCollection
*/
private function getMockTradeDataCollection(array $data = array())
{
$tradeDataCollection = new ArrayCollection();
foreach ($data as $item) {
$tradeData = $this->getMockTradeData($item);
$tradeDataCollection->add($tradeData);
}
return $tradeDataCollection;
}
public function testGetTradeDataCollectionForJob()
{
$riaRepo = new RiaRepository();
$ria = $riaRepo->findOneBy(array('email' => '*****@*****.**'));
$jobRepo = new JobRepository();
$job = $jobRepo->findOneBy(array('user_id' => $ria->getId()));
$tradeDataCollection = $this->repository->getTradeDataCollectionForJob($job);
$this->assertCount(3, $tradeDataCollection);
$securityRepo = new SecurityRepository();
$accountRepo = new AccountRepository();
$tradeDataExpectedArray = array(array('job_id' => $job->getId(), 'security_id' => $securityRepo->findOneBySymbol('RWX')->getId(), 'account_id' => $accountRepo->findOneByAccountNumber('916985328')->getId(), 'account_number' => '916985328', 'account_type' => TradeData::ACCOUNT_TYPE_CASH_ACCOUNT, 'security_type' => TradeData::SECURITY_TYPE_EQUITY, 'action' => TradeData::ACTION_SELL, 'quantity_type' => TradeData::QUANTITY_TYPE_ALL_SHARES, 'quantity' => '56', 'symbol' => 'RWX'), array('job_id' => $job->getId(), 'security_id' => $securityRepo->findOneBySymbol('VCIT')->getId(), 'account_id' => $accountRepo->findOneByAccountNumber('480888811')->getId(), 'account_number' => '480888811', 'account_type' => TradeData::ACCOUNT_TYPE_CASH_ACCOUNT, 'security_type' => TradeData::SECURITY_TYPE_EQUITY, 'action' => TradeData::ACTION_BUY, 'quantity_type' => TradeData::QUANTITY_TYPE_SHARES, 'quantity' => '12', 'symbol' => 'VCIT'), array('job_id' => $job->getId(), 'security_id' => $securityRepo->findOneBySymbol('BND')->getId(), 'account_id' => $accountRepo->findOneByAccountNumber('122223334')->getId(), 'account_number' => '122223334', 'account_type' => TradeData::ACCOUNT_TYPE_CASH_ACCOUNT, 'security_type' => TradeData::SECURITY_TYPE_EQUITY, 'action' => TradeData::ACTION_BUY, 'quantity_type' => TradeData::QUANTITY_TYPE_SHARES, 'quantity' => '1', 'symbol' => 'BND'), array('job_id' => '59', 'security_id' => $securityRepo->findOneBySymbol('VGIT')->getId(), 'account_id' => $accountRepo->findOneByAccountNumber('916985328')->getId(), 'account_number' => '916985328', 'account_type' => TradeData::ACCOUNT_TYPE_CASH_ACCOUNT, 'security_type' => TradeData::SECURITY_TYPE_EQUITY, 'action' => TradeData::ACTION_SELL, 'quantity_type' => TradeData::QUANTITY_TYPE_ALL_SHARES, 'quantity' => '10', 'symbol' => 'VGIT'));
$tradeDataCollectionExpected = new ArrayCollection();
foreach ($tradeDataExpectedArray as $key => $tradeDataExpected) {
$tradeData = new TradeData();
$tradeData->loadFromArray($tradeDataExpected);
$tradeDataCollectionExpected->add($tradeData, $key);
}
foreach ($tradeDataCollection as $tradeData) {
$tradeData->setId(null);
$tradeDataExpected = $tradeDataCollectionExpected->current();
$this->assertEquals($tradeData, $tradeDataExpected);
$tradeDataCollectionExpected->next();
}
}