MC-20423: [Integration Test] Import Table Rates to be used in Configuration Settings

Author: DianaRusin

dev/tests/integration/testsuite/Magento/OfflineShipping/Controller/Adminhtml/System/Config/ImportExportTableratesTest.php

@@ -9,10 +9,12 @@
 namespace Magento\OfflineShipping\Controller\Adminhtml\System\Config;
 
 use Magento\Framework\App\Request\Http as HttpRequest;
+use Magento\Framework\Filesystem\Directory\WriteInterface;
 use Magento\Framework\Message\MessageInterface;
 use Magento\Framework\View\LayoutInterface;
 use Magento\OfflineShipping\Block\Adminhtml\Carrier\Tablerate\Grid;
-use Magento\OfflineShipping\Model\ResourceModel\Carrier\Tablerate;
+use Magento\OfflineShipping\Model\ResourceModel\Carrier\Tablerate\Collection;
+use Magento\Store\Model\StoreManagerInterface;
 use Magento\TestFramework\ObjectManager;
 use Magento\TestFramework\Helper\Bootstrap;
 use Magento\Framework\Filesystem;
@@ -36,9 +38,14 @@ class ImportExportTableratesTest extends \Magento\TestFramework\TestCase\Abstrac
     private $fileSystem;
 
     /**
-     * @var DirectoryList
+     * @var StoreManagerInterface
      */
-    private $varDirectory;
+    private $storeManager;
+
+    /**
+     * @var int
+     */
+    private $websiteId;
 
     /**
      * @inheritdoc
@@ -47,7 +54,8 @@ protected function setUp()
     {
         $this->objectManager = Bootstrap::getObjectManager();
         $this->fileSystem = $this->objectManager->get(Filesystem::class);
-        $this->varDirectory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_DIR);
+        $this->storeManager = Bootstrap::getObjectManager()->get(StoreManagerInterface::class);
+        $this->websiteId = $this->storeManager->getWebsite()->getId();
 
         parent::setUp();
     }
@@ -85,34 +93,38 @@ public function testImportExportTablerates(): void
             ]
         )->setMethod(HttpRequest::METHOD_POST);
 
-        $this->dispatch('backend/admin/system_config/save/section/carriers/website/1/');
+        $this->dispatch('backend/admin/system_config/save/section/carriers/website/' . $this->websiteId . '/');
         $this->assertSessionMessages(
             $this->equalTo([(string)__('You saved the configuration.')]),
             MessageInterface::TYPE_SUCCESS
         );
 
-        $tablerateResourceModel = $this->objectManager->create(Tablerate::class);
-        $connection = $tablerateResourceModel->getConnection();
+        /** @var Collection $tablerateCollection */
+        $tablerateCollection = $this->objectManager->create(Collection::class);
+        $tablerateData = $tablerateCollection->setConditionFilter('package_weight')->getItems()[0]->getData();
+        $this->assertEquals('666.0000', $tablerateData['price']);
+        $this->assertEquals('USA', $tablerateData['dest_country']);
+        $this->assertEquals('10.0000', $tablerateData['condition_value']);
 
-        $selectData = $connection->select()->from($tablerateResourceModel->getTable('shipping_tablerate'));
-        $this->assertNotEmpty($connection->fetchRow($selectData));
-
-        $exportCsv = $this->getTablerateCsv();
-        $exportCsvContent = $this->varDirectory->openFile($exportCsv['value'], 'r')->readAll();
+        $exportCsvContent = $this->getTablerateCsv();
         $importCsvContent = $tmpDirectory->openFile($importCsvPath, 'r')->readAll();
 
         $this->assertEquals($importCsvContent, $exportCsvContent);
     }
 
     /**
-     * @return array
+     * @return string
      */
-    private function getTablerateCsv(): array
+    private function getTablerateCsv(): string
     {
+        /** @var WriteInterface $varDirectory */
+        $varDirectory = $this->fileSystem->getDirectoryWrite(DirectoryList::VAR_DIR);
+
         /** @var Grid $gridBlock */
         $gridBlock = $this->objectManager->get(LayoutInterface::class)->createBlock(Grid::class);
-        $exportCsv = $gridBlock->setWebsiteId(1)->setConditionName('package_weight')->getCsvFile();
+        $exportCsv = $gridBlock->setWebsiteId($this->websiteId)->setConditionName('package_weight')->getCsvFile();
+        $exportCsvContent = $varDirectory->openFile($exportCsv['value'], 'r')->readAll();
 
-        return $exportCsv;
+        return $exportCsvContent;
     }
 }

dev/tests/integration/testsuite/Magento/OfflineShipping/_files/tablerate_create_file_in_tmp.php

@@ -8,11 +8,13 @@
 use Magento\Framework\App\Filesystem\DirectoryList;
 use Magento\Framework\Filesystem;
 use Magento\TestFramework\Helper\Bootstrap;
+use Magento\Framework\Filesystem\Directory\WriteInterface;
 
 $importCsv = 'tablerates.csv';
 $objectManager = Bootstrap::getObjectManager();
+/** @var Filesystem $fileSystem */
 $fileSystem = $objectManager->get(Filesystem::class);
-
+/** @var WriteInterface $tmpDirectory */
 $tmpDirectory = $fileSystem->getDirectoryWrite(DirectoryList::SYS_TMP);
 $importCsvPath = $tmpDirectory->getAbsolutePath($importCsv);
 

dev/tests/integration/testsuite/Magento/OfflineShipping/_files/tablerate_create_file_in_tmp_rollback.php

@@ -8,10 +8,13 @@
 use Magento\Framework\App\Filesystem\DirectoryList;
 use Magento\Framework\Filesystem;
 use Magento\TestFramework\Helper\Bootstrap;
+use Magento\Framework\Filesystem\Directory\WriteInterface;
 
 $objectManager = Bootstrap::getObjectManager();
+/** @var Filesystem $fileSystem */
 $fileSystem = $objectManager->get(Filesystem::class);
+/** @var WriteInterface $tmpDirectory */
 $tmpDirectory = $fileSystem->getDirectoryWrite(DirectoryList::SYS_TMP);
 $fileName = 'tablerates.csv';
 
-unlink($tmpDirectory->getAbsolutePath($fileName));
+$tmpDirectory->delete($fileName);