Handle state in OAuth2 client

1 files changed, 5 insertions, 2 deletions

diff --git a/tests/units/Core/Http/OAuth2Test.php b/tests/units/Core/Http/OAuth2Test.php
index c68ae116..5a9c0ac1 100644
--- a/tests/units/Core/Http/OAuth2Test.php
+++ b/tests/units/Core/Http/OAuth2Test.php
@@ -10,7 +10,8 @@ class OAuth2Test extends Base
     {
         $oauth = new OAuth2($this->container);
         $oauth->createService('A', 'B', 'C', 'D', 'E', array('f', 'g'));
-        $this->assertEquals('D?response_type=code&client_id=A&redirect_uri=C&scope=f+g', $oauth->getAuthorizationUrl());
+        $state = $oauth->getState();
+        $this->assertEquals('D?response_type=code&client_id=A&redirect_uri=C&scope=f+g&state='.$state, $oauth->getAuthorizationUrl());
     }
 
     public function testAuthHeader()
@@ -27,12 +28,15 @@ class OAuth2Test extends Base
 
     public function testAccessToken()
     {
+        $oauth = new OAuth2($this->container);
+
         $params = array(
             'code' => 'something',
             'client_id' => 'A',
             'client_secret' => 'B',
             'redirect_uri' => 'C',
             'grant_type' => 'authorization_code',
+            'state' => $oauth->getState(),
         );
 
         $response = json_encode(array(
@@ -46,7 +50,6 @@ class OAuth2Test extends Base
             ->with('E', $params, array('Accept: application/json'))
             ->will($this->returnValue($response));
 
-        $oauth = new OAuth2($this->container);
         $oauth->createService('A', 'B', 'C', 'D', 'E', array('f', 'g'));
         $oauth->getAccessToken('something');
     }