Change the EntityManger to ModelManger adapter into a trait

src/Entity/AccessTokenRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\AccessTokenManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * AccessTokenRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class AccessTokenRepository extends AbstractEntityRepository implements AccessTokenManagerInterface
+class AccessTokenRepository extends EntityRepository implements AccessTokenManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

src/Entity/AuthorizeRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\AuthorizeManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * AuthorizeRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class AuthorizeRepository extends AbstractEntityRepository implements AuthorizeManagerInterface
+class AuthorizeRepository extends EntityRepository implements AuthorizeManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

src/Entity/ClientRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\ClientManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * ClientRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class ClientRepository extends AbstractEntityRepository implements ClientManagerInterface
+class ClientRepository extends EntityRepository implements ClientManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

src/Entity/CodeRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\CodeManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * CodeRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class CodeRepository extends AbstractEntityRepository implements CodeManagerInterface
+class CodeRepository extends EntityRepository implements CodeManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

src/Entity/AbstractEntityRepository.php → src/Entity/ModelManagerEntityRepository.php

@@ -21,7 +21,7 @@
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class AbstractEntityRepository extends EntityRepository implements ModelManagerInterface
+trait ModelManagerEntityRepository
 {
     public function createModel(ModelInterface $model)
     {

src/Entity/RefreshTokenRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\RefreshTokenManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * RefreshTokenRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class RefreshTokenRepository extends AbstractEntityRepository implements RefreshTokenManagerInterface
+class RefreshTokenRepository extends EntityRepository implements RefreshTokenManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

src/Entity/ScopeRepository.php

@@ -12,13 +12,15 @@
 namespace AuthBucket\Bundle\OAuth2Bundle\Entity;
 
 use AuthBucket\OAuth2\Model\ScopeManagerInterface;
+use Doctrine\ORM\EntityRepository;
 
 /**
  * ScopeRepository.
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class ScopeRepository extends AbstractEntityRepository implements ScopeManagerInterface
+class ScopeRepository extends EntityRepository implements ScopeManagerInterface
 {
+    use ModelManagerEntityRepository;
 }

tests/TestBundle/Entity/UserRepository.php

@@ -11,7 +11,9 @@
 
 namespace AuthBucket\Bundle\OAuth2Bundle\Tests\TestBundle\Entity;
 
-use AuthBucket\Bundle\OAuth2Bundle\Entity\AbstractEntityRepository;
+use AuthBucket\Bundle\OAuth2Bundle\Entity\ModelManagerEntityRepository;
+use AuthBucket\OAuth2\Model\ModelManagerInterface;
+use Doctrine\ORM\EntityRepository;
 use Symfony\Bridge\Doctrine\Security\User\UserLoaderInterface;
 use Symfony\Component\Security\Core\Exception\UsernameNotFoundException;
 use Symfony\Component\Security\Core\User\UserInterface;
@@ -22,8 +24,10 @@
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class UserRepository extends AbstractEntityRepository implements UserLoaderInterface
+class UserRepository extends EntityRepository implements UserLoaderInterface, ModelManagerInterface
 {
+    use ModelManagerEntityRepository;
+
     public function createUser()
     {
         $class = $this->getClassName();