B2B-2677: [MediaGallery]Implement data caching for GraphQL results on resolver level

- Add testCacheIsInvalidatedWhenUpdatingMediaGalleryEntriesOnAProductViaImport (should fail)
- Refactor

Author: danmooney2

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

@@ -23,6 +23,7 @@
 use Magento\Framework\ObjectManagerInterface;
 use Magento\GraphQlResolverCache\Model\Resolver\Result\CacheKey\Calculator\ProviderInterface;
 use Magento\GraphQlResolverCache\Model\Resolver\Result\Type as GraphQlResolverCache;
+use Magento\ImportExport\Model\Import;
 use Magento\Integration\Model\Integration;
 use Magento\TestFramework\Fixture\DataFixture;
 use Magento\TestFramework\Helper\Bootstrap;
@@ -418,22 +419,75 @@ public function testCacheIsInvalidatedOnProductDeletion()
     #[
         DataFixture(ProductFixture::class, ['sku' => 'product1', 'media_gallery_entries' => [[]]], as: 'product'),
     ]
-    public function testCacheIsNotInvalidatedWhenUpdatingProductSpecificAttributeViaImport()
+    public function testCacheIsInvalidatedWhenUpdatingMediaGalleryEntriesOnAProductViaImport()
     {
         // first, create an integration so that cache is not cleared in
         // Magento\TestFramework\Authentication\OauthHelper::_createIntegration before making the API call
-        /** @var $integrationService \Magento\Integration\Api\IntegrationServiceInterface */
-        $integrationService = $this->objectManager->get(\Magento\Integration\Api\IntegrationServiceInterface::class);
+        $integration = $this->createNewOauthIntegration();
 
-        $params = [
-            'all_resources' => true,
-            'integration_id' => 1,
-            'status' => Integration::STATUS_ACTIVE,
-            'name' => 'Integration' . microtime()
+        $product = $this->productRepository->get('product1');
+
+        $this->assertCount(
+            1,
+            $product->getMediaGalleryEntries()
+        );
+
+        $query = $this->getProductWithMediaGalleryQuery($product);
+        $response = $this->graphQlQuery($query);
+
+        $this->assertCount(
+            1,
+            $response['products']['items'][0]['media_gallery']
+        );
+
+        $this->assertMediaGalleryResolverCacheRecordExists($product);
+
+        $serviceInfo = [
+            'rest' => [
+                'resourcePath' => '/V1/import/csv',
+                'httpMethod' => \Magento\Framework\Webapi\Rest\Request::HTTP_METHOD_POST,
+            ],
         ];
 
-        $integration = $integrationService->create($params);
-        $integration->setStatus(\Magento\Integration\Model\Integration::STATUS_ACTIVE)->save();
+        $requestData = [
+            'source' => [
+                'entity' => 'catalog_product',
+                'behavior' => 'append',
+                'validationStrategy' => 'validation-stop-on-errors',
+                'allowedErrorCount' => '10',
+                Import::FIELD_NAME_IMG_FILE_DIR =>
+                    '/Users/dmooney/Code/vagrant-magento/magento2ce/var/import/images/product_images',
+                'csvData' => base64_encode("sku,base_image\nproduct1,design_patterns.jpeg\n"),
+            ],
+        ];
+
+        $response = $this->_webApiCall($serviceInfo, $requestData, 'rest', null, $integration);
+
+        $this->assertEquals(
+            'Entities Processed: 1',
+            $response[0]
+        );
+
+        // assert product has updated media gallery entry count
+        $this->productRepository->cleanCache();
+        $updatedProduct = $this->productRepository->get('product1');
+        $this->assertCount(
+            2,
+            $updatedProduct->getMediaGalleryEntries()
+        );
+
+        // assert media gallery resolver cache is invalidated
+        $this->assertMediaGalleryResolverCacheRecordDoesNotExist($product);
+    }
+
+    #[
+        DataFixture(ProductFixture::class, ['sku' => 'product1', 'media_gallery_entries' => [[]]], as: 'product'),
+    ]
+    public function testCacheIsNotInvalidatedWhenUpdatingProductSpecificAttributeViaImport()
+    {
+        // first, create an integration so that cache is not cleared in
+        // Magento\TestFramework\Authentication\OauthHelper::_createIntegration before making the API call
+        $integration = $this->createNewOauthIntegration();
 
         $product = $this->productRepository->get('product1');
 
@@ -489,6 +543,29 @@ public function testCacheIsNotInvalidatedWhenUpdatingProductSpecificAttributeVia
         $this->assertMediaGalleryResolverCacheRecordExists($product);
     }
 
+    /**
+     *
+     * @return Integration
+     * @throws \Magento\Framework\Exception\IntegrationException
+     */
+    private function createNewOauthIntegration(): Integration
+    {
+        /** @var $integrationService \Magento\Integration\Api\IntegrationServiceInterface */
+        $integrationService = $this->objectManager->get(\Magento\Integration\Api\IntegrationServiceInterface::class);
+
+        $params = [
+            'all_resources' => true,
+            'integration_id' => 1,
+            'status' => Integration::STATUS_ACTIVE,
+            'name' => 'Integration' . microtime()
+        ];
+
+        $integration = $integrationService->create($params);
+        $integration->setStatus(\Magento\Integration\Model\Integration::STATUS_ACTIVE)->save();
+
+        return $integration;
+    }
+
     /**
      * Assert that media gallery cache record exists for the $product.
      *