ACP2E-636: Cannot get graphql customer token

Author: glo23503

dev/tests/api-functional/testsuite/Magento/GraphQl/LoginAsCustomerGraphQl/GenerateLoginCustomerTokenTest.php

@@ -62,6 +62,34 @@ public function testGenerateCustomerValidTokenLoginAsCustomerEnabled()
         $this->assertIsArray($response['generateCustomerTokenAsAdmin']);
     }
 
+    /**
+     * Verify customer token as admin with store
+     *
+     * @magentoApiDataFixture Magento/LoginAsCustomer/_files/admin.php
+     * @magentoConfigFixture admin_store login_as_customer/general/enabled 1
+     * @magentoApiDataFixture Magento/LoginAsCustomer/_files/customer_with_second_store.php
+     * @throws Exception
+     */
+    public function testGenerateCustomerValidTokenAsAdminWithStore()
+    {
+        $customerEmail = '[email protected]';
+
+        $mutation = $this->getQuery($customerEmail);
+
+        $response = $this->graphQlMutation(
+            $mutation,
+            [],
+            '',
+            $this->getAdminHeaderAuthenticationWithStore(
+                'TestAdmin1',
+                \Magento\TestFramework\Bootstrap::ADMIN_PASSWORD,
+                'fixture_second_store'
+            )
+        );
+        $this->assertArrayHasKey('generateCustomerTokenAsAdmin', $response);
+        $this->assertIsArray($response['generateCustomerTokenAsAdmin']);
+    }
+
     /**
      * Verify with Admin email ID and Magento_LoginAsCustomer::login is disabled
      *
@@ -224,4 +252,28 @@ private function getAdminHeaderAuthentication(string $userName, string $password
             );
         }
     }
+
+    /**
+     * To get admin access token with store
+     *
+     * @param string $userName
+     * @param string $password
+     * @param string $storeCode
+     * @return string[]
+     * @throws AuthenticationException
+     */
+    private function getAdminHeaderAuthenticationWithStore(string $userName, string $password, string $storeCode)
+    {
+        try {
+            $adminAccessToken = $this->adminTokenService->createAdminAccessToken($userName, $password);
+            return ['Authorization' => 'Bearer ' . $adminAccessToken, 'store' => $storeCode];
+        } catch (\Exception $e) {
+            throw new AuthenticationException(
+                __(
+                    'The account sign-in was incorrect or your account is disabled temporarily. '
+                    . 'Please wait and try again later.'
+                )
+            );
+        }
+    }
 }

dev/tests/integration/testsuite/Magento/LoginAsCustomer/_files/customer_with_second_store.php

@@ -0,0 +1,67 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+use Magento\Customer\Model\CustomerRegistry;
+use Magento\Customer\Api\Data\CustomerExtensionFactory;
+use Magento\Store\Model\Store;
+use Magento\TestFramework\Helper\Bootstrap;
+use Magento\TestFramework\Workaround\Override\Fixture\Resolver;
+
+Resolver::getInstance()->requireDataFixture('Magento/Store/_files/second_store.php');
+
+$objectManager = Bootstrap::getObjectManager();
+$store = $objectManager->get(Store::class);
+$store->load('fixture_second_store', 'code');
+
+//$objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
+/** @var $repository \Magento\Customer\Api\CustomerRepositoryInterface */
+$repository = $objectManager->create(\Magento\Customer\Api\CustomerRepositoryInterface::class);
+$customer = $objectManager->create(\Magento\Customer\Model\Customer::class);
+
+/** @var CustomerRegistry $customerRegistry */
+$customerRegistry = $objectManager->get(CustomerRegistry::class);
+/** @var Magento\Customer\Model\Customer $customer */
+$customer->setWebsiteId($store->getWebsiteId())
+    ->setId(10)
+    ->setEmail('[email protected]')
+    ->setPassword('password')
+    ->setGroupId($store->getGroupId())
+    ->setStoreId($store->getId())
+    ->setIsActive(1)
+    ->setPrefix('Mr.')
+    ->setFirstname('John')
+    ->setMiddlename('A')
+    ->setLastname('Smith')
+    ->setSuffix('Esq.')
+    ->setDefaultBilling(1)
+    ->setDefaultShipping(1)
+    ->setTaxvat('12')
+    ->setGender(0);
+
+$extension = $customer->getExtensionAttributes();
+if ($extension === null) {
+    $extension = $objectManager->get(CustomerExtensionFactory::class)->create();
+}
+
+$extension->setAssistanceAllowed(2);
+$customer->setExtensionAttributes($extension);
+
+$customer->isObjectNew(true);
+$customer->save();
+
+$customerRegistry->remove($customer->getId());
+/** @var \Magento\JwtUserToken\Api\RevokedRepositoryInterface $revokedRepo */
+$revokedRepo = \Magento\TestFramework\Helper\Bootstrap::getObjectManager()
+    ->get(\Magento\JwtUserToken\Api\RevokedRepositoryInterface::class);
+$revokedRepo->saveRevoked(
+    new \Magento\JwtUserToken\Api\Data\Revoked(
+        \Magento\Authorization\Model\UserContextInterface::USER_TYPE_CUSTOMER,
+        (int) $customer->getId(),
+        time() - 3600 * 24
+    )
+);