mirror of
https://github.com/wallabag/wallabag.git
synced 2024-06-01 19:33:09 +00:00
fdc90ceb17
Instead of using a HTTP request we just login user like FOSUser does. It allows us to mock service in container for functional tests. Also, fix a bad config name in fos_user for firewall And finally, add functional test to PocketImport
73 lines
2.2 KiB
PHP
73 lines
2.2 KiB
PHP
<?php
|
|
|
|
namespace Tests\Wallabag\CoreBundle\Controller;
|
|
|
|
use Tests\Wallabag\CoreBundle\WallabagCoreTestCase;
|
|
|
|
class SecurityControllerTest extends WallabagCoreTestCase
|
|
{
|
|
public function testLoginWithout2Factor()
|
|
{
|
|
$this->logInAs('admin');
|
|
$client = $this->getClient();
|
|
$client->followRedirects();
|
|
|
|
$crawler = $client->request('GET', '/config');
|
|
$this->assertContains('config.form_rss.description', $crawler->filter('body')->extract(['_text'])[0]);
|
|
}
|
|
|
|
public function testLoginWith2Factor()
|
|
{
|
|
$client = $this->getClient();
|
|
|
|
if (!$client->getContainer()->getParameter('twofactor_auth')) {
|
|
$this->markTestSkipped('twofactor_auth is not enabled.');
|
|
|
|
return;
|
|
}
|
|
|
|
$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->logInAsUsingHttp('admin');
|
|
$crawler = $client->request('GET', '/config');
|
|
$this->assertContains('scheb_two_factor.trusted', $crawler->filter('body')->extract(['_text'])[0]);
|
|
|
|
// restore user
|
|
$user = $em
|
|
->getRepository('WallabagUserBundle:User')
|
|
->findOneByUsername('admin');
|
|
$user->setTwoFactorAuthentication(false);
|
|
$em->persist($user);
|
|
$em->flush();
|
|
}
|
|
|
|
public function testTrustedComputer()
|
|
{
|
|
$client = $this->getClient();
|
|
|
|
if (!$client->getContainer()->getParameter('twofactor_auth')) {
|
|
$this->markTestSkipped('twofactor_auth is not enabled.');
|
|
|
|
return;
|
|
}
|
|
|
|
$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'));
|
|
}
|
|
}
|