PB-366: Templates Lower Level Test Coverage

Author: davemacaulay

app/code/Magento/PageBuilder/Model/TemplateRepository.php

@@ -121,15 +121,8 @@ public function getList(SearchCriteriaInterface $criteria)
 
         $searchResults = $this->searchResultsFactory->create();
         $searchResults->setSearchCriteria($criteria);
-
-        $items = [];
-        foreach ($collection as $model) {
-            $items[] = $model;
-        }
-
-        $searchResults->setItems($items);
+        $searchResults->setItems($collection->getItems());
         $searchResults->setTotalCount($collection->getSize());
-
         return $searchResults;
     }
 

dev/tests/integration/testsuite/Magento/PageBuilder/Controller/Adminhtml/Template/DeleteTest.php

@@ -67,7 +67,8 @@ public function testDeleteAction()
             $this->searchCriteriaBuilder->addFilter('name', 'Test Template')->create()
         );
         $this->assertEquals(1, $findTemplate->getTotalCount());
-        $templateId = $findTemplate->getItems()[0]->getId();
+        $items = $findTemplate->getItems();
+        $templateId = reset($items)->getId();
 
         $this->getRequest()->setPostValue(['template_id' => $templateId])->setMethod(HttpRequest::METHOD_GET);
         $this->deleteController->execute();