feat(Cart): add product type reference to line item

Closes #280
commercetools · Jan 6, 2017 · e795540 · e795540
1 parent e9cc4a7
commit e795540
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 2 deletions.
diff --git a/src/Model/Cart/LineItem.php b/src/Model/Cart/LineItem.php
@@ -15,6 +15,7 @@
 use Commercetools\Core\Model\TaxCategory\TaxRate;
 use Commercetools\Core\Model\CustomField\CustomFieldObject;
 use Commercetools\Core\Model\Common\TaxedItemPrice;
+use Commercetools\Core\Model\ProductType\ProductTypeReference;
 
 /**
  * @package Commercetools\Core\Model\Cart
@@ -50,6 +51,8 @@
  * @method LineItem setTaxedPrice(TaxedItemPrice $taxedPrice = null)
  * @method string getPriceMode()
  * @method LineItem setPriceMode(string $priceMode = null)
+ * @method ProductTypeReference getProductType()
+ * @method LineItem setProductType(ProductTypeReference $productType = null)
  */
 class LineItem extends JsonObject
 {
@@ -76,7 +79,8 @@ public function fieldDefinitions()
             'discountedPricePerQuantity' => [
                 static::TYPE => '\Commercetools\Core\Model\Cart\DiscountedPricePerQuantityCollection'
             ],
-            'priceMode' => [static::TYPE => 'string']
+            'priceMode' => [static::TYPE => 'string'],
+            'productType' => [static::TYPE => '\Commercetools\Core\Model\ProductType\ProductTypeReference']
         ];
     }
 

diff --git a/tests/fixtures/models.yaml b/tests/fixtures/models.yaml
@@ -45,6 +45,7 @@ lineItem:
     - discountedPricePerQuantity
     - totalPrice
     - priceMode
+    - productType
 
 customLineItem:
   domain: cart

diff --git a/tests/integration/Cart/CartUpdateRequestTest.php b/tests/integration/Cart/CartUpdateRequestTest.php
@@ -142,7 +142,10 @@ public function testLineItem()
         $this->deleteRequest->setVersion($cart->getVersion());
 
         $this->assertSame($product->getId(), $cart->getLineItems()->current()->getProductId());
-
+        $this->assertSame(
+            $product->getProductType()->getId(),
+            $cart->getLineItems()->current()->getProductType()->getId()
+        );
         $request = CartUpdateRequest::ofIdAndVersion($cart->getId(), $cart->getVersion())
             ->addAction(
                 CartChangeLineItemQuantityAction::ofLineItemIdAndQuantity($cart->getLineItems()->current()->getId(), 2)

diff --git a/tests/integration/Order/OrderUpdateRequestTest.php b/tests/integration/Order/OrderUpdateRequestTest.php
@@ -110,6 +110,11 @@ public function testChangeState()
         $cartDraft = $this->getCartDraft();
         $order = $this->createOrder($cartDraft);
 
+        $this->assertSame(
+            $this->getProduct()->getProductType()->getId(),
+            $order->getLineItems()->current()->getProductType()->getId()
+        );
+
         $request = OrderUpdateRequest::ofIdAndVersion($order->getId(), $order->getVersion())
             ->addAction(OrderChangeOrderStateAction::ofOrderState(OrderState::COMPLETE))
         ;