/**
* @param string $key
* @param B2bCustomer $entity
*/
public function fillEntityData($key, $entity)
{
$addressRepo = $this->templateManager->getEntityRepository('Oro\\Bundle\\AddressBundle\\Entity\\Address');
$userRepo = $this->templateManager->getEntityRepository('Oro\\Bundle\\UserBundle\\Entity\\User');
$contactRepo = $this->templateManager->getEntityRepository('OroCRM\\Bundle\\ContactBundle\\Entity\\Contact');
$leadRepo = $this->templateManager->getEntityRepository('OroCRM\\Bundle\\SalesBundle\\Entity\\Lead');
$accountRepo = $this->templateManager->getEntityRepository('OroCRM\\Bundle\\AccountBundle\\Entity\\Account');
$channelRepo = $this->templateManager->getEntityRepository('OroCRM\\Bundle\\ChannelBundle\\Entity\\Channel');
$opportunityRepo = $this->templateManager->getEntityRepository('OroCRM\\Bundle\\SalesBundle\\Entity\\Opportunity');
switch ($key) {
case 'Jerry Coleman':
$entity->setName('Jerry Coleman');
$entity->addLead($leadRepo->getEntity('Jerry Coleman'));
$entity->setContact($contactRepo->getEntity('Jerry Coleman'));
$entity->setAccount($accountRepo->getEntity('Coleman'));
$entity->setOwner($userRepo->getEntity('John Doo'));
$entity->setBillingAddress($addressRepo->getEntity('Jerry Coleman'));
$entity->setShippingAddress($addressRepo->getEntity('Jerry Coleman'));
$entity->setDataChannel($channelRepo->getEntity('B2B channel|b2b'));
$entity->setCreatedAt(new \DateTime());
$entity->setUpdatedAt(new \DateTime());
$entity->addOpportunity($opportunityRepo->getEntity('Jerry Coleman'));
return;
}
parent::fillEntityData($key, $entity);
}
public function testLeadsInteraction()
{
$result = $this->entity->getLeads();
$this->assertInstanceOf('Doctrine\\Common\\Collections\\ArrayCollection', $result);
$this->assertCount(0, $result);
$lead = $this->getMock('OroCRM\\Bundle\\SalesBundle\\Entity\\Lead');
$this->entity->addLead($lead);
$this->assertCount(1, $this->entity->getLeads());
$this->assertTrue($this->entity->getLeads()->contains($lead));
$this->entity->removeLead($lead);
$result = $this->entity->getLeads();
$this->assertCount(0, $result);
$newCollection = new ArrayCollection();
$this->entity->setLeads($newCollection);
$this->assertNotSame($result, $this->entity->getLeads());
$this->assertSame($newCollection, $this->entity->getLeads());
}
/**
* Append leads to B2bCustomer
*
* @param B2bCustomer $b2bCustomer
* @param Lead[] $leads
*/
protected function appendLeads(B2bCustomer $b2bCustomer, array $leads)
{
foreach ($leads as $lead) {
$b2bCustomer->addLead($lead);
}
}
