diff --git a/src/Resources/views/Generator/TopRepositoryTemplate.html.twig b/src/Resources/views/Generator/TopRepositoryTemplate.html.twig index 50e0f32..dee29a4 100644 --- a/src/Resources/views/Generator/TopRepositoryTemplate.html.twig +++ b/src/Resources/views/Generator/TopRepositoryTemplate.html.twig @@ -15,12 +15,14 @@ use Doctrine\Persistence\ManagerRegistry; use Symfony\Component\Uid\Uuid; use Symfony\Contracts\Cache\CacheInterface; use Symfony\Contracts\Cache\ItemInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterByInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterInInterface; /** * Generated class for filter query builders * */ -class {{ entityClassname }}Base extends {{ extendClass }} +class {{ entityClassname }}Base extends {{ extendClass }} implements FilterByInterface, FilterInInterface { const NAME = '{{ entityDql }}'; protected static $parameterIndex = 0; diff --git a/tests/Repository/MyClassRepositoryBase.php b/tests/Repository/MyClassRepositoryBase.php index 878f3d5..d40db2b 100644 --- a/tests/Repository/MyClassRepositoryBase.php +++ b/tests/Repository/MyClassRepositoryBase.php @@ -15,12 +15,14 @@ use Symfony\Component\Uid\Uuid; use Symfony\Contracts\Cache\CacheInterface; use Symfony\Contracts\Cache\ItemInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterByInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterInInterface; /** * Generated class for filter query builders * */ -class MyClassRepositoryBase extends \Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository +class MyClassRepositoryBase extends \Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository implements FilterByInterface, FilterInInterface { const NAME = 'myClass'; protected static $parameterIndex = 0; diff --git a/tests/Service/ExpectedMyClassRepositoryBase.txt b/tests/Service/ExpectedMyClassRepositoryBase.txt index 878f3d5..d40db2b 100644 --- a/tests/Service/ExpectedMyClassRepositoryBase.txt +++ b/tests/Service/ExpectedMyClassRepositoryBase.txt @@ -15,12 +15,14 @@ use Doctrine\Persistence\ManagerRegistry; use Symfony\Component\Uid\Uuid; use Symfony\Contracts\Cache\CacheInterface; use Symfony\Contracts\Cache\ItemInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterByInterface; +use Tbn\QueryBuilderRepositoryGeneratorBundle\Interface\FilterInInterface; /** * Generated class for filter query builders * */ -class MyClassRepositoryBase extends \Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository +class MyClassRepositoryBase extends \Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository implements FilterByInterface, FilterInInterface { const NAME = 'myClass'; protected static $parameterIndex = 0;