MC-41550: product URL rewrite for specific website not generated after mass update

Author: victor-v-rad

dev/tests/integration/testsuite/Magento/CatalogUrlRewrite/Plugin/Catalog/Model/Product/UpdateProductWebsiteUrlRewritesTest.php

@@ -8,6 +8,7 @@
 namespace Magento\CatalogUrlRewrite\Plugin\Catalog\Model\Product;
 
 use Magento\Catalog\Api\Data\ProductInterface;
+use Magento\Catalog\Model\ResourceModel\Product\Collection as ProductCollection;
 use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Catalog\Model\Product\Action;
 use Magento\Store\Api\StoreWebsiteRelationInterface;
@@ -69,4 +70,49 @@ public function testUpdateUrlRewrites()
             $url
         );
     }
+
+    /**
+     * @magentoDataFixture Magento/Catalog/_files/product_simple_with_url_key.php
+     * @magentoDataFixture Magento/Store/_files/second_website_with_store_group_and_store.php
+     */
+    public function testUpdateUrlRewritesForSecondProduct()
+    {
+        /** @var Website $website */
+        $websiteRepository = Bootstrap::getObjectManager()->get(WebsiteRepository::class);
+        $website = $websiteRepository->get('test');
+        $productRepository = Bootstrap::getObjectManager()->get(ProductRepositoryInterface::class);
+        // test the first product
+        $product =  $productRepository->get('simple1', false, null, true);
+        // store filter condition about the first product in collection
+        $productCollection = Bootstrap::getObjectManager()->get(ProductCollection::class);
+        $productCollection->addFieldToFilter('entity_id', $product->getId());
+        $this->action->updateWebsites([$product->getId()], [$website->getId()], 'add');
+        $storeIds = $this->storeWebsiteRelation->getStoreByWebsiteId($website->getId());
+        $this->assertStringContainsString(
+            $product->getUrlKey() . '.html',
+            $product->setStoreId(reset($storeIds))->getProductUrl()
+        );
+        $this->action->updateWebsites([$product->getId()], [$website->getId()], 'remove');
+        $product->setRequestPath('');
+        $url = $product->setStoreId(reset($storeIds))->getProductUrl();
+        $this->assertStringNotContainsString(
+            $product->getUrlKey() . '.htmll',
+            $url
+        );
+        // test the second product
+        $product =  $productRepository->get('simple2', false, null, true);
+        $this->action->updateWebsites([$product->getId()], [$website->getId()], 'add');
+        $storeIds = $this->storeWebsiteRelation->getStoreByWebsiteId($website->getId());
+        $this->assertStringContainsString(
+            $product->getUrlKey() . '.html',
+            $product->setStoreId(reset($storeIds))->getProductUrl()
+        );
+        $this->action->updateWebsites([$product->getId()], [$website->getId()], 'remove');
+        $product->setRequestPath('');
+        $url = $product->setStoreId(reset($storeIds))->getProductUrl();
+        $this->assertStringNotContainsString(
+            $product->getUrlKey() . '.html',
+            $url
+        );
+    }
 }