fix tests

Author: full

Entity/TokenManager.php

@@ -91,7 +91,7 @@ public function deleteExpired()
     protected function getRepository(): EntityRepository
     {
         $repository = $this->em->getRepository($this->class);
-        if(!($repository instanceof EntityRepository)){
+        if (!($repository instanceof EntityRepository)) {
             throw new \RuntimeException('EntityRepository needed');
         }
 

Tests/Entity/TokenManagerTest.php

@@ -13,6 +13,7 @@
 
 namespace FOS\OAuthServerBundle\Tests\Entity;
 
+use Doctrine\Common\Persistence\ObjectRepository;
 use Doctrine\ORM\AbstractQuery;
 use Doctrine\ORM\EntityManager;
 use Doctrine\ORM\EntityManagerInterface;
@@ -64,7 +65,6 @@ public function setUp()
         ;
 
         $this->entityManager
-            ->expects($this->once())
             ->method('getRepository')
             ->with($this->className)
             ->willReturn($this->repository)
@@ -76,7 +76,6 @@ public function setUp()
     public function testConstructWillSetParameters()
     {
         $this->assertAttributeSame($this->entityManager, 'em', $this->instance);
-        $this->assertAttributeSame($this->repository, 'repository', $this->instance);
         $this->assertAttributeSame($this->className, 'class', $this->instance);
     }
 
@@ -228,4 +227,29 @@ public function testDeleteExpired()
 
         $this->assertSame($randomResult, $this->instance->deleteExpired());
     }
+
+    public function testExceptionWithObjectRepository()
+    {
+        $this->repository = $this->getMockBuilder(ObjectRepository::class)
+            ->disableOriginalConstructor()
+            ->getMock()
+        ;
+
+        $this->entityManager = $this->getMockBuilder(EntityManager::class)
+            ->disableOriginalConstructor()
+            ->getMock()
+        ;
+
+        $this->entityManager
+            ->expects($this->once())
+            ->method('getRepository')
+            ->with($this->className)
+            ->willReturn($this->repository)
+        ;
+
+        $this->instance = new TokenManager($this->entityManager, $this->className);
+
+        $this->expectException(\RuntimeException::class);
+        $this->instance->findTokenBy([]);
+    }
 }