init

dev/tests/integration/testsuite/Magento/CustomerGraphQl/Model/Context/AddUserInfoToContextTest.php

@@ -0,0 +1,55 @@
+<?php
+/**
+ * Copyright 2023 Adobe
+ * All Rights Reserved.
+ */
+declare(strict_types=1);
+
+namespace Magento\CustomerGraphQl\Model\Context;
+
+use Magento\Authorization\Model\UserContextInterface;
+use Magento\Authorization\Model\UserContextInterfaceFactory;
+use Magento\Customer\Model\Session;
+use Magento\Customer\Test\Fixture\Customer;
+use Magento\GraphQl\Model\Query\ContextParametersInterface;
+use Magento\TestFramework\Fixture\DataFixture;
+use Magento\TestFramework\Fixture\DataFixtureStorageManager;
+use Magento\TestFramework\Helper\Bootstrap;
+use PHPUnit\Framework\TestCase;
+
+class AddUserInfoToContextTest extends TestCase
+{
+    #[
+        DataFixture(Customer::class, as: 'customer'),
+    ]
+    public function testExecute()
+    {
+        $objectManager = Bootstrap::getObjectManager();
+        $service = $objectManager->get(AddUserInfoToContext::class);
+        $parameters = $objectManager->get(ContextParametersInterface::class);
+
+        /** @var \Magento\Customer\Model\Customer $customer */
+        $customer = DataFixtureStorageManager::getStorage()->get('customer');
+        $userId = $customer->getId();
+        $userType = UserContextInterface::USER_TYPE_CUSTOMER;
+
+        $context = $this->createMock(UserContextInterface::class);
+        $context->method('getUserId')->willReturn($userId);
+        $context->method('getUserType')->willReturn($userType);
+
+        $service->setUserContext($context);
+        $returnedParameters = $service->execute($parameters);
+
+        $this->assertEquals($userId, $returnedParameters->getUserId());
+        $this->assertEquals($userType, $returnedParameters->getUserType());
+
+        $extensionAttributes = $returnedParameters->getExtensionAttributesData();
+        $this->assertArrayHasKey('is_customer', $extensionAttributes);
+        $this->assertTrue($extensionAttributes['is_customer']);
+
+        $session = $objectManager->get(Session::class);
+
+        $this->assertEquals($session->getCustomer()->getData(), $customer->getData());
+        $this->assertEquals($session->getCustomerGroupId(), $customer->getGroupId());
+    }
+}