MC-17015: [GraphQL] Product Categories query doesn't return all categories where product is visible

- fix functional tests

Author: danielrenaud

dev/tests/api-functional/testsuite/Magento/GraphQl/Catalog/ProductViewTest.php

@@ -282,11 +282,11 @@ public function testQueryAllFieldsSimpleProduct()
         $this->assertWebsites($product, $response['products']['items'][0]['websites']);
         self::assertEquals(
             'Movable Position 2',
-            $responseObject->getData('products/items/0/categories/1/name')
+            $responseObject->getData('products/items/0/categories/0/name')
         );
         self::assertEquals(
             'Filter category',
-            $responseObject->getData('products/items/0/categories/2/name')
+            $responseObject->getData('products/items/0/categories/1/name')
         );
         $storeManager = ObjectManager::getInstance()->get(\Magento\Store\Model\StoreManagerInterface::class);
         self::assertEquals(

dev/tests/api-functional/testsuite/Magento/GraphQl/ConfigurableProduct/ConfigurableProductViewTest.php

@@ -300,6 +300,7 @@ private function assertConfigurableVariants($actualResponse)
                 isset($variantArray['product']['id']),
                 'variant product elements don\'t contain id key'
             );
+            $variantProductId = $variantArray['product']['id'];
             $indexValue = $variantArray['product']['sku'];
             unset($variantArray['product']['id']);
             $this->assertTrue(
@@ -310,14 +311,19 @@ private function assertConfigurableVariants($actualResponse)
             /** @var \Magento\Catalog\Model\Product $childProduct */
             $childProduct = $productRepository->get($indexValue);
 
-            /** @var  \Magento\Catalog\Api\Data\ProductLinkInterface[] */
-            $links = $childProduct->getExtensionAttributes()->getCategoryLinks();
-            $this->assertCount(1, $links, "Precondition failed, incorrect number of categories.");
-            $id =$links[0]->getCategoryId();
-
-            $actualValue
-                = $actualResponse['variants'][$variantKey]['product']['categories'][0];
-            $this->assertEquals($actualValue, ['id' => $id]);
+            switch( $variantProductId ){
+                case 10:
+                    $this->assertEmpty(
+                        $actualResponse['variants'][$variantKey]['product']['categories'],
+                        'No category is expected for product, that not visible individually'
+                    );
+                    break;
+                case 20:
+                    $this->assertEquals(
+                        $actualResponse['variants'][$variantKey]['product']['categories'][0],
+                        ['id' => 333]
+                    );
+            }
             unset($variantArray['product']['categories']);
 
             $mediaGalleryEntries = $childProduct->getMediaGalleryEntries();

dev/tests/integration/testsuite/Magento/ConfigurableProduct/_files/product_configurable_with_category_and_weight.php

@@ -19,6 +19,7 @@
 \Magento\TestFramework\Helper\Bootstrap::getInstance()->reinitialize();
 
 require __DIR__ . '/configurable_attribute.php';
+require __DIR__ . '/../../Catalog/_files/category.php';
 
 /** @var ProductRepositoryInterface $productRepository */
 $productRepository = Bootstrap::getObjectManager()
@@ -36,6 +37,10 @@
 $associatedProductIds = [];
 $productIds = [10, 20];
 array_shift($options); //remove the first option which is empty
+$visibility = [
+    10 => Visibility::VISIBILITY_IN_CATALOG,
+    20 => Visibility::VISIBILITY_NOT_VISIBLE,
+];
 
 foreach ($options as $option) {
     /** @var $product Product */
@@ -49,11 +54,11 @@
         ->setSku('simple_' . $productId)
         ->setPrice($productId)
         ->setTestConfigurable($option->getValue())
-        ->setVisibility(Visibility::VISIBILITY_NOT_VISIBLE)
+        ->setVisibility($visibility[$productId])
         ->setStatus(Status::STATUS_ENABLED)
         ->setStockData(['use_config_manage_stock' => 1, 'qty' => 100, 'is_qty_decimal' => 0, 'is_in_stock' => 1]);
     $eavAttributeValues = [
-        'category_ids' => [2]
+        'category_ids' => [333]
         ];
     foreach ($eavAttributeValues as $eavCategoryAttributeCode => $eavCategoryAttributeValues) {
         $product->setCustomAttribute($eavCategoryAttributeCode, $eavCategoryAttributeValues);
@@ -210,5 +215,5 @@
 
 $categoryLinkManagement->assignProductToCategories(
     $product->getSku(),
-    [2]
+    [333]
 );