[Tests] Add test for Grid with custom repository method

src/Bundle/test/src/Repository/BookRepository.php

@@ -15,6 +15,7 @@
 
 use App\Entity\Book;
 use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
+use Doctrine\ORM\QueryBuilder;
 use Doctrine\Persistence\ManagerRegistry;
 use Sylius\Bundle\ResourceBundle\Doctrine\ORM\ResourceRepositoryTrait;
 
@@ -26,4 +27,12 @@ public function __construct(ManagerRegistry $registry)
     {
         parent::__construct($registry, Book::class);
     }
+
+    public function createListQueryBuilder(): QueryBuilder
+    {
+        $queryBuilder = $this->createQueryBuilder('o');
+        $queryBuilder->andWhere($queryBuilder->expr()->isNotNull('o.archivedAt'));
+
+        return $queryBuilder;
+    }
 }

src/Bundle/test/src/Subscription/Entity/Subscription.php

@@ -14,6 +14,7 @@
 namespace App\Subscription\Entity;
 
 use App\Subscription\Form\Type\SubscriptionType;
+use App\Subscription\Repository\SubscriptionRepository;
 use Doctrine\ORM\Mapping as ORM;
 use Sylius\Component\Resource\Metadata\ApplyStateMachineTransition;
 use Sylius\Component\Resource\Metadata\Create;
@@ -40,12 +41,15 @@
 #[ApplyStateMachineTransition(stateMachineTransition: 'reject')]
 #[Delete]
 #[Show(template: 'subscription/show.html.twig')]
-#[ORM\Entity]
+#[ORM\Entity(repositoryClass: SubscriptionRepository::class)]
 class Subscription implements ResourceInterface
 {
     #[ORM\Column(type: 'string')]
     public string $state = 'new';
 
+    #[ORM\Column(type: 'datetime', nullable: true)]
+    public ?\DateTimeInterface $archivedAt = null;
+
     public function __construct(
         #[ORM\Id]
         #[ORM\Column(type: 'uuid', unique: true)]

src/Bundle/test/src/Subscription/Grid/SubscriptionGrid.php

@@ -36,6 +36,7 @@ public static function getName(): string
     public function buildGrid(GridBuilderInterface $gridBuilder): void
     {
         $gridBuilder
+            ->setRepositoryMethod('createListQueryBuilder', ['showArchived' => true])
             ->orderBy('email', 'asc')
             ->addField(
                 StringField::create('email')

src/Bundle/test/src/Subscription/Repository/SubscriptionRepository.php

@@ -0,0 +1,44 @@
+<?php
+
+/*
+ * This file is part of the Sylius package.
+ *
+ * (c) Paweł Jędrzejewski
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+declare(strict_types=1);
+
+namespace App\Subscription\Repository;
+
+use App\Subscription\Entity\Subscription;
+use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
+use Doctrine\ORM\QueryBuilder;
+use Doctrine\Persistence\ManagerRegistry;
+use Sylius\Bundle\ResourceBundle\Doctrine\ORM\ResourceRepositoryTrait;
+use Sylius\Component\Resource\Repository\RepositoryInterface;
+
+final class SubscriptionRepository extends ServiceEntityRepository implements RepositoryInterface
+{
+    use ResourceRepositoryTrait;
+
+    public function __construct(ManagerRegistry $registry)
+    {
+        parent::__construct($registry, Subscription::class);
+    }
+
+    public function createListQueryBuilder(bool $showArchived = false): QueryBuilder
+    {
+        $queryBuilder = $this->createQueryBuilder('o');
+
+        if (!$showArchived) {
+            $queryBuilder->andWhere(
+                $queryBuilder->expr()->isNull('o.archivedAt'),
+            );
+        }
+
+        return $queryBuilder;
+    }
+}

src/Bundle/test/src/Tests/Controller/SubscriptionUiTest.php

@@ -55,6 +55,11 @@ public function it_allows_browsing_subscriptions(): void
         $this->assertStringContainsString(sprintf('<a href="/admin/subscriptions/%s">Show</a>', $subscriptions['subscription_doc']->getId()), $content);
         $this->assertStringContainsString(sprintf('<a href="/admin/subscriptions/%s/edit">Edit</a>', $subscriptions['subscription_doc']->getId()), $content);
         $this->assertStringContainsString(sprintf('<form action="/admin/subscriptions/%s" method="post">', $subscriptions['subscription_doc']->getId()), $content);
+
+        $this->assertStringContainsString('<td>[email protected]</td>', $content);
+        $this->assertStringContainsString(sprintf('<a href="/admin/subscriptions/%s">Show</a>', $subscriptions['subscription_archived']->getId()), $content);
+        $this->assertStringContainsString(sprintf('<a href="/admin/subscriptions/%s/edit">Edit</a>', $subscriptions['subscription_archived']->getId()), $content);
+        $this->assertStringContainsString(sprintf('<form action="/admin/subscriptions/%s" method="post">', $subscriptions['subscription_archived']->getId()), $content);
     }
 
     /** @test */

src/Bundle/test/src/Tests/DataFixtures/ORM/subscriptions.yml

@@ -3,3 +3,6 @@ App\Subscription\Entity\Subscription:
         email: '[email protected]'
     subscription_doc:
         email: '[email protected]'
+    subscription_archived:
+        email: '[email protected]'
+        archivedAt: '<dateTimeBetween("-200 days", "now")>'