59 lines
1.8 KiB
PHP
59 lines
1.8 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
namespace Wallabag\CoreBundle\Tests\Controller;
|
||
|
|
|
||
|
|
use Wallabag\CoreBundle\Tests\WallabagCoreTestCase;
|
||
|
|
|
||
|
|
class SecurityControllerTest extends WallabagCoreTestCase
|
||
|
|
{
|
||
|
|
public function testLoginWithout2Factor()
|
||
|
|
{
|
||
|
|
$this->logInAs('admin');
|
||
|
|
$client = $this->getClient();
|
||
|
|
$client->followRedirects();
|
||
|
|
|
||
|
|
$client->request('GET', '/config');
|
||
|
|
$this->assertContains('RSS', $client->getResponse()->getContent());
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testLoginWith2Factor()
|
||
|
|
{
|
||
|
|
$client = $this->getClient();
|
||
|
|
$client->followRedirects();
|
||
|
|
|
||
|
|
$em = $client->getContainer()->get('doctrine.orm.entity_manager');
|
||
|
|
$user = $em
|
||
|
|
->getRepository('WallabagUserBundle:User')
|
||
|
|
->findOneByUsername('admin');
|
||
|
|
$user->setTwoFactorAuthentication(true);
|
||
|
|
$em->persist($user);
|
||
|
|
$em->flush();
|
||
|
|
|
||
|
|
$this->logInAs('admin');
|
||
|
|
$client->request('GET', '/config');
|
||
|
|
$this->assertContains('trusted computer', $client->getResponse()->getContent());
|
||
|
|
|
||
|
|
// restore user
|
||
|
|
$user = $em
|
||
|
|
->getRepository('WallabagUserBundle:User')
|
||
|
|
->findOneByUsername('admin');
|
||
|
|
$user->setTwoFactorAuthentication(false);
|
||
|
|
$em->persist($user);
|
||
|
|
$em->flush();
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testTrustedComputer()
|
||
|
|
{
|
||
|
|
$client = $this->getClient();
|
||
|
|
$em = $client->getContainer()->get('doctrine.orm.entity_manager');
|
||
|
|
$user = $em
|
||
|
|
->getRepository('WallabagUserBundle:User')
|
||
|
|
->findOneByUsername('admin');
|
||
|
|
|
||
|
|
$date = new \DateTime();
|
||
|
|
$user->addTrustedComputer('ABCDEF', $date->add(new \DateInterval('P1M')));
|
||
|
|
$this->assertTrue($user->isTrustedComputer('ABCDEF'));
|
||
|
|
$this->assertFalse($user->isTrustedComputer('FEDCBA'));
|
||
|
|
}
|
||
|
|
}
|