/** * @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'); }