Revert "no encryption on this grant"

This reverts commit 7522a46.
hafezdivandari · Oct 28, 2024 · b7751db · b7751db
1 parent 7522a46
commit b7751db
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/Grant/DeviceCodeGrantTest.php b/tests/Grant/DeviceCodeGrantTest.php
@@ -20,6 +20,7 @@
 use League\OAuth2\Server\Repositories\ScopeRepositoryInterface;
 use LeagueTests\Stubs\AccessTokenEntity;
 use LeagueTests\Stubs\ClientEntity;
+use LeagueTests\Stubs\CryptTraitStub;
 use LeagueTests\Stubs\DeviceCodeEntity;
 use LeagueTests\Stubs\RefreshTokenEntity;
 use LeagueTests\Stubs\ScopeEntity;
@@ -35,6 +36,13 @@ class DeviceCodeGrantTest extends TestCase
     private const DEFAULT_SCOPE = 'basic';
     private const INTERVAL_RATE = 10;
 
+    protected CryptTraitStub $cryptStub;
+
+    public function setUp(): void
+    {
+        $this->cryptStub = new CryptTraitStub();
+    }
+
     public function testGetIdentifier(): void
     {
         $deviceCodeRepositoryMock = $this->getMockBuilder(DeviceCodeRepositoryInterface::class)->getMock();
@@ -92,6 +100,7 @@ public function testRespondToDeviceAuthorizationRequest(): void
 
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setScopeRepository($scopeRepositoryMock);
 
         $request = (new ServerRequest())->withParsedBody([
@@ -136,6 +145,7 @@ public function testRespondToDeviceAuthorizationRequestWithVerificationUriComple
 
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setScopeRepository($scopeRepositoryMock);
 
         $request = (new ServerRequest())->withParsedBody([
@@ -262,6 +272,8 @@ public function testCompleteDeviceAuthorizationRequest(): void
             'http://foo/bar',
         );
 
+        $grant->setEncryptionKey($this->cryptStub->getKey());
+
         $grant->completeDeviceAuthorizationRequest($deviceCode->getIdentifier(), 'userId', true);
 
         $this::assertEquals('userId', $deviceCode->getUserIdentifier());
@@ -311,6 +323,8 @@ public function testDeviceAuthorizationResponse(): void
             'http://foo/bar'
         );
 
+        $deviceCodeGrant->setEncryptionKey($this->cryptStub->getKey());
+
         $server->enableGrantType($deviceCodeGrant);
 
         $response = $server->respondToDeviceAuthorizationRequest($serverRequest, new Response());
@@ -376,6 +390,7 @@ public function testRespondToAccessTokenRequest(): void
         $grant->setAccessTokenRepository($accessTokenRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $grant->completeDeviceAuthorizationRequest($deviceCodeEntity->getIdentifier(), 'baz', true);
@@ -454,6 +469,7 @@ public function testRespondToRequestMissingDeviceCode(): void
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $serverRequest = (new ServerRequest())->withParsedBody([
@@ -501,6 +517,7 @@ public function testIssueSlowDownError(): void
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $serverRequest = (new ServerRequest())->withParsedBody([
@@ -548,6 +565,7 @@ public function testIssueAuthorizationPendingError(): void
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $serverRequest = (new ServerRequest())->withParsedBody([
@@ -595,6 +613,7 @@ public function testIssueExpiredTokenError(): void
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $serverRequest = (new ServerRequest())->withParsedBody([
@@ -638,6 +657,7 @@ public function testSettingDeviceCodeIntervalRate(): void
 
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setIntervalVisibility(true);
 
@@ -692,6 +712,7 @@ public function testIssueAccessDeniedError(): void
         $grant->setClientRepository($clientRepositoryMock);
         $grant->setScopeRepository($scopeRepositoryMock);
         $grant->setDefaultScope(self::DEFAULT_SCOPE);
+        $grant->setEncryptionKey($this->cryptStub->getKey());
         $grant->setPrivateKey(new CryptKey('file://' . __DIR__ . '/../Stubs/private.key'));
 
         $grant->completeDeviceAuthorizationRequest($deviceCode->getIdentifier(), '1', false);