MAGETWO-50912: Custom Customer Attributes not saving at checkout

Author: Stanislav Idolov

app/code/Magento/Checkout/view/frontend/web/js/model/new-customer-address.js

@@ -6,16 +6,17 @@
 /*global alert*/
 define([], function() {
     /**
-     * @param addressData
+     * @param {Object} addressData
      * Returns new address object
      */
     return function (addressData) {
         var identifier = Date.now();
+
         return {
             email: addressData.email,
             countryId: (addressData.country_id) ? addressData.country_id : window.checkoutConfig.defaultCountryId,
-            regionId: (addressData.region && addressData.region.region_id)
-                ? addressData.region.region_id
+            regionId: (addressData.region && addressData.region.region_id) ?
+                addressData.region.region_id
                 : window.checkoutConfig.defaultRegionId,
             regionCode: (addressData.region) ? addressData.region.region_code : null,
             region: (addressData.region) ? addressData.region.region : null,
@@ -33,25 +34,26 @@ define([], function() {
             suffix: addressData.suffix,
             vatId: addressData.vat_id,
             saveInAddressBook: addressData.save_in_address_book,
-            isDefaultShipping: function() {
+            customAttributes: addressData.custom_attributes,
+            isDefaultShipping: function () {
                 return addressData.default_shipping;
             },
-            isDefaultBilling: function() {
+            isDefaultBilling: function () {
                 return addressData.default_billing;
             },
-            getType: function() {
+            getType: function () {
                 return 'new-customer-address'
             },
-            getKey: function() {
+            getKey: function () {
                 return this.getType();
             },
-            getCacheKey: function() {
+            getCacheKey: function () {
                 return this.getType() + identifier;
             },
-            isEditable: function() {
+            isEditable: function () {
                 return true;
             },
-            canUseForBilling: function() {
+            canUseForBilling: function () {
                 return true;
             }
         }

app/code/Magento/Customer/Model/Address.php

@@ -47,6 +47,11 @@ class Address extends \Magento\Customer\Model\Address\AbstractAddress
      */
     protected $indexerRegistry;
 
+    /**
+     * @var \Magento\Customer\Model\Address\CustomAttributeListInterface
+     */
+    protected $attributeList;
+
     /**
      * @param \Magento\Framework\Model\Context $context
      * @param \Magento\Framework\Registry $registry
@@ -347,4 +352,27 @@ public function reindex()
         $indexer = $this->indexerRegistry->get(Customer::CUSTOMER_GRID_INDEXER_ID);
         $indexer->reindexRow($this->getCustomerId());
     }
+
+    /**
+     * {@inheritdoc}
+     */
+    protected function getCustomAttributesCodes()
+    {
+        return array_keys($this->getAttributeList()->getAttributes());
+    }
+
+    /**
+     * Get new AttributeList dependency for application code.
+     * @return \Magento\Customer\Model\Address\CustomAttributeListInterface
+     * @deprecated
+     */
+    private function getAttributeList()
+    {
+        if (!$this->attributeList) {
+            $this->attributeList = \Magento\Framework\App\ObjectManager::getInstance()->get(
+                \Magento\Customer\Model\Address\CustomAttributeListInterface::class
+            );
+        }
+        return $this->attributeList;
+    }
 }

app/code/Magento/Customer/Model/Address/CustomAttributeList.php

@@ -0,0 +1,17 @@
+<?php
+/**
+ * Copyright © 2016 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Customer\Model\Address;
+
+class CustomAttributeList implements CustomAttributeListInterface
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function getAttributes()
+    {
+        return [];
+    }
+}

app/code/Magento/Customer/Model/Address/CustomAttributeListInterface.php

@@ -0,0 +1,16 @@
+<?php
+/**
+ * Copyright © 2016 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Customer\Model\Address;
+
+interface CustomAttributeListInterface
+{
+    /**
+     * Retrieve list of customer addresses custom attributes
+     *
+     * @return array
+     */
+    public function getAttributes();
+}

app/code/Magento/Customer/etc/di.xml

@@ -43,6 +43,8 @@
                 type="Magento\Customer\Model\Metadata\AddressMetadataManagement" />
     <preference for="Magento\Customer\Api\CustomerManagementInterface"
                 type="Magento\Customer\Model\CustomerManagement" />
+    <preference for="Magento\Customer\Model\Address\CustomAttributeListInterface"
+                type="Magento\Customer\Model\Address\CustomAttributeList" />
     <type name="Magento\Customer\Model\Session">
         <arguments>
             <argument name="configShare" xsi:type="object">Magento\Customer\Model\Config\Share\Proxy</argument>

app/code/Magento/Customer/view/frontend/web/js/model/customer/address.js

@@ -32,6 +32,7 @@ define([], function() {
             vatId: addressData.vat_id,
             sameAsBilling: addressData.same_as_billing,
             saveInAddressBook: addressData.save_in_address_book,
+            customAttributes: addressData.custom_attributes,
             isDefaultShipping: function() {
                 return addressData.default_shipping;
             },