Added grabNumRecords function

TavoNiievez · TavoNiievez · commit ef145f33876b · 2020-12-11T10:30:13.000-05:00
diff --git a/src/Codeception/Module/Symfony.php b/src/Codeception/Module/Symfony.php
@@ -840,6 +840,33 @@ public function seeNumRecords(int $expectedNum, string $className, array $criter
         );
     }
 
+    /**
+     * Retrieves number of records from database
+     * 'id' is the default search parameter.
+     *
+     * ```php
+     * <?php
+     * $I->grabNumRecords('User::class', ['name' => 'davert']);
+     * ```
+     *
+     * @param string $entityClass  The entity class
+     * @param array $criteria      Optional query criteria
+     * @return integer
+     */
+    public function grabNumRecords(string $entityClass, array $criteria = [])
+    {
+        $em         = $this->_getEntityManager();
+        $repository = $em->getRepository($entityClass);
+
+        if (empty($criteria)) {
+            return (int)$repository->createQueryBuilder('a')
+                ->select('count(a.id)')
+                ->getQuery()
+                ->getSingleScalarResult();
+        }
+        return $repository->count($criteria);
+    }
+
     /**
      * Invalidate the current session.
      * ```php
