MC-42970: [Magento Cloud][Premier Support] Graphql request can add product to cart that is NOT assigned to a website of a multi site store

Author: dhorytskyi

app/code/Magento/QuoteGraphQl/Model/Cart/AddSimpleProductToCart.php

@@ -8,8 +8,7 @@
 namespace Magento\QuoteGraphQl\Model\Cart;
 
 use Exception;
-use Magento\Catalog\Api\Data\ProductInterface;
-use Magento\Catalog\Model\ResourceModel\Product\CollectionFactory as ProductCollectionFactory;
+use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Framework\Exception\NoSuchEntityException;
 use Magento\Framework\GraphQl\Exception\GraphQlInputException;
 use Magento\Framework\GraphQl\Exception\GraphQlNoSuchEntityException;
@@ -22,24 +21,24 @@
 class AddSimpleProductToCart
 {
     /**
-     * @var ProductCollectionFactory
+     * @var ProductRepositoryInterface
      */
-    private $productCollectionFactory;
+    private $productRepository;
 
     /**
      * @var BuyRequestBuilder
      */
     private $buyRequestBuilder;
 
     /**
-     * @param ProductCollectionFactory $productCollectionFactory
+     * @param ProductRepositoryInterface $productRepository
      * @param BuyRequestBuilder $buyRequestBuilder
      */
     public function __construct(
-        ProductCollectionFactory $productCollectionFactory,
-        BuyRequestBuilder        $buyRequestBuilder
+        ProductRepositoryInterface $productRepository,
+        BuyRequestBuilder $buyRequestBuilder
     ) {
-        $this->productCollectionFactory = $productCollectionFactory;
+        $this->productRepository = $productRepository;
         $this->buyRequestBuilder = $buyRequestBuilder;
     }
 
@@ -56,13 +55,12 @@ public function execute(Quote $cart, array $cartItemData): void
         $cartItemData['model'] = $cart;
         $sku = $this->extractSku($cartItemData);
 
-        $productCollection = $this->productCollectionFactory->create()
-            ->addAttributeToFilter(ProductInterface::SKU, $sku)
-            ->addWebsiteFilter([$cart->getStore()->getWebsiteId()])
-            ->load();
-        /** @var ProductInterface $product */
-        $product = $productCollection->getFirstItem();
-        if (!$product->getId()) {
+        try {
+            $product = $this->productRepository->get($sku, false, null, true);
+        } catch (NoSuchEntityException $e) {
+            throw new GraphQlNoSuchEntityException(__('Could not find a product with SKU "%sku"', ['sku' => $sku]));
+        }
+        if (!in_array($cart->getStore()->getWebsiteId(), $product->getWebsiteIds())) {
             throw new GraphQlNoSuchEntityException(__('Could not find a product with SKU "%sku"', ['sku' => $sku]));
         }