Merge branch 'B2B-2020' into B2B-2024

Author: oshmyheliuk

dev/tests/integration/testsuite/Magento/AdvancedPricingImportExport/Model/Export/AdvancedPricingTest.php

@@ -6,19 +6,19 @@
 
 namespace Magento\AdvancedPricingImportExport\Model\Export;
 
-use Magento\Framework\App\Filesystem\DirectoryList;
-use Magento\Framework\Exception\NoSuchEntityException;
-use Magento\Framework\File\Csv;
-use Magento\TestFramework\Indexer\TestCase;
-use Magento\TestFramework\Helper\Bootstrap;
-use Magento\Framework\Filesystem;
 use Magento\AdvancedPricingImportExport\Model\Export\AdvancedPricing as ExportAdvancedPricing;
+use Magento\AdvancedPricingImportExport\Model\Import\AdvancedPricing as ImportAdvancedPricing;
 use Magento\Catalog\Api\ProductRepositoryInterface;
 use Magento\Catalog\Model\Product;
+use Magento\Framework\App\Filesystem\DirectoryList;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\Framework\Filesystem;
+use Magento\Framework\Filesystem\Directory\Write;
 use Magento\ImportExport\Model\Export\Adapter\Csv as ExportAdapterCsv;
-use Magento\AdvancedPricingImportExport\Model\Import\AdvancedPricing as ImportAdvancedPricing;
-use Magento\ImportExport\Model\Import\Source\Csv as ImportSourceCsv;
 use Magento\ImportExport\Model\Import;
+use Magento\ImportExport\Model\Import\Source\Csv as ImportSourceCsv;
+use Magento\TestFramework\Helper\Bootstrap;
+use Magento\TestFramework\Indexer\TestCase;
 
 /**
  * Test for \Magento\AdvancedPricingImportExport\Model\Export\AdvancedPricing
@@ -41,6 +41,11 @@ class AdvancedPricingTest extends TestCase
      */
     protected $fileSystem;
 
+    /**
+     * @var Write
+     */
+    private $directory;
+
     // @codingStandardsIgnoreStart
     public static function setUpBeforeClass(): void
     {
@@ -64,6 +69,7 @@ protected function setUp(): void
 
         $this->objectManager = Bootstrap::getObjectManager();
         $this->fileSystem = $this->objectManager->get(Filesystem::class);
+        $this->directory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_IMPORT_EXPORT);
         $this->model = $this->objectManager->create(ExportAdvancedPricing::class);
     }
 
@@ -179,7 +185,7 @@ public function testExportImportOfAdvancedPricing(): void
     {
         $simpleSku = 'simple';
         $secondSimpleSku = 'second_simple';
-        $csvfile = uniqid('importexport_') . '.csv';
+        $csvfile = $this->directory->getAbsolutePath(uniqid('importexport_') . '.csv');
         $exportContent = $this->exportData($csvfile);
         $this->assertStringContainsString(
             \sprintf('%s,"All Websites [USD]","ALL GROUPS",10.0000,3.00,Discount', $secondSimpleSku),
@@ -269,10 +275,7 @@ private function updateTierPriceDataInCsv(string $csvfile): void
             ],
         ];
 
-        /** @var Csv $csv */
-        $csv = $this->objectManager->get(Csv::class);
-        $varDirectory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_DIR);
-        $csv->appendData($varDirectory->getAbsolutePath($csvfile), $csvNewData);
+        $this->updateCsvFile($csvfile, $csvNewData);
     }
 
     /**
@@ -281,16 +284,15 @@ private function updateTierPriceDataInCsv(string $csvfile): void
      */
     private function exportData($csvFile)
     {
-        $this->model->setWriter(
-            Bootstrap::getObjectManager()
-                ->create(
-                    ExportAdapterCsv::class,
-                    ['fileSystem' => $this->fileSystem, 'destination' => $csvFile]
-                )
-        );
+        $writer = Bootstrap::getObjectManager()->create(ExportAdapterCsv::class, ['fileSystem' => $this->fileSystem]);
+
+        $this->model->setWriter($writer);
         $exportContent = $this->model->export();
         $this->assertNotEmpty($exportContent);
 
+        $driver = $this->directory->getDriver();
+        $driver->filePutContents($this->directory->getAbsolutePath($csvFile), $exportContent);
+
         return $exportContent;
     }
 
@@ -301,12 +303,11 @@ private function importData($csvFile)
     {
         /** @var ImportAdvancedPricing $importModel */
         $importModel = $this->objectManager->create(ImportAdvancedPricing::class);
-        $directory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_DIR);
         $source = $this->objectManager->create(
             ImportSourceCsv::class,
             [
                 'file' => $csvFile,
-                'directory' => $directory
+                'directory' => $this->directory
             ]
         );
         $errors = $importModel->setParameters(
@@ -366,4 +367,24 @@ private function removeImportedProducts(array $skus): void
         $registry->unregister('isSecureArea');
         $registry->register('isSecureArea', false);
     }
+
+    /**
+     * Appends csv data to the file
+     *
+     * @param string $filePath
+     * @param array $csv
+     * @return void
+     */
+    private function updateCsvFile(string $filePath, array $csv): void
+    {
+        $driver = $this->directory->getDriver();
+        $driver->deleteFile($filePath);
+        $fileResource = $driver->fileOpen($filePath, 'w');
+
+        foreach ($csv as $dataRow) {
+            $driver->filePutCsv($fileResource, $dataRow);
+        }
+
+        $driver->fileClose($fileResource);
+    }
 }

dev/tests/integration/testsuite/Magento/AdvancedPricingImportExport/Model/Import/AdvancedPricingTest.php

@@ -175,6 +175,7 @@ public function testImportDelete()
         $index = 0;
         $ids = [];
         $origPricingData = [];
+        $skus = ['simple'];
         while (isset($skus[$index])) {
             $ids[$index] = $productRepository->get($skus[$index])->getId();
             $origPricingData[$index] = $this->objectManager->create(\Magento\Catalog\Model\Product::class)
@@ -192,12 +193,16 @@ public function testImportDelete()
         $exportModel->setWriter(
             \Magento\TestFramework\Helper\Bootstrap::getObjectManager()->create(
                 \Magento\ImportExport\Model\Export\Adapter\Csv::class,
-                ['fileSystem' => $this->fileSystem, 'destination' => $csvfile]
+                ['fileSystem' => $this->fileSystem]
             )
         );
-        $this->assertNotEmpty($exportModel->export());
+        $exportContent = $exportModel->export();
+        $this->assertNotEmpty($exportContent);
 
-        $errors = $this->doImport($csvfile, DirectoryList::VAR_DIR, Import::BEHAVIOR_DELETE, true);
+        $directory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_IMPORT_EXPORT);
+        $directory->getDriver()->filePutContents($directory->getAbsolutePath($csvfile), $exportContent);
+
+        $errors = $this->doImport($csvfile, DirectoryList::VAR_IMPORT_EXPORT, Import::BEHAVIOR_DELETE, true);
 
         $this->assertTrue(
             $errors->getErrorsCount() == 0,