Allow single model as well

Tofandel · Jun 5, 2024 · 6ae8e85 · 6ae8e85
1 parent a0dbadf
commit 6ae8e85
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/src/Models/Behaviors/HasRelated.php b/src/Models/Behaviors/HasRelated.php
@@ -65,8 +65,10 @@ public function loadRelated(string $browserName): Collection
      *
      * @param array<int, Model|array> $items
      */
-    public function saveRelated(array|Collection $items, string $browserName): void
+    public function saveRelated(array|Collection|Model $items, string $browserName): void
     {
+        $items = collect($items);
+
         /** @var Collection<int, RelatedItem> $itemsToProcess */
         $itemsToProcess = $this->relatedItems()->where('browser_name', $browserName)->get();
 

diff --git a/tests/integration/Controllers/Tables/BrowserColumnTest.php b/tests/integration/Controllers/Tables/BrowserColumnTest.php
@@ -27,12 +27,7 @@ public function testWithSingleValue(): void
         $category = $this->createCategory();
 
         $this->author->saveRelated(
-            [
-                [
-                    'id' => $category->id,
-                    'endpointType' => Category::class,
-                ],
-            ],
+            $category,
             'categories'
         );
 

diff --git a/tests/integration/NplusOneRelatedTest.php b/tests/integration/NplusOneRelatedTest.php
@@ -35,7 +35,7 @@ public function testSingleRelatedItem(): void
                 'published' => true,
             ]);
 
-            $storeArray[] = ['endpointType' => Writer::class, 'id' => $writer->id];
+            $storeArray[] = $writer;
         }
 
         $letter->saveRelated($storeArray, 'dummyBrowser');