<?php

use GuzzleHttp\Client;

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

class Google_AccessToken_RevokeTest extends BaseTest
{
  public function testRevokeAccessGuzzle5()
  {
    $this->onlyGuzzle5();

    $accessToken = 'ACCESS_TOKEN';
    $refreshToken = 'REFRESH_TOKEN';
    $token = '';

    $response = $this->getMock('GuzzleHttp\Message\ResponseInterface');
    $response->expects($this->exactly(3))
      ->method('getStatusCode')
      ->will($this->returnValue(200));
    $http = $this->getMock('GuzzleHttp\ClientInterface');
    $http->expects($this->exactly(3))
      ->method('send')
      ->will($this->returnCallback(
            function ($request) use (&$token, $response) {
              parse_str((string) $request->getBody(), $fields);
              $token = isset($fields['token']) ? $fields['token'] : null;

              return $response;
            }
        ));

    $requestToken = null;
    $request = $this->getMock('GuzzleHttp\Message\RequestInterface');
    $request->expects($this->exactly(3))
        ->method('getBody')
        ->will($this->returnCallback(
            function () use (&$requestToken) {
              return 'token='.$requestToken;
            }));
    $http->expects($this->exactly(3))
      ->method('createRequest')
      ->will($this->returnCallback(
            function ($method, $url, $params) use (&$requestToken, $request) {
              parse_str((string) $params['body'], $fields);
              $requestToken = isset($fields['token']) ? $fields['token'] : null;

              return $request;
            }
        ));

    $t = array(
      'access_token' => $accessToken,
      'created' => time(),
      'expires_in' => '3600'
    );

    // Test with access token.
    $revoke = new Google_AccessToken_Revoke($http);
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($accessToken, $token);

    // Test with refresh token.
    $revoke = new Google_AccessToken_Revoke($http);
    $t = array(
      'access_token' => $accessToken,
      'refresh_token' => $refreshToken,
      'created' => time(),
      'expires_in' => '3600'
    );
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($refreshToken, $token);

    // Test with token string.
    $revoke = new Google_AccessToken_Revoke($http);
    $t = $accessToken;
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($accessToken, $token);
  }

  public function testRevokeAccessGuzzle6()
  {
    $this->onlyGuzzle6();

    $accessToken = 'ACCESS_TOKEN';
    $refreshToken = 'REFRESH_TOKEN';
    $token = '';

    $response = $this->getMock('Psr\Http\Message\ResponseInterface');
    $response->expects($this->exactly(3))
      ->method('getStatusCode')
      ->will($this->returnValue(200));
    $http = $this->getMock('GuzzleHttp\ClientInterface');
    $http->expects($this->exactly(3))
      ->method('send')
      ->will($this->returnCallback(
            function ($request) use (&$token, $response) {
              parse_str((string) $request->getBody(), $fields);
              $token = isset($fields['token']) ? $fields['token'] : null;

              return $response;
            }
        ));

    $t = array(
      'access_token' => $accessToken,
      'created' => time(),
      'expires_in' => '3600'
    );

    // Test with access token.
    $revoke = new Google_AccessToken_Revoke($http);
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($accessToken, $token);

    // Test with refresh token.
    $revoke = new Google_AccessToken_Revoke($http);
    $t = array(
      'access_token' => $accessToken,
      'refresh_token' => $refreshToken,
      'created' => time(),
      'expires_in' => '3600'
    );
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($refreshToken, $token);

    // Test with token string.
    $revoke = new Google_AccessToken_Revoke($http);
    $t = $accessToken;
    $this->assertTrue($revoke->revokeToken($t));
    $this->assertEquals($accessToken, $token);
  }
}