f91f4dda13
ResponseHeaderBag::hasCookie() and ResponseHeaderBag::getCookie() were removed
330 lines
14 KiB
PHP
330 lines
14 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony framework.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* This source file is subject to the MIT license that is bundled
|
|
* with this source code in the file LICENSE.
|
|
*/
|
|
|
|
namespace Symfony\Tests\Component\Security\Http\RememberMe;
|
|
|
|
use Symfony\Component\Security\Http\RememberMe\RememberMeServicesInterface;
|
|
|
|
use Symfony\Component\Security\Core\Authentication\Token\RememberMeToken;
|
|
use Symfony\Component\HttpFoundation\HeaderBag;
|
|
use Symfony\Component\Security\Core\Exception\UsernameNotFoundException;
|
|
use Symfony\Component\Security\Core\Authentication\RememberMe\PersistentToken;
|
|
use Symfony\Component\HttpFoundation\Request;
|
|
use Symfony\Component\HttpFoundation\Response;
|
|
use Symfony\Component\HttpFoundation\ResponseHeaderBag;
|
|
use Symfony\Component\Security\Http\RememberMe\PersistentTokenBasedRememberMeServices;
|
|
use Symfony\Component\Security\Core\Exception\TokenNotFoundException;
|
|
use Symfony\Component\Security\Core\Exception\CookieTheftException;
|
|
|
|
class PersistentTokenBasedRememberMeServicesTest extends \PHPUnit_Framework_TestCase
|
|
{
|
|
public function testAutoLoginReturnsNullWhenNoCookie()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo'));
|
|
|
|
$this->assertNull($service->autoLogin(new Request()));
|
|
}
|
|
|
|
public function testAutoLoginThrowsExceptionOnInvalidCookie()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null, 'always_remember_me' => false, 'remember_me_parameter' => 'foo'));
|
|
$request = new Request;
|
|
$request->request->set('foo', 'true');
|
|
$request->cookies->set('foo', 'foo');
|
|
|
|
$this->assertNull($service->autoLogin($request));
|
|
$this->assertTrue($request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME)->isCleared());
|
|
}
|
|
|
|
public function testAutoLoginThrowsExceptionOnNonExistentToken()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null, 'always_remember_me' => false, 'remember_me_parameter' => 'foo'));
|
|
$request = new Request;
|
|
$request->request->set('foo', 'true');
|
|
$request->cookies->set('foo', $this->encodeCookie(array(
|
|
$series = 'fooseries',
|
|
$tokenValue = 'foovalue',
|
|
)));
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('loadTokenBySeries')
|
|
->will($this->throwException(new TokenNotFoundException('Token not found.')))
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$this->assertNull($service->autoLogin($request));
|
|
$this->assertTrue($request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME)->isCleared());
|
|
}
|
|
|
|
public function testAutoLoginReturnsNullOnNonExistentUser()
|
|
{
|
|
$userProvider = $this->getProvider();
|
|
$service = $this->getService($userProvider, array('name' => 'foo', 'path' => null, 'domain' => null, 'always_remember_me' => true, 'lifetime' => 3600, 'secure' => false, 'httponly' => false));
|
|
$request = new Request;
|
|
$request->cookies->set('foo', $this->encodeCookie(array('fooseries', 'foovalue')));
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('loadTokenBySeries')
|
|
->will($this->returnValue(new PersistentToken('fooclass', 'fooname', 'fooseries', 'foovalue', new \DateTime())))
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$userProvider
|
|
->expects($this->once())
|
|
->method('loadUserByUsername')
|
|
->will($this->throwException(new UsernameNotFoundException('user not found')))
|
|
;
|
|
|
|
$this->assertNull($service->autoLogin($request));
|
|
$this->assertTrue($request->attributes->has(RememberMeServicesInterface::COOKIE_ATTR_NAME));
|
|
}
|
|
|
|
public function testAutoLoginThrowsExceptionOnStolenCookieAndRemovesItFromThePersistentBackend()
|
|
{
|
|
$userProvider = $this->getProvider();
|
|
$service = $this->getService($userProvider, array('name' => 'foo', 'path' => null, 'domain' => null, 'always_remember_me' => true));
|
|
$request = new Request;
|
|
$request->cookies->set('foo', $this->encodeCookie(array('fooseries', 'foovalue')));
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('loadTokenBySeries')
|
|
->will($this->returnValue(new PersistentToken('fooclass', 'foouser', 'fooseries', 'anotherFooValue', new \DateTime())))
|
|
;
|
|
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('deleteTokenBySeries')
|
|
->with($this->equalTo('fooseries'))
|
|
->will($this->returnValue(null))
|
|
;
|
|
|
|
try {
|
|
$service->autoLogin($request);
|
|
$this->fail('Expected CookieTheftException was not thrown.');
|
|
} catch (CookieTheftException $theft) { }
|
|
|
|
$this->assertTrue($request->attributes->has(RememberMeServicesInterface::COOKIE_ATTR_NAME));
|
|
}
|
|
|
|
public function testAutoLoginDoesNotAcceptAnExpiredCookie()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null, 'always_remember_me' => true, 'lifetime' => 3600));
|
|
$request = new Request;
|
|
$request->cookies->set('foo', $this->encodeCookie(array('fooseries', 'foovalue')));
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('loadTokenBySeries')
|
|
->with($this->equalTo('fooseries'))
|
|
->will($this->returnValue(new PersistentToken('fooclass', 'username', 'fooseries', 'foovalue', new \DateTime('yesterday'))))
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$this->assertNull($service->autoLogin($request));
|
|
$this->assertTrue($request->attributes->has(RememberMeServicesInterface::COOKIE_ATTR_NAME));
|
|
}
|
|
|
|
public function testAutoLogin()
|
|
{
|
|
$user = $this->getMock('Symfony\Component\Security\Core\User\UserInterface');
|
|
$user
|
|
->expects($this->once())
|
|
->method('getRoles')
|
|
->will($this->returnValue(array('ROLE_FOO')))
|
|
;
|
|
|
|
$userProvider = $this->getProvider();
|
|
$userProvider
|
|
->expects($this->once())
|
|
->method('loadUserByUsername')
|
|
->with($this->equalTo('foouser'))
|
|
->will($this->returnValue($user))
|
|
;
|
|
|
|
$service = $this->getService($userProvider, array('name' => 'foo', 'path' => null, 'domain' => null, 'secure' => false, 'httponly' => false, 'always_remember_me' => true, 'lifetime' => 3600));
|
|
$request = new Request;
|
|
$request->cookies->set('foo', $this->encodeCookie(array('fooseries', 'foovalue')));
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('loadTokenBySeries')
|
|
->with($this->equalTo('fooseries'))
|
|
->will($this->returnValue(new PersistentToken('fooclass', 'foouser', 'fooseries', 'foovalue', new \DateTime())))
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$returnedToken = $service->autoLogin($request);
|
|
|
|
$this->assertInstanceOf('Symfony\Component\Security\Core\Authentication\Token\RememberMeToken', $returnedToken);
|
|
$this->assertSame($user, $returnedToken->getUser());
|
|
$this->assertEquals('fookey', $returnedToken->getKey());
|
|
$this->assertTrue($request->attributes->has(RememberMeServicesInterface::COOKIE_ATTR_NAME));
|
|
}
|
|
|
|
public function testLogout()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => '/foo', 'domain' => 'foodomain.foo'));
|
|
$request = new Request();
|
|
$request->cookies->set('foo', $this->encodeCookie(array('fooseries', 'foovalue')));
|
|
$response = new Response();
|
|
$token = $this->getMock('Symfony\Component\Security\Core\Authentication\Token\TokenInterface');
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('deleteTokenBySeries')
|
|
->with($this->equalTo('fooseries'))
|
|
->will($this->returnValue(null))
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$service->logout($request, $response, $token);
|
|
|
|
$cookie = $request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME);
|
|
$this->assertTrue($cookie->isCleared());
|
|
$this->assertEquals('/foo', $cookie->getPath());
|
|
$this->assertEquals('foodomain.foo', $cookie->getDomain());
|
|
}
|
|
|
|
public function testLogoutSimplyIgnoresNonSetRequestCookie()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null));
|
|
$request = new Request;
|
|
$response = new Response;
|
|
$token = $this->getMock('Symfony\Component\Security\Core\Authentication\Token\TokenInterface');
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->never())
|
|
->method('deleteTokenBySeries')
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$service->logout($request, $response, $token);
|
|
|
|
$cookie = $request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME);
|
|
$this->assertTrue($cookie->isCleared());
|
|
$this->assertNull($cookie->getPath());
|
|
$this->assertNull($cookie->getDomain());
|
|
}
|
|
|
|
public function testLogoutSimplyIgnoresInvalidCookie()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null));
|
|
$request = new Request;
|
|
$request->cookies->set('foo', 'somefoovalue');
|
|
$response = new Response;
|
|
$token = $this->getMock('Symfony\Component\Security\Core\Authentication\Token\TokenInterface');
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->never())
|
|
->method('deleteTokenBySeries')
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$service->logout($request, $response, $token);
|
|
|
|
$this->assertTrue($request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME)->isCleared());
|
|
}
|
|
|
|
public function testLoginFail()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'path' => null, 'domain' => null));
|
|
$request = new Request();
|
|
|
|
$this->assertFalse($request->attributes->has(RememberMeServicesInterface::COOKIE_ATTR_NAME));
|
|
$service->loginFail($request);
|
|
$this->assertTrue($request->attributes->get(RememberMeServicesInterface::COOKIE_ATTR_NAME)->isCleared());
|
|
}
|
|
|
|
public function testLoginSuccessSetsCookieWhenLoggedInWithNonRememberMeTokenInterfaceImplementation()
|
|
{
|
|
$service = $this->getService(null, array('name' => 'foo', 'domain' => 'myfoodomain.foo', 'path' => '/foo/path', 'secure' => true, 'httponly' => true, 'lifetime' => 3600, 'always_remember_me' => true));
|
|
$request = new Request;
|
|
$response = new Response;
|
|
|
|
$account = $this->getMock('Symfony\Component\Security\Core\User\UserInterface');
|
|
$account
|
|
->expects($this->once())
|
|
->method('getUsername')
|
|
->will($this->returnValue('foo'))
|
|
;
|
|
$token = $this->getMock('Symfony\Component\Security\Core\Authentication\Token\TokenInterface');
|
|
$token
|
|
->expects($this->any())
|
|
->method('getUser')
|
|
->will($this->returnValue($account))
|
|
;
|
|
|
|
$tokenProvider = $this->getMock('Symfony\Component\Security\Core\Authentication\RememberMe\TokenProviderInterface');
|
|
$tokenProvider
|
|
->expects($this->once())
|
|
->method('createNewToken')
|
|
;
|
|
$service->setTokenProvider($tokenProvider);
|
|
|
|
$cookies = $response->headers->getCookies();
|
|
$this->assertEquals(0, count($cookies));
|
|
|
|
$service->loginSuccess($request, $response, $token);
|
|
|
|
$cookies = $response->headers->getCookies(ResponseHeaderBag::COOKIES_ARRAY);
|
|
$cookie = $cookies['myfoodomain.foo']['/foo/path']['foo'];
|
|
$this->assertFalse($cookie->isCleared());
|
|
$this->assertTrue($cookie->isSecure());
|
|
$this->assertTrue($cookie->isHttpOnly());
|
|
$this->assertTrue($cookie->getExpiresTime() > time() + 3590 && $cookie->getExpiresTime() < time() + 3610);
|
|
$this->assertEquals('myfoodomain.foo', $cookie->getDomain());
|
|
$this->assertEquals('/foo/path', $cookie->getPath());
|
|
}
|
|
|
|
protected function encodeCookie(array $parts)
|
|
{
|
|
$service = $this->getService();
|
|
$r = new \ReflectionMethod($service, 'encodeCookie');
|
|
$r->setAccessible(true);
|
|
|
|
return $r->invoke($service, $parts);
|
|
}
|
|
|
|
protected function getService($userProvider = null, $options = array(), $logger = null)
|
|
{
|
|
if (null === $userProvider) {
|
|
$userProvider = $this->getProvider();
|
|
}
|
|
|
|
return new PersistentTokenBasedRememberMeServices(array($userProvider), 'fookey', 'fookey', $options, $logger);
|
|
}
|
|
|
|
protected function getProvider()
|
|
{
|
|
$provider = $this->getMock('Symfony\Component\Security\Core\User\UserProviderInterface');
|
|
$provider
|
|
->expects($this->any())
|
|
->method('supportsClass')
|
|
->will($this->returnValue(true))
|
|
;
|
|
|
|
return $provider;
|
|
}
|
|
}
|