X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/72c8b138e12a0596c56538e4c43e52847bfec2e0..refs/pull/3593/head:/tests/TestCase.php

diff --git a/tests/TestCase.php b/tests/TestCase.php
index 95e58b267..9ffce917e 100644
--- a/tests/TestCase.php
+++ b/tests/TestCase.php
@@ -428,4 +428,17 @@ abstract class TestCase extends BaseTestCase
 
         $this->assertDatabaseHas('activities', $detailsToCheck);
     }
+
+    /**
+     * @return Entity[]
+     */
+    protected function getEachEntityType(): array
+    {
+        return [
+            'page' => Page::query()->first(),
+            'chapter' => Chapter::query()->first(),
+            'book' => Book::query()->first(),
+            'bookshelf' => Bookshelf::query()->first(),
+        ];
+    }
 }