MC-37582: Fix Failing WebAPI GraphQL tests when DB table prefixes are enabled

- fix test

Author: dthampy

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

@@ -11,9 +11,11 @@
 use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Catalog\Model\CategoryRepository;
 use Magento\Catalog\Model\ResourceModel\Category\Collection as CategoryCollection;
+use Magento\Framework\App\ResourceConnection;
 use Magento\Framework\EntityManager\MetadataPool;
 use Magento\Store\Model\Store;
 use Magento\Store\Model\StoreManagerInterface;
+use Magento\TestFramework\Helper\Bootstrap;
 use Magento\TestFramework\ObjectManager;
 use Magento\TestFramework\TestCase\GraphQlAbstract;
 
@@ -564,10 +566,12 @@ public function testCategoryImage(?string $imagePrefix)
             ->addAttributeToFilter('name', ['eq' => 'Parent Image Category'])
             ->getFirstItem();
         $categoryId = $categoryModel->getId();
+        /** @var ResourceConnection $resourceConnection */
+        $resourceConnection = Bootstrap::getObjectManager()->create(ResourceConnection::class);
+        $connection = $resourceConnection->getConnection();
 
         if ($imagePrefix !== null) {
             // update image to account for different stored image formats
-            $connection = $categoryCollection->getConnection();
             $productLinkField = $this->metadataPool
                 ->getMetadata(\Magento\Catalog\Api\Data\ProductInterface::class)
                 ->getLinkField();
@@ -577,20 +581,20 @@ public function testCategoryImage(?string $imagePrefix)
             $imageAttributeValue = $imagePrefix . basename($categoryModel->getImage());
 
             if (!empty($imageAttributeValue)) {
-                $query = sprintf(
+                $sqlQuery = sprintf(
                     'UPDATE %s SET `value` = "%s" ' .
                     'WHERE `%s` = %d ' .
                     'AND `store_id`= %d ' .
                     'AND `attribute_id` = ' .
                     '(SELECT `ea`.`attribute_id` FROM %s ea WHERE `ea`.`attribute_code` = "image" LIMIT 1)',
-                    $connection->getTableName('catalog_category_entity_varchar'),
+                    $resourceConnection->getTableName('catalog_category_entity_varchar'),
                     $imageAttributeValue,
                     $productLinkField,
                     $categoryModel->getData($productLinkField),
                     $defaultStoreId,
-                    $connection->getTableName('eav_attribute')
+                    $resourceConnection->getTableName('eav_attribute')
                 );
-                $connection->query($query);
+                $connection->query($sqlQuery);
             }
         }
 

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

@@ -12,11 +12,13 @@
 use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Catalog\Model\CategoryRepository;
 use Magento\Catalog\Model\ResourceModel\Category\Collection as CategoryCollection;
+use Magento\Framework\App\ResourceConnection;
 use Magento\Framework\DataObject;
 use Magento\Framework\EntityManager\MetadataPool;
 use Magento\Store\Model\Store;
 use Magento\Store\Model\StoreManagerInterface;
 use Magento\TestFramework\ObjectManager;
+use Magento\TestFramework\Helper\Bootstrap;
 use Magento\TestFramework\TestCase\GraphQl\ResponseContainsErrorsException;
 use Magento\TestFramework\TestCase\GraphQlAbstract;
 
@@ -587,9 +589,12 @@ public function testCategoryImage(?string $imagePrefix)
             ->getFirstItem();
         $categoryId = $categoryModel->getId();
 
+        /** @var ResourceConnection $resourceConnection */
+        $resourceConnection = Bootstrap::getObjectManager()->create(ResourceConnection::class);
+        $connection = $resourceConnection->getConnection();
+
         if ($imagePrefix !== null) {
-            // update image to account for different stored image formats
-            $connection = $categoryCollection->getConnection();
+            // update image to account for different stored image format
             $productLinkField = $this->metadataPool
                 ->getMetadata(\Magento\Catalog\Api\Data\ProductInterface::class)
                 ->getLinkField();
@@ -599,20 +604,20 @@ public function testCategoryImage(?string $imagePrefix)
             $imageAttributeValue = $imagePrefix . basename($categoryModel->getImage());
 
             if (!empty($imageAttributeValue)) {
-                $query = sprintf(
+                $sqlQuery = sprintf(
                     'UPDATE %s SET `value` = "%s" ' .
                     'WHERE `%s` = %d ' .
                     'AND `store_id`= %d ' .
                     'AND `attribute_id` = ' .
                     '(SELECT `ea`.`attribute_id` FROM %s ea WHERE `ea`.`attribute_code` = "image" LIMIT 1)',
-                    $connection->getTableName('catalog_category_entity_varchar'),
+                    $resourceConnection->getTableName('catalog_category_entity_varchar'),
                     $imageAttributeValue,
                     $productLinkField,
                     $categoryModel->getData($productLinkField),
                     $defaultStoreId,
-                    $connection->getTableName('eav_attribute')
+                    $resourceConnection->getTableName('eav_attribute')
                 );
-                $connection->query($query);
+                $connection->query($sqlQuery);
             }
         }