SonataAdminBundle/Tests/Security/Handler/AclSecurityHandlerTest.php

<?php

/*
 * This file is part of the Sonata package.
 *
 * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */
namespace Sonata\AdminBundle\Tests\Admin\Security\Acl\Permission;

use Sonata\AdminBundle\Security\Handler\AclSecurityHandler;

class AclSecurityHandlerTest extends \PHPUnit_Framework_TestCase
{
    public function testAcl()
    {
        $securityContext = $this->getMock('Symfony\Component\Security\Core\SecurityContextInterface');
        $securityContext->expects($this->any())
            ->method('isGranted')
            ->will($this->returnValue(true));

        $handler = new AclSecurityHandler($securityContext);

        $this->assertTrue($handler->isGranted(array('TOTO')));
        $this->assertTrue($handler->isGranted('TOTO'));

        $securityContext = $this->getMock('Symfony\Component\Security\Core\SecurityContextInterface');
        $securityContext->expects($this->any())
            ->method('isGranted')
            ->will($this->returnValue(false));

        $handler = new AclSecurityHandler($securityContext);

        $this->assertFalse($handler->isGranted(array('TOTO')));
        $this->assertFalse($handler->isGranted('TOTO'));
    }

  public function testBuildInformation()
  {
      $informations = array(
          'EDIT' => array('EDIT')
      );

      $securityContext = $this->getMock('Symfony\Component\Security\Core\SecurityContextInterface');
      $admin = $this->getMock('Sonata\AdminBundle\Admin\AdminInterface');
      $admin->expects($this->once())
          ->method('getCode')
          ->will($this->returnValue('test'));

      $admin->expects($this->once())
          ->method('getSecurityInformation')
          ->will($this->returnValue($informations));

      $handler = new AclSecurityHandler($securityContext);

      $results = $handler->buildSecurityInformation($admin);

      $this->assertArrayHasKey('ROLE_TEST_EDIT', $results);
  }
}