Add count method to InMemoryRepository

README.md

@@ -53,7 +53,7 @@ The following methods on Doctrine's `EntityManagerInterface` should all work as
 - getCache (will always return `null`)
 - isOpen (will always return `true`)
 
-All methods on the `ObjectRepository` (for various findBy operations) should also work.
+All methods on the `ObjectRepository` (for various findBy operations) should also work, as well as the non-interface `count($criteria)` method.
 `ObjectRepository` also implements the `Selectable` interface (as `EntityRepository` does, which is the returned type from `EntityManager`), so it's also possible to use the `matching(Criteria)` method.
 
 The following methods are **not** supported at this time:

src/InMemoryRepository.php

@@ -199,6 +199,19 @@ public function findBy(array $criteria, ?array $orderBy = null, $limit = null, $
         return $this->doMatch($crit);
     }
 
+    /**
+     * Counts entities by a set of critieria.
+     *
+     * NOTE: this is not part of the official interface; there's
+     * a Doctrine-internal TODO to make it so.
+     *
+     * @param array<string, mixed> $criteria
+     */
+    public function count(array $criteria): int
+    {
+        return count($this->findBy($criteria));
+    }
+
     /**
      * Finds a single object by a set of criteria.
      *

tests/InMemoryRepositoryTest.php

@@ -74,6 +74,18 @@ public function testSimpleFindBy(): void
         $this->assertSame('[email protected]', $results[1]->getEmail());
     }
 
+    public function testCount(): void
+    {
+        $repo = $this->getFixture();
+        self::assertSame(5, $repo->count([]));
+        self::assertSame(2, $repo->count(['lastName' => 'last']));
+        self::assertSame(3, $repo->count(['lastName' => 'other']));
+        self::assertSame(1, $repo->count(['email' => '[email protected]', 'lastName' => 'last']));
+        self::assertSame(0, $repo->count(['email' => '[email protected]', 'lastName' => 'other']));
+        self::assertSame(1, $repo->count(['email' => '[email protected]']));
+        self::assertSame(0, $repo->count(['email' => '[email protected]']));
+    }
+
     public function testFindByWithArrayValue(): void
     {
         $repo = $this->getFixture();