wallabag/src/Wallabag/CoreBundle/Tests/Controller/SecurityControllerTest.php

65 lines
2.0 KiB
PHP
Raw Normal View History

2015-10-14 17:10:12 +02:00
<?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();
2015-10-15 13:17:21 +02:00
if ($client->getContainer()->getParameter('twofactor_auth')) {
$client->followRedirects();
2015-10-14 17:10:12 +02:00
2015-10-15 13:17:21 +02:00
$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();
}
2015-10-14 17:10:12 +02:00
}
public function testTrustedComputer()
{
$client = $this->getClient();
2015-10-15 13:17:21 +02:00
if ($client->getContainer()->getParameter('twofactor_auth')) {
$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'));
}
2015-10-14 17:10:12 +02:00
}
}