Test create initial category (#907)

Author: jordisala1991

src/Document/CategoryManager.php

@@ -102,6 +102,10 @@ public function getAllRootCategories(bool $loadChildren = true): array
             ->execute();
         \assert(\is_array($rootCategories));
 
+        if ([] === $rootCategories) {
+            return $this->getRootCategoriesForContext(null);
+        }
+
         $categories = [];
 
         foreach ($rootCategories as $category) {

src/Entity/CategoryManager.php

@@ -103,6 +103,10 @@ public function getAllRootCategories(bool $loadChildren = true): array
             ->getQuery()
             ->getResult();
 
+        if ([] === $rootCategories) {
+            return $this->getRootCategoriesForContext(null);
+        }
+
         $categories = [];
 
         foreach ($rootCategories as $category) {

tests/Functional/Admin/CategoryAdminTest.php

@@ -92,6 +92,29 @@ public static function provideFormUrlsCases(): iterable
         yield 'Remove Category' => ['/admin/tests/app/category/1/delete', [], 'btn_delete'];
     }
 
+    public function testCreateFirstCategory(): void
+    {
+        $client = self::createClient();
+
+        $client->request('GET', '/admin/tests/app/category/create', [
+            'uniqid' => 'category',
+        ]);
+
+        static::assertSame(1, $this->countCategories());
+
+        $client->submitForm('btn_create_and_list', [
+            'category[name]' => 'Name',
+        ]);
+        $client->followRedirect();
+
+        self::assertResponseIsSuccessful();
+
+        $client->request('GET', '/admin/tests/app/category/tree');
+
+        self::assertResponseIsSuccessful();
+        static::assertSame(2, $this->countCategories());
+    }
+
     /**
      * @psalm-suppress UndefinedPropertyFetch
      */
@@ -116,4 +139,18 @@ private function prepareData(): void
 
         $manager->flush();
     }
+
+    /**
+     * @psalm-suppress UndefinedPropertyFetch
+     */
+    private function countCategories(): int
+    {
+        // TODO: Simplify this when dropping support for Symfony 4.
+        // @phpstan-ignore-next-line
+        $container = method_exists(static::class, 'getContainer') ? static::getContainer() : static::$container;
+        $manager = $container->get('doctrine.orm.entity_manager');
+        \assert($manager instanceof EntityManagerInterface);
+
+        return $manager->getRepository(Category::class)->count([]);
+    }
 }