From 008702fb0067c93226e1607319002bbcb2259a71 Mon Sep 17 00:00:00 2001
From: Jens Schulze <jens.schulze@commercetools.de>
Date: Wed, 28 Oct 2015 14:55:37 +0100
Subject: [PATCH] feat(ImportOrder): add custom fields to ImportOrder

---
 src/Model/Common/PriceDraftCollection.php | 2 +-
 src/Model/Order/ImportOrder.php           | 4 ++++
 src/Model/Order/LineItemImportDraft.php   | 4 ++++
 tests/fixtures/models.yaml                | 2 ++
 4 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/Model/Common/PriceDraftCollection.php b/src/Model/Common/PriceDraftCollection.php
index f39bf7611e..aa933a09a0 100644
--- a/src/Model/Common/PriceDraftCollection.php
+++ b/src/Model/Common/PriceDraftCollection.php
@@ -8,9 +8,9 @@
 /**
  * @package Commercetools\Core\Model\Common
  *
- * @method PriceDraft getAt($offset)
  * @method PriceDraftCollection add(PriceDraft $element)
  * @method PriceDraft current()
+ * @method PriceDraft getAt($offset)
  */
 class PriceDraftCollection extends Collection
 {
diff --git a/src/Model/Order/ImportOrder.php b/src/Model/Order/ImportOrder.php
index a48ab613d9..945a462a88 100644
--- a/src/Model/Order/ImportOrder.php
+++ b/src/Model/Order/ImportOrder.php
@@ -13,6 +13,7 @@
 use Commercetools\Core\Model\CustomerGroup\CustomerGroupReference;
 use Commercetools\Core\Model\Cart\ShippingInfo;
 use Commercetools\Core\Model\Common\DateTimeDecorator;
+use Commercetools\Core\Model\CustomField\CustomFieldObjectDraft;
 
 /**
  * @package Commercetools\Core\Model\Order
@@ -48,6 +49,8 @@
  * @method ImportOrder setShippingInfo(ShippingInfo $shippingInfo = null)
  * @method DateTimeDecorator getCompletedAt()
  * @method ImportOrder setCompletedAt(\DateTime $completedAt = null)
+ * @method CustomFieldObjectDraft getCustom()
+ * @method ImportOrder setCustom(CustomFieldObjectDraft $custom = null)
  */
 class ImportOrder extends JsonObject
 {
@@ -73,6 +76,7 @@ public function fieldDefinitions()
                 static::TYPE => '\DateTime',
                 static::DECORATOR => '\Commercetools\Core\Model\Common\DateTimeDecorator'
             ],
+            'custom' => [static::TYPE => '\Commercetools\Core\Model\CustomField\CustomFieldObjectDraft']
         ];
     }
 }
diff --git a/src/Model/Order/LineItemImportDraft.php b/src/Model/Order/LineItemImportDraft.php
index ff5d72601c..4ffd093873 100644
--- a/src/Model/Order/LineItemImportDraft.php
+++ b/src/Model/Order/LineItemImportDraft.php
@@ -10,6 +10,7 @@
 use Commercetools\Core\Model\Common\Price;
 use Commercetools\Core\Model\Channel\ChannelReference;
 use Commercetools\Core\Model\TaxCategory\TaxRate;
+use Commercetools\Core\Model\CustomField\CustomFieldObjectDraft;
 
 /**
  * @package Commercetools\Core\Model\Order
@@ -29,6 +30,8 @@
  * @method LineItemImportDraft setSupplyChannel(ChannelReference $supplyChannel = null)
  * @method TaxRate getTaxRate()
  * @method LineItemImportDraft setTaxRate(TaxRate $taxRate = null)
+ * @method CustomFieldObjectDraft getCustom()
+ * @method LineItemImportDraft setCustom(CustomFieldObjectDraft $custom = null)
  */
 class LineItemImportDraft extends JsonObject
 {
@@ -43,6 +46,7 @@ public function fieldDefinitions()
             'state' => [static::TYPE => '\Commercetools\Core\Model\Order\ItemStateCollection'],
             'supplyChannel' => [static::TYPE => '\Commercetools\Core\Model\Channel\ChannelReference'],
             'taxRate' => [static::TYPE => '\Commercetools\Core\Model\TaxCategory\TaxRate'],
+            'custom' => [static::TYPE => '\Commercetools\Core\Model\CustomField\CustomFieldObjectDraft']
         ];
     }
 }
diff --git a/tests/fixtures/models.yaml b/tests/fixtures/models.yaml
index f54b1b7a1c..ea3574b319 100644
--- a/tests/fixtures/models.yaml
+++ b/tests/fixtures/models.yaml
@@ -823,6 +823,7 @@ importOrder:
     - paymentState
     - shippingInfo
     - completedAt
+    - custom
 
 lineItemImportDraft:
   domain: order
@@ -836,6 +837,7 @@ lineItemImportDraft:
     - state
     - supplyChannel
     - taxRate
+    - custom
 
 productVariantImportDraft:
   domain: order