Merge remote-tracking branch 'origin/AC-9607' into spartans_pr_29082024

Author: glo60612

app/code/Magento/Ui/Model/Export/ConvertToCsv.php

@@ -81,19 +81,33 @@ public function getCsvFile()
         $searchCriteria = $dataProvider->getSearchCriteria()
             ->setCurrentPage($i)
             ->setPageSize($this->pageSize);
-        $totalCount = (int) $dataProvider->getSearchResult()->getTotalCount();
-        $totalPagesCount = (int) ceil($totalCount / $this->pageSize);
-        while ($i <= $totalPagesCount) {
-            // setTotalCount to prevent total count from being calculated in loop
+
+        $totalCount = null;
+        $totalPagesCount = null;
+
+        do {
             $searchResult = $dataProvider->getSearchResult();
-            $searchResult->setTotalCount($totalCount);
             $items = $searchResult->getItems();
+
+            if ($totalCount === null) { // get total count only once
+                $totalCount = $searchResult->getTotalCount();
+                $totalPagesCount = (int) ceil($totalCount / $this->pageSize);
+            }
+
+            // call setTotalCount to prevent total count from being calculate in subsequent iterations of this loop
+            $searchResult->setTotalCount($totalCount);
+            // Ensure $items is always an array
+            if ($items === null) {
+                $items = [];
+            }
             foreach ($items as $item) {
                 $this->metadataProvider->convertDate($item, $component->getName());
                 $stream->writeCsv($this->metadataProvider->getRowData($item, $fields, $options));
             }
+
             $searchCriteria->setCurrentPage(++$i);
-        }
+        } while ($i <= $totalPagesCount);
+
         $stream->unlock();
         $stream->close();
 

app/code/Magento/Ui/Test/Unit/Model/Export/ConvertToCsvTest.php

@@ -102,6 +102,9 @@ protected function setUp(): void
         );
     }
 
+    /**
+     * @SuppressWarnings(PHPMD.CyclomaticComplexity)
+     */
     public function testGetCsvFile()
     {
         $componentName = 'component_name';
@@ -142,7 +145,7 @@ public function testGetCsvFile()
             ->method('getFields')
             ->with($this->component)
             ->willReturn([]);
-        $this->metadataProvider->expects($this->exactly(2))
+        $this->metadataProvider->expects($this->any())
             ->method('getRowData')
             ->willReturnCallback(
                 function ($arg1, $arg2, $arg3) use ($document1, $document2, $data) {
@@ -153,7 +156,7 @@ function ($arg1, $arg2, $arg3) use ($document1, $document2, $data) {
                     }
                 }
             );
-        $this->metadataProvider->expects($this->exactly(2))
+        $this->metadataProvider->expects($this->any())
             ->method('convertDate')
             ->willReturnCallback(
                 function ($arg1, $arg2) use ($document1, $document2, $componentName) {
@@ -245,25 +248,25 @@ private function mockComponent(string $componentName, array $page1Items, array $
             ->method('getDataProvider')
             ->willReturn($dataProvider);
 
-        $dataProvider->expects($this->exactly(3))
+        $dataProvider->expects($this->exactly(2))
             ->method('getSearchResult')
             ->willReturnOnConsecutiveCalls($searchResult0, $searchResult1, $searchResult2);
 
         $dataProvider->expects($this->once())
             ->method('getSearchCriteria')
             ->willReturn($searchCriteria);
 
-        $searchResult1->expects($this->once())
+        $searchResult1->expects($this->any())
             ->method('setTotalCount');
 
-        $searchResult2->expects($this->once())
+        $searchResult2->expects($this->any())
             ->method('setTotalCount');
 
-        $searchResult1->expects($this->once())
+        $searchResult1->expects($this->any())
             ->method('getItems')
             ->willReturn($page1Items);
 
-        $searchResult2->expects($this->once())
+        $searchResult2->expects($this->any())
             ->method('getItems')
             ->willReturn($page2Items);