AC-14095: Fix web api test

Author: wip44850

dev/tests/api-functional/testsuite/Magento/GraphQl/GraphQlCache/CacheIdFactorProviders/Customer/CustomerGroupProviderTest.php

@@ -1,17 +1,23 @@
 <?php
 /**
- * Copyright © Magento, Inc. All rights reserved.
- * See COPYING.txt for license details.
+ * Copyright 2021 Adobe
+ * All Rights Reserved.
  */
 declare(strict_types=1);
 
 namespace Magento\GraphQl\GraphQlCache\CacheIdFactorProviders\Customer;
 
+use Magento\Catalog\Test\Fixture\Product as ProductFixture;
 use Magento\Customer\Api\CustomerRepositoryInterface;
 use Magento\Customer\Model\Group;
+use Magento\Customer\Test\Fixture\Customer as CustomerFixture;
 use Magento\Framework\Exception\AuthenticationException;
+use Magento\Framework\Exception\EmailNotConfirmedException;
 use Magento\GraphQlCache\Model\CacheId\CacheIdCalculator;
 use Magento\Integration\Api\CustomerTokenServiceInterface;
+use Magento\TestFramework\Fixture\DataFixture;
+use Magento\TestFramework\Fixture\DataFixtureStorage;
+use Magento\TestFramework\Fixture\DataFixtureStorageManager;
 use Magento\TestFramework\Helper\Bootstrap;
 use Magento\TestFramework\TestCase\GraphQlAbstract;
 
@@ -35,57 +41,70 @@ class CustomerGroupProviderTest extends GraphQlAbstract
      */
     private $customerRepository;
 
+    /**
+     * @var DataFixtureStorage
+     */
+    private $fixtures;
+
     protected function setUp(): void
     {
-        $objectManager = Bootstrap::getObjectManager();
-        $this->customerTokenService = $objectManager->get(CustomerTokenServiceInterface::class);
-        $this->customerGroup = $objectManager->get(Group::class);
-        $this->customerRepository = $objectManager->get(CustomerRepositoryInterface::class);
+        $this->customerTokenService = Bootstrap::getObjectManager()->get(CustomerTokenServiceInterface::class);
+        $this->customerGroup = Bootstrap::getObjectManager()->get(Group::class);
+        $this->customerRepository = Bootstrap::getObjectManager()->get(CustomerRepositoryInterface::class);
+        $this->fixtures = Bootstrap::getObjectManager()->get(DataFixtureStorageManager::class)->getStorage();
     }
 
     /**
      * Tests that cache id header changes based on customer group and remains consistent for same customer group
-     *
-     * @magentoApiDataFixture Magento/Customer/_files/customer.php
-     * @magentoApiDataFixture Magento/GraphQl/Catalog/_files/simple_product.php
      */
-    public function testCacheIdHeaderWithCustomerGroup()
+    #[
+        DataFixture(ProductFixture::class, as: 'product'),
+        DataFixture(CustomerFixture::class, as: 'customer'),
+    ]
+    public function testCacheIdHeaderWithCustomerGroup(): void
     {
-        $sku = 'simple_product';
+        $customerEmail = $this->fixtures->get('customer')->getEmail();
         $query = <<<QUERY
- {
-           products(filter: {sku: {eq: "{$sku}"}})
-           {
-               items {
-                   id
-                   name
-                   sku
-                   description {
-                   html
+         {
+               products(filter: {sku: {eq: "{$this->fixtures->get('product')->getSku()}"}})
+               {
+                   items {
+                       id
+                       name
+                       sku
+                       description {
+                       html
+                       }
                    }
                }
            }
-       }
-QUERY;
-        $response = $this->graphQlQueryWithResponseHeaders($query, [], '', $this->getHeaderMap());
+        QUERY;
+        $response = $this->graphQlQueryWithResponseHeaders(
+            $query,
+            [],
+            '',
+            $this->getHeaderMap($customerEmail)
+        );
         $this->assertArrayHasKey(CacheIdCalculator::CACHE_ID_HEADER, $response['headers']);
         $cacheId = $response['headers'][CacheIdCalculator::CACHE_ID_HEADER];
         $this->assertTrue((boolean)preg_match('/^[0-9a-f]{64}$/i', $cacheId));
-        $groupCode = 'Retailer';
-        $customer = $this->customerRepository->get('[email protected]');
-        $customerGroupId = $this->customerGroup->load($groupCode, 'customer_group_code')->getId();
+        $customer = $this->customerRepository->get($customerEmail);
+        $customerGroupId = $this->customerGroup->load('Retailer', 'customer_group_code')->getId();
         // change the customer group of this customer from the default group
         $customer->setGroupId($customerGroupId);
         $this->customerRepository->save($customer);
         $responseAfterCustomerGroupChange = $this->graphQlQueryWithResponseHeaders(
             $query,
             [],
             '',
-            $this->getHeaderMap()
+            $this->getHeaderMap($customerEmail)
+        );
+        $this->assertArrayHasKey(
+            CacheIdCalculator::CACHE_ID_HEADER,
+            $responseAfterCustomerGroupChange['headers']
         );
-        $this->assertArrayHasKey(CacheIdCalculator::CACHE_ID_HEADER, $responseAfterCustomerGroupChange['headers']);
         $cacheIdCustomerGroupChange = $responseAfterCustomerGroupChange['headers'][CacheIdCalculator::CACHE_ID_HEADER];
-        // Verify that the the cache id generated is a 64 character long
+        // Verify that the cache id generated is a 64 character long
         $this->assertTrue((boolean)preg_match('/^[0-9a-f]{64}$/i', $cacheId));
         // check that the cache ids generated before and after customer group changes are not equal
         $this->assertNotEquals($cacheId, $cacheIdCustomerGroupChange);
@@ -96,28 +115,28 @@ public function testCacheIdHeaderWithCustomerGroup()
             $query,
             [],
             '',
-            $this->getHeaderMap()
+            $this->getHeaderMap($customerEmail)
         );
-        $cacheIdDefaultCustomerGroup = $responseDefaultCustomerGroup['headers'][CacheIdCalculator::CACHE_ID_HEADER];
         //Verify that the cache id is same as original $cacheId
-        $this->assertEquals($cacheIdDefaultCustomerGroup, $cacheId);
+        $this->assertEquals(
+            $responseDefaultCustomerGroup['headers'][CacheIdCalculator::CACHE_ID_HEADER],
+            $cacheId
+        );
     }
 
     /**
      * Authentication header map
      *
      * @param string $username
-     * @param string $password
-     *
      * @return array
      *
      * @throws AuthenticationException
+     * @throws EmailNotConfirmedException
      */
-    private function getHeaderMap(string $username = '[email protected]', string $password = 'password'): array
+    private function getHeaderMap(string $username): array
     {
-        $customerToken = $this->customerTokenService->createCustomerAccessToken($username, $password);
+        $customerToken = $this->customerTokenService->createCustomerAccessToken($username, 'password');
 
         return ['Authorization' => 'Bearer ' . $customerToken];
     }
-
 }