Suggestions mentioned by Rogyar V1

Author: Krishna Kumar

app/code/Magento/LoginAsCustomerGraphQl/Model/LoginAsCustomer/CreateCustomerToken.php

@@ -11,7 +11,7 @@
 use Magento\Framework\Exception\LocalizedException;
 use Magento\Framework\GraphQl\Exception\GraphQlInputException;
 use Magento\Integration\Model\Oauth\TokenFactory;
-use Magento\Store\Model\StoreManagerInterface;
+use Magento\Store\Api\Data\StoreInterface;
 use Exception;
 
 /**
@@ -31,23 +31,15 @@ class CreateCustomerToken
      */
     private $tokenModelFactory;
 
-    /**
-     * @var StoreManagerInterface
-     */
-    private $storeManager;
-
     /**
      * CreateCustomerToken constructor.
-     * @param StoreManagerInterface $storeManager
      * @param TokenFactory $tokenModelFactory
      * @param CustomerFactory $customerFactory
      */
     public function __construct(
-        StoreManagerInterface $storeManager,
         TokenFactory $tokenModelFactory,
         CustomerFactory $customerFactory
     ) {
-        $this->storeManager = $storeManager;
         $this->tokenModelFactory = $tokenModelFactory;
         $this->customerFactory= $customerFactory;
     }
@@ -56,14 +48,14 @@ public function __construct(
      * Get admin user token
      *
      * @param string $email
+     * @param StoreInterface $store
      * @return array
+     * @throws GraphQlInputException
      * @throws LocalizedException
      */
-    public function execute(string $email): array
+    public function execute(string $email, StoreInterface $store): array
     {
-        $websiteID = $this->storeManager->getStore()->getWebsiteId();
-
-        $customer = $this->customerFactory->create()->setWebsiteId($websiteID)->loadByEmail($email);
+        $customer = $this->customerFactory->create()->setWebsiteId((int)$store->getId())->loadByEmail($email);
 
         /* Check if customer email exist */
         if (!$customer->getId()) {

app/code/Magento/LoginAsCustomerGraphQl/Model/Resolver/RequestCustomerToken.php

@@ -81,9 +81,9 @@ public function resolve(
 
         /* Get input params */
         try {
-            $args = $args['input'] ?: [];
+            $args = $args['input'];
         } catch (NoSuchEntityException $e) {
-            throw new GraphQlNoSuchEntityException(__('Check input params.'));
+            throw new GraphQlInputException(__('Check input params.'));
         }
 
         if (empty(trim($args['customer_email'], " "))) {
@@ -97,8 +97,10 @@ public function resolve(
                 __('Login as Customer is disabled.')
             );
         }
-
-        return $this->createCustomerToken->execute($args['customer_email']);
+        return $this->createCustomerToken->execute(
+            $args['customer_email'],
+            $context->getExtensionAttributes()->getStore()
+        );
     }
 
     /**

app/code/Magento/LoginAsCustomerGraphQl/composer.json

@@ -8,8 +8,7 @@
         "magento/module-integration": "*",
         "magento/module-store": "*",
         "magento/module-customer": "*",
-        "magento/module-catalog-graph-ql": "*",
-        "magento/framework": "*"
+        "magento/module-catalog-graph-ql": "*"
     },
     "type": "magento2-module",
     "license": [

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

@@ -84,8 +84,6 @@ public function testGenerateCustomerValidTokenLoginAsCustomerDisabled()
             '',
             $this->getAdminHeaderAuthentication('TestAdmin1', 'Zilker777')
         );
-        $this->assertArrayHasKey('generateCustomerTokenAsAdmin', $response);
-        $this->assertIsArray($response['generateCustomerTokenAsAdmin']);
     }
 
     /**
@@ -105,14 +103,12 @@ public function testGenerateCustomerTokenLoginWithCustomerCredentials()
 
         $mutation = $this->getQuery($customerEmail);
 
-        $response = $this->graphQlMutation(
+        $this->graphQlMutation(
             $mutation,
             [],
             '',
             $this->getCustomerHeaderAuthentication($customerEmail, $password)
         );
-        $this->assertArrayHasKey('generateCustomerTokenAsAdmin', $response);
-        $this->assertIsArray($response['generateCustomerTokenAsAdmin']);
     }
 
     /**
@@ -137,7 +133,7 @@ public function testGenerateCustomerTokenInvalidData(
         $this->expectExceptionMessage($message);
 
         $mutation = $this->getQuery($customerEmail);
-        $response = $this->graphQlMutation(
+        $this->graphQlMutation(
             $mutation,
             [],
             '',
@@ -218,7 +214,6 @@ private function getAdminHeaderAuthentication(string $userName, string $password
     {
         try {
             $adminAccessToken = $this->adminTokenService->createAdminAccessToken($userName, $password);
-
             return ['Authorization' => 'Bearer ' . $adminAccessToken];
         } catch (\Exception $e) {
             throw new AuthenticationException(