GraphQL-57: Manage Address Book

Author: Valeriy Nayda

app/code/Magento/CustomerGraphQl/Model/Resolver/CreateCustomerAddress.php

@@ -14,6 +14,8 @@
 use Magento\CustomerGraphQl\Model\Customer\Address\CustomerAddressDataProvider;
 use Magento\CustomerGraphQl\Model\Customer\CheckCustomerAccount;
 use Magento\Framework\Api\DataObjectHelper;
+use Magento\Framework\Exception\InputException;
+use Magento\Framework\GraphQl\Exception\GraphQlInputException;
 use Magento\Framework\GraphQl\Schema\Type\ResolveInfo;
 use Magento\Framework\GraphQl\Config\Element\Field;
 use Magento\Framework\GraphQl\Query\ResolverInterface;
@@ -103,13 +105,20 @@ public function resolve(
      * @param int $customerId
      * @param array $addressData
      * @return AddressInterface
+     * @throws GraphQlInputException
      */
     private function createCustomerAddress(int $customerId, array $addressData) : AddressInterface
     {
         /** @var AddressInterface $address */
         $address = $this->addressInterfaceFactory->create();
         $this->dataObjectHelper->populateWithArray($address, $addressData, AddressInterface::class);
         $address->setCustomerId($customerId);
-        return $this->addressRepository->save($address);
+
+        try {
+            $address = $this->addressRepository->save($address);
+        } catch (InputException $e) {
+            throw new GraphQlInputException(__($e->getMessage()), $e);
+        }
+        return $address;
     }
 }

dev/tests/api-functional/testsuite/Magento/GraphQl/Customer/CreateCustomerAddressTest.php

@@ -34,7 +34,7 @@ protected function setUp()
     }
 
     /**
-     * @magentoApiDataFixture Magento/Customer/_files/customer.php
+     * @magentoApiDataFixture Magento/Customer/_files/customer_without_addresses.php
      * @SuppressWarnings(PHPMD.ExcessiveMethodLength)
      */
     public function testCreateCustomerAddress()
@@ -165,7 +165,7 @@ public function testCreateCustomerAddressIfUserIsNotAuthorized()
      * Verify customers with valid credentials create new address
      * with missing required Firstname attribute
      *
-     * @magentoApiDataFixture Magento/Customer/_files/customer.php
+     * @magentoApiDataFixture Magento/Customer/_files/customer_without_addresses.php
      * @expectedException \Exception
      * @expectedExceptionMessage Required parameters are missing: firstname
      */

dev/tests/integration/testsuite/Magento/Customer/_files/customer_without_addresses.php

@@ -0,0 +1,31 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+use Magento\Customer\Model\Customer;
+use Magento\Customer\Model\CustomerRegistry;
+use Magento\TestFramework\Helper\Bootstrap;
+
+/** @var CustomerRegistry $customerRegistry */
+$customerRegistry = Bootstrap::getObjectManager()->get(CustomerRegistry::class);
+/** @var Customer $customer */
+$customer = Bootstrap::getObjectManager()->get(Customer::class);
+
+$customer->setWebsiteId(1)
+    ->setId(1)
+    ->setEmail('customer@example.com')
+    ->setPassword('password')
+    ->setGroupId(1)
+    ->setStoreId(1)
+    ->setIsActive(1)
+    ->setPrefix('Mr.')
+    ->setFirstname('John')
+    ->setMiddlename('A')
+    ->setLastname('Smith')
+    ->setSuffix('Esq.')
+    ->setGender(0)
+    ->save();
+$customerRegistry->remove($customer->getId());

dev/tests/integration/testsuite/Magento/Customer/_files/customer_without_addresses_rollback.php

@@ -0,0 +1,34 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+use Magento\Customer\Api\CustomerRepositoryInterface;
+use Magento\Customer\Model\CustomerRegistry;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\Framework\Registry;
+use Magento\TestFramework\Helper\Bootstrap;
+
+/** @var CustomerRegistry $customerRegistry */
+$customerRegistry = Bootstrap::getObjectManager()->get(CustomerRegistry::class);
+/** @var CustomerRepositoryInterface $customerRepository */
+$customerRepository = Bootstrap::getObjectManager()->get(CustomerRepositoryInterface::class);
+
+/** @var Registry $registry */
+$registry = Bootstrap::getObjectManager()->get(Registry::class);
+$registry->unregister('isSecureArea');
+$registry->register('isSecureArea', true);
+
+try {
+    $customerRepository->deleteById(1);
+} catch (NoSuchEntityException $e) {
+    /**
+     * Tests which are wrapped with MySQL transaction clear all data by transaction rollback.
+     */
+}
+$customerRegistry->remove(1);
+
+$registry->unregister('isSecureArea');
+$registry->register('isSecureArea', false);