public function testSetProductUnit() { $productUnit = (new ProductUnit())->setCode('rfp-unit-code'); $requestProductItem = new RequestProductItem(); $this->assertNull($requestProductItem->getProductUnitCode()); $requestProductItem->setProductUnit($productUnit); $this->assertEquals($productUnit->getCode(), $requestProductItem->getProductUnitCode()); }
/** * @param int $productId * @param int $quantity * @param string $unitCode * @param OptionalPrice $price * @return RequestProductItem */ protected function getRequestProductItem($productId, $quantity = null, $unitCode = null, OptionalPrice $price = null) { $requestProductItem = new RequestProductItem(); $requestProductItem->setRequestProduct($this->getRequestProduct($productId)); if (null !== $quantity) { $requestProductItem->setQuantity($quantity); } if (null !== $unitCode) { $requestProductItem->setProductUnit($this->getEntity('OroB2B\\Bundle\\ProductBundle\\Entity\\ProductUnit', $unitCode, 'code')); } if (null !== $price) { $requestProductItem->setPrice($price); } return $requestProductItem; }