CABI-365::AdobeAdminIms to AdobeIms code migration-updated unit tests

Author: mahesh-singh-rajawat

app/code/Magento/AdobeIms/Model/LogOut.php

@@ -107,7 +107,7 @@ public function execute(?string $accessToken = null) : bool
                 $accessToken = $session->getAdobeAccessToken();
             }
             if (!empty($accessToken)) {
-                return $this->adminLogoutFromIms($accessToken);
+                return $this->logoutAdminFromIms($accessToken);
             }
             $accessToken = $accessToken ?? $this->getAccessToken->execute();
             if (empty($accessToken)) {
@@ -149,7 +149,7 @@ private function externalLogOut(string $accessToken): void
      * @return bool
      * @throws LocalizedException
      */
-    private function adminLogoutFromIms(string $accessToken): bool
+    private function logoutAdminFromIms(string $accessToken): bool
     {
         if (!$this->checkUserProfile($accessToken)) {
             throw new LocalizedException(

app/code/Magento/AdobeIms/Test/Unit/Model/LogOutTest.php

@@ -9,13 +9,13 @@
 namespace Magento\AdobeIms\Test\Unit\Model;
 
 use Exception;
+use Magento\AdobeIms\Model\GetProfile;
 use Magento\AdobeIms\Model\LogOut;
+use Magento\Backend\Model\Auth\StorageInterface;
 use Magento\AdobeImsApi\Api\ConfigInterface;
-use Magento\AdobeImsApi\Api\Data\UserProfileInterface;
 use Magento\AdobeImsApi\Api\FlushUserTokensInterface;
 use Magento\AdobeImsApi\Api\GetAccessTokenInterface;
-use Magento\AdobeImsApi\Api\UserProfileRepositoryInterface;
-use Magento\Authorization\Model\UserContextInterface;
+use Magento\Backend\Model\Auth;
 use Magento\Framework\HTTP\Client\Curl;
 use Magento\Framework\HTTP\Client\CurlFactory;
 use PHPUnit\Framework\MockObject\MockObject;
@@ -60,6 +60,16 @@ class LogOutTest extends TestCase
      */
     private $model;
 
+    /**
+     * @var Auth|MockObject
+     */
+    private $auth;
+
+    /**
+     * @var GetProfile|MockObject
+     */
+    private $profile;
+
     /**
      * @inheritdoc
      */
@@ -70,12 +80,16 @@ protected function setUp(): void
         $this->loggerInterfaceMock = $this->createMock(LoggerInterface::class);
         $this->getToken = $this->createMock(GetAccessTokenInterface::class);
         $this->flushTokens = $this->createMock(FlushUserTokensInterface::class);
+        $this->profile = $this->createMock(GetProfile::class);
+        $this->auth = $this->createMock(Auth::class);
         $this->model = new LogOut(
             $this->loggerInterfaceMock,
             $this->configInterfaceMock,
             $this->curlFactoryMock,
             $this->getToken,
-            $this->flushTokens
+            $this->flushTokens,
+            $this->profile,
+            $this->auth
         );
     }
 
@@ -104,7 +118,15 @@ public function testExecute(): void
 
         $this->flushTokens->expects($this->once())
             ->method('execute');
-
+        $session = $this->getMockBuilder(StorageInterface::class)
+            ->addMethods(['getAdobeAccessToken'])
+            ->getMockForAbstractClass();
+        $session->expects($this->once())
+            ->method('getAdobeAccessToken')
+            ->willReturn(null);
+        $this->auth->expects($this->once())
+            ->method('getAuthStorage')
+            ->willReturn($session);
         $this->assertEquals(true, $this->model->execute());
     }
 
@@ -135,7 +157,15 @@ public function testExecuteWithError(): void
 
         $this->flushTokens->expects($this->never())
             ->method('execute');
-
+        $session = $this->getMockBuilder(StorageInterface::class)
+            ->addMethods(['getAdobeAccessToken'])
+            ->getMockForAbstractClass();
+        $session->expects($this->once())
+            ->method('getAdobeAccessToken')
+            ->willReturn(null);
+        $this->auth->expects($this->once())
+            ->method('getAuthStorage')
+            ->willReturn($session);
         $this->assertEquals(false, $this->model->execute());
     }
 
@@ -161,7 +191,15 @@ public function testExecuteWithException(): void
         $curl->expects($this->once())
             ->method('getStatus')
             ->willReturn(self::HTTP_FOUND);
-
+        $session = $this->getMockBuilder(StorageInterface::class)
+            ->addMethods(['getAdobeAccessToken'])
+            ->getMockForAbstractClass();
+        $session->expects($this->once())
+            ->method('getAdobeAccessToken')
+            ->willReturn(null);
+        $this->auth->expects($this->once())
+            ->method('getAuthStorage')
+            ->willReturn($session);
         $this->flushTokens->expects($this->once())
             ->method('execute')
             ->willThrowException(new Exception('Could not save user profile.'));