magento/graphql-ce:#678 Send email to friend

Author: lenaorobei

app/code/Magento/SendFriendGraphQl/Model/Provider/GetVisibleProduct.php

@@ -5,7 +5,6 @@
  */
 declare(strict_types=1);
 
-
 namespace Magento\SendFriendGraphQl\Model\Provider;
 
 use Magento\Catalog\Api\ProductRepositoryInterface;
@@ -15,7 +14,7 @@
 use Magento\Framework\Exception\NoSuchEntityException;
 
 /**
- * Class GetProduct
+ * Returns product if it is visible in catalog.
  */
 class GetVisibleProduct
 {

app/code/Magento/SendFriendGraphQl/Model/SendFriend/SendEmail.php

@@ -16,6 +16,7 @@
 use Magento\Framework\GraphQl\Exception\GraphQlNoSuchEntityException;
 use Magento\SendFriend\Model\SendFriend;
 use Magento\SendFriend\Model\SendFriendFactory;
+use Magento\SendFriendGraphQl\Model\Provider\GetVisibleProduct;
 
 /**
  * Send Product Email to Friend(s)
@@ -43,21 +44,30 @@ class SendEmail
     private $eventManager;
 
     /**
+     * @var GetVisibleProduct
+     */
+    private $visibleProductProvider;
+
+    /**
+     * SendEmail constructor.
      * @param DataObjectFactory $dataObjectFactory
      * @param ProductRepositoryInterface $productRepository
      * @param SendFriendFactory $sendFriendFactory
      * @param ManagerInterface $eventManager
+     * @param GetVisibleProduct $visibleProductProvider
      */
     public function __construct(
         DataObjectFactory $dataObjectFactory,
         ProductRepositoryInterface $productRepository,
         SendFriendFactory $sendFriendFactory,
-        ManagerInterface $eventManager
+        ManagerInterface $eventManager,
+        GetVisibleProduct $visibleProductProvider
     ) {
         $this->dataObjectFactory = $dataObjectFactory;
         $this->productRepository = $productRepository;
         $this->sendFriendFactory = $sendFriendFactory;
         $this->eventManager = $eventManager;
+        $this->visibleProductProvider = $visibleProductProvider;
     }
 
     /**
@@ -81,7 +91,7 @@ public function execute(int $productId, array $senderData, array $recipientsData
             );
         }
 
-        $product = $this->getProduct($productId);
+        $product = $this->visibleProductProvider->execute($productId);
 
         $this->eventManager->dispatch('sendfriend_product', ['product' => $product]);
 
@@ -117,26 +127,4 @@ private function validateSendFriendModel(SendFriend $sendFriend, array $senderDa
             throw new GraphQlInputException(__(implode($validationResult)));
         }
     }
-
-    /**
-     * Get product
-     *
-     * @param int $productId
-     * @return ProductInterface
-     * @throws GraphQlNoSuchEntityException
-     */
-    private function getProduct(int $productId): ProductInterface
-    {
-        try {
-            $product = $this->productRepository->getById($productId);
-            if (!$product->isVisibleInCatalog()) {
-                throw new GraphQlNoSuchEntityException(
-                    __("The product that was requested doesn't exist. Verify the product and try again.")
-                );
-            }
-        } catch (NoSuchEntityException $e) {
-            throw new GraphQlNoSuchEntityException(__($e->getMessage()), $e);
-        }
-        return $product;
-    }
 }

dev/tests/api-functional/testsuite/Magento/GraphQl/SendFriend/SendFriendTest.php

@@ -125,7 +125,6 @@ public function testSendWithoutExistProduct()
               }';
         $query = $this->getQuery($productId, $recipients);
 
-        $this->expectException(\Exception::class);
         $this->expectExceptionMessage(
             'The product that was requested doesn\'t exist. Verify the product and try again.'
         );
@@ -276,9 +275,10 @@ public function testSendProductWithoutVisibility()
                   email:"[email protected]"
               }';
         $query = $this->getQuery($productId, $recipients);
-
-        $response = $this->graphQlMutation($query, [], '', $this->getHeaderMap());
-        $this->assertResponse($response);
+        $this->expectExceptionMessage(
+            'The product that was requested doesn\'t exist. Verify the product and try again.'
+        );
+        $this->graphQlMutation($query, [], '', $this->getHeaderMap());
     }
 
     /**