ACP2E-873 Allowed Country per website issue

Author: rajan-adobe

app/code/Magento/Customer/Model/ResourceModel/AddressRepository.php

@@ -126,7 +126,9 @@ public function save(\Magento\Customer\Api\Data\AddressInterface $address)
         } else {
             $addressModel->updateData($address);
         }
-        $addressModel->setStoreId($customerModel->getStoreId());
+        if ($customerModel->getSharingConfig()->isWebsiteScope()) {
+            $addressModel->setStoreId($customerModel->getStoreId());
+        }
 
         $errors = $addressModel->validate();
         if ($errors !== true) {