- $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();
+ $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();
+ }