simplify tests

hafezdivandari · Nov 1, 2024 · 60d8e51 · 60d8e51
1 parent 1b005d3
commit 60d8e51
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/tests/Grant/DeviceCodeGrantTest.php b/tests/Grant/DeviceCodeGrantTest.php
@@ -350,19 +350,8 @@ public function testRespondToAccessTokenRequest(): void
         $scope = new ScopeEntity();
         $scope->setIdentifier('foo');
 
-        $deviceCodeRepositoryMock = $this->getMockBuilder(DeviceCodeRepositoryInterface::class)->getMock();
-        $deviceCodeEntity = new DeviceCodeEntity();
-
-        $deviceCodeEntity->setUserIdentifier('baz');
-        $deviceCodeEntity->setUserCode('123456');
-
-        $deviceCodeRepositoryMock->method('getDeviceCodeEntityByDeviceCode')
-            ->with('deviceCodeEntityIdentifier')
-            ->willReturn($deviceCodeEntity);
-
         $accessTokenEntity = new AccessTokenEntity();
         $accessTokenEntity->addScope($scope);
-
         $accessTokenRepositoryMock = $this->getMockBuilder(AccessTokenRepositoryInterface::class)->getMock();
         $accessTokenRepositoryMock->method('getNewToken')->willReturn($accessTokenEntity);
         $accessTokenRepositoryMock->method('persistNewAccessToken')->willReturnSelf();
@@ -371,6 +360,17 @@ public function testRespondToAccessTokenRequest(): void
         $refreshTokenRepositoryMock->method('persistNewRefreshToken')->willReturnSelf();
         $refreshTokenRepositoryMock->method('getNewRefreshToken')->willReturn(new RefreshTokenEntity());
 
+        $deviceCodeRepositoryMock = $this->getMockBuilder(DeviceCodeRepositoryInterface::class)->getMock();
+        $deviceCodeEntity = new DeviceCodeEntity();
+        $deviceCodeEntity->setUserIdentifier('baz');
+        $deviceCodeEntity->setIdentifier('deviceCodeEntityIdentifier');
+        $deviceCodeEntity->setUserCode('123456');
+        $deviceCodeEntity->setExpiryDateTime(new DateTimeImmutable('+1 hour'));
+        $deviceCodeEntity->setClient($client);
+        $deviceCodeRepositoryMock->method('getDeviceCodeEntityByDeviceCode')
+            ->with('deviceCodeEntityIdentifier')
+            ->willReturn($deviceCodeEntity);
+
         $scopeRepositoryMock = $this->getMockBuilder(ScopeRepositoryInterface::class)->getMock();
         $scopeRepositoryMock->method('getScopeEntityByIdentifier')->willReturn($scope);
         $scopeRepositoryMock->method('finalizeScopes')->willReturnArgument(0);