/**
* @param string $key
* @param ContactAddress $entity
*/
public function fillEntityData($key, $entity)
{
$typeRepository = $this->templateManager->getEntityRepository('Oro\\Bundle\\AddressBundle\\Entity\\AddressType');
$billingType = $typeRepository->getEntity(AddressType::TYPE_BILLING);
$shippingType = $typeRepository->getEntity(AddressType::TYPE_SHIPPING);
switch ($key) {
case 'Jerry Coleman':
$entity->addType($billingType)->addType($shippingType);
break;
case 'John Smith':
$entity->addType($billingType);
break;
case 'John Doo':
$entity->addType($shippingType);
break;
}
parent::fillEntityData($key, $entity);
}
public function testSetAddressType()
{
$contact = new Contact();
$shippingType = new AddressType('shipping');
$addressOne = new ContactAddress();
$addressOne->addType($shippingType);
$contact->addAddress($addressOne);
$addressTwo = new ContactAddress();
$contact->addAddress($addressTwo);
$contact->setAddressType($addressTwo, $shippingType);
$this->assertFalse($addressOne->hasTypeWithName('shipping'));
$this->assertTrue($addressTwo->hasTypeWithName('shipping'));
}
