MC-20684: Admin: Add/remove product from other storeviews and websites

Author: ysapiga

dev/tests/integration/testsuite/Magento/Catalog/Model/Product/UpdateProductWebsiteTest.php

@@ -9,7 +9,7 @@
 
 use Magento\Catalog\Api\Data\ProductInterface;
 use Magento\Catalog\Api\ProductRepositoryInterface;
-use Magento\Catalog\Model\ResourceModel\Product\WebsiteFactory;
+use Magento\Catalog\Model\ResourceModel\Product\Website\Link;
 use Magento\Framework\Exception\NoSuchEntityException;
 use Magento\Framework\ObjectManagerInterface;
 use Magento\Store\Api\WebsiteRepositoryInterface;
@@ -26,8 +26,8 @@ class UpdateProductWebsiteTest extends TestCase
     /** @var ObjectManagerInterface */
     private $objectManager;
 
-    /** @var  WebsiteFactory */
-    private $websiteProductsResourceFactory;
+    /** @var  Link */
+    private $productWebsiteLink;
 
     /** @var WebsiteRepositoryInterface */
     private $websiteRepository;
@@ -43,7 +43,7 @@ protected function setUp()
         parent::setUp();
 
         $this->objectManager = Bootstrap::getObjectManager();
-        $this->websiteProductsResourceFactory = $this->objectManager->get(WebsiteFactory::class);
+        $this->productWebsiteLink = $this->objectManager->get(Link::class);
         $this->websiteRepository = $this->objectManager->get(WebsiteRepositoryInterface::class);
         $this->productRepository = $this->objectManager->get(ProductRepositoryInterface::class);
     }
@@ -58,7 +58,10 @@ public function testAssignProductToWebsite(): void
         $defaultWebsiteId = $this->websiteRepository->get('base')->getId();
         $secondWebsiteId = $this->websiteRepository->get('test')->getId();
         $product = $this->updateProductWebsites('simple2', [$defaultWebsiteId, $secondWebsiteId]);
-        $this->assertProductWebsites((int)$product->getId(), [$defaultWebsiteId, $secondWebsiteId]);
+        $this->assertEquals(
+            [$defaultWebsiteId, $secondWebsiteId],
+            $this->productWebsiteLink->getWebsiteIdsByProductId($product->getId())
+        );
     }
 
     /**
@@ -68,11 +71,9 @@ public function testAssignProductToWebsite(): void
      */
     public function testUnassignProductFromWebsite(): void
     {
-        $product = $this->productRepository->get('simple-on-two-websites');
         $secondWebsiteId = $this->websiteRepository->get('test')->getId();
-        $product->setWebsiteIds([$secondWebsiteId]);
-        $product = $this->productRepository->save($product);
-        $this->assertProductWebsites((int)$product->getId(), [$secondWebsiteId]);
+        $product = $this->updateProductWebsites('simple-on-two-websites', [$secondWebsiteId]);
+        $this->assertEquals([$secondWebsiteId], $this->productWebsiteLink->getWebsiteIdsByProductId($product->getId()));
     }
 
     /**
@@ -102,17 +103,4 @@ private function updateProductWebsites(string $productSku, array $websiteIds): P
 
         return $this->productRepository->save($product);
     }
-
-    /**
-     * Assert that websites attribute was correctly saved
-     *
-     * @param int $productId
-     * @param array $expectedData
-     * @return void
-     */
-    private function assertProductWebsites(int $productId, array $expectedData): void
-    {
-        $websiteResource = $this->websiteProductsResourceFactory->create();
-        $this->assertEquals($expectedData, $websiteResource->getWebsites([$productId])[$productId]);
-    }
 }