/**
* @param ObjectManager $manager
* @param string $productReference
* @param string $unitReference
* @param int $precision
* @return ProductUnitPrecision
*/
protected function createProductUnitPrecision(ObjectManager $manager, $productReference, $unitReference, $precision = 0)
{
/** @var Product $productReference */
$product = $this->getReference($productReference);
/** @var ProductUnit $unitReference */
$unit = $this->getReference($unitReference);
$productUnitPrecision = new ProductUnitPrecision();
$productUnitPrecision->setProduct($product);
$productUnitPrecision->setPrecision($precision);
$productUnitPrecision->setUnit($unit);
$manager->persist($productUnitPrecision);
$this->addReference(sprintf('product_unit_precision.%s', implode('.', [$product->getSku(), $unit->getCode()])), $productUnitPrecision);
return $productUnitPrecision;
}
public function testPostRemove()
{
$entity = new ProductUnitPrecision();
$product = new Product();
$unit = new ProductUnit();
$entity->setProduct($product)->setUnit($unit);
$event = $this->getMockBuilder('Doctrine\\ORM\\Event\\LifecycleEventArgs')->disableOriginalConstructor()->getMock();
$event->expects($this->once())->method('getEntity')->will($this->returnValue($entity));
$repository = $this->getMockBuilder('OroB2B\\Bundle\\PricingBundle\\Entity\\Repository\\ProductPriceRepository')->disableOriginalConstructor()->getMock();
$repository->expects($this->once())->method('deleteByProductUnit')->with($product, $unit);
$em = $this->getMockBuilder('Doctrine\\ORM\\EntityManager')->disableOriginalConstructor()->getMock();
$em->expects($this->once())->method('getRepository')->with($this->productPriceClass)->will($this->returnValue($repository));
$event->expects($this->once())->method('getEntityManager')->will($this->returnValue($em));
$this->eventDispatcher->expects($this->exactly(2))->method('dispatch');
$this->eventDispatcher->expects($this->at(0))->method('dispatch')->with(ProductPricesRemoveBefore::NAME, $this->isInstanceOf('OroB2B\\Bundle\\PricingBundle\\Event\\ProductPricesRemoveBefore'));
$this->eventDispatcher->expects($this->at(1))->method('dispatch')->with(ProductPricesRemoveAfter::NAME, $this->isInstanceOf('OroB2B\\Bundle\\PricingBundle\\Event\\ProductPricesRemoveAfter'));
$this->listener->postRemove($event);
}
/**
* {@inheritdoc}
*/
public function load(ObjectManager $manager)
{
$locator = $this->container->get('file_locator');
$filePath = $locator->locate('@OroB2BProductBundle/Migrations/Data/Demo/ORM/data/product_unit_precisions.csv');
if (is_array($filePath)) {
$filePath = current($filePath);
}
$handler = fopen($filePath, 'r');
$headers = fgetcsv($handler, 1000, ',');
while (($data = fgetcsv($handler, 1000, ',')) !== false) {
$row = array_combine($headers, array_values($data));
$product = $this->getProductBySku($manager, $row['productCode']);
$productUnit = $this->getProductUnit($manager, $row['unitCode']);
$productUnitPrecision = new ProductUnitPrecision();
$productUnitPrecision->setProduct($product)->setUnit($productUnit)->setPrecision($row['precision']);
$manager->persist($productUnitPrecision);
}
fclose($handler);
$manager->flush();
}
/**
* Add unitPrecisions
*
* @param ProductUnitPrecision $unitPrecision
* @return Product
*/
public function addUnitPrecision(ProductUnitPrecision $unitPrecision)
{
$existingUnitPrecision = $this->getUnitPrecision($unitPrecision->getUnit()->getCode());
if ($existingUnitPrecision) {
$existingUnitPrecision->setPrecision($unitPrecision->getPrecision());
} else {
$unitPrecision->setProduct($this);
$this->unitPrecisions->add($unitPrecision);
}
return $this;
}
/**
* @param boolean $withProductUnitPrecision
* @return Product
*/
protected function createExpectedProductEntity($withProductUnitPrecision = false)
{
$expectedProduct = new StubProduct();
$productUnit = new ProductUnit();
$productUnit->setCode('kg');
if ($withProductUnitPrecision) {
$productUnitPrecision = new ProductUnitPrecision();
$productUnitPrecision->setProduct($expectedProduct)->setUnit($productUnit)->setPrecision(3);
$expectedProduct->addUnitPrecision($productUnitPrecision);
}
return $expectedProduct->setSku('test sku');
}
/**
* @param boolean $withProductUnitPrecision
* @param boolean $withNamesAndDescriptions
* @return Product
*/
protected function createExpectedProductEntity($withProductUnitPrecision = false, $withNamesAndDescriptions = false)
{
$expectedProduct = new StubProduct();
$productUnit = new ProductUnit();
$productUnit->setCode('kg');
if ($withProductUnitPrecision) {
$productUnitPrecision = new ProductUnitPrecision();
$productUnitPrecision->setProduct($expectedProduct)->setUnit($productUnit)->setPrecision(3);
$expectedProduct->addUnitPrecision($productUnitPrecision);
}
if ($withNamesAndDescriptions) {
$expectedProduct->addName($this->createLocalizedValue('first name'))->addName($this->createLocalizedValue('second name'))->addDescription($this->createLocalizedValue(null, 'first description'))->addDescription($this->createLocalizedValue(null, 'second description'));
}
return $expectedProduct->setSku('test sku');
}
