Merge branch 'MAGNIMEX-156-units' into 'master'

Magnimex 156 units

See merge request !174

Author: kalashnikau-uladzimir

app/code/Magento/AdvancedPricingImportExport/Test/Unit/Model/Export/AdvancedPricingTest.php

@@ -5,7 +5,6 @@
  */
 namespace Magento\AdvancedPricingImportExport\Test\Unit\Model\Export;
 
-use Magento\Framework\TestFramework\Unit\Helper\ObjectManager as ObjectManagerHelper;
 use \Magento\Store\Model\Store;
 
 /**
@@ -98,8 +97,14 @@ class ExportTest extends \PHPUnit_Framework_TestCase
      */
     protected $rowCustomizer;
 
+    /**
+     * @var \Magento\CatalogImportExport\Model\Import\Product\StoreResolver|\PHPUnit_Framework_MockObject_MockObject
+     */
     protected $storeResolver;
 
+    /**
+     * @var \Magento\Customer\Api\GroupRepositoryInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
     protected $groupRepository;
 
     /**
@@ -112,14 +117,14 @@ class ExportTest extends \PHPUnit_Framework_TestCase
      */
     protected $advancedPricing;
 
-    protected $_entityTypeCode;
-
-
     /**
      * @var StubProduct|\Magento\CatalogImportExport\Model\Export\Product
      */
     protected $object;
 
+    /**
+     * Set Up
+     */
     protected function setUp()
     {
         $this->localeDate = $this->getMock(
@@ -204,9 +209,12 @@ protected function setUp()
             '',
             false
         );
-        $this->_setColFactory = $this->getMock(
-            '\Magento\Eav\Model\Resource\Entity\Attribute\Set\CollectionFactory',
-            ['create'],
+        $this->attrSetColFactory = $this->getMock(
+            'Magento\Eav\Model\Resource\Entity\Attribute\Set\CollectionFactory',
+            [
+                'create',
+                'setEntityTypeFilter',
+            ],
             [],
             '',
             false
@@ -327,7 +335,7 @@ protected function setUp()
             $this->collection,
             $this->exportConfig,
             $this->productFactory,
-            $this->_setColFactory,
+            $this->attrSetColFactory,
             $this->categoryColFactory,
             $this->itemFactory,
             $this->optionColFactory,
@@ -340,7 +348,10 @@ protected function setUp()
         );
     }
 
-    public function testExportCountZeroBreakInternalCalls()
+    /**
+     * Test export with zero condition
+     */
+    public function testExportZeroConditionCalls()
     {
         $page = 1;
         $itemsPerPage = 10;
@@ -364,7 +375,10 @@ public function testExportCountZeroBreakInternalCalls()
         $this->advancedPricing->export();
     }
 
-    public function testExportCurPageEqualToLastBreakInternalCalls()
+    /**
+     * Test export for current page
+     */
+    public function testExportCurrentPageCalls()
     {
         $curPage = $lastPage = $page = 1;
         $itemsPerPage = 10;
@@ -379,23 +393,25 @@ public function testExportCurPageEqualToLastBreakInternalCalls()
         $this->abstractCollection->expects($this->once())->method('getCurPage')->willReturn($curPage);
         $this->abstractCollection->expects($this->once())->method('getLastPageNumber')->willReturn($lastPage);
         $headers = ['headers'];
-        $this->advancedPricing->expects($this->once())->method('_getHeaderColumns')->willReturn($headers);
-        $this->writer->expects($this->once())->method('setHeaderCols')->with($headers);
+        $this->advancedPricing->expects($this->any())->method('_getHeaderColumns')->willReturn($headers);
+        $this->writer->expects($this->any())->method('setHeaderCols')->with($headers);
+        $webSite = 'All Websites [USD]';
+        $userGroup = 'General';
+        $this->advancedPricing->expects($this->any())->method('_getWebsiteCode')->willReturn($webSite);
+        $this->advancedPricing->expects($this->any())->method('_getCustomerGroupById')->willReturn($userGroup);
         $data = [
-            'sku' => 'simpletest',
-            'group_price_website' => '0',
-            'group_price_customer_group' => '1',
-            'group_price' => '100',
-            'tier_price_website' => '0',
-            'tier_price_customer_group' => '2',
-            'tier_price_qty' => '2',
-            'tier_price' => '23',
+            [
+                'sku' => 'simpletest',
+                'group_price_website' => $webSite,
+                'group_price_customer_group' => $userGroup,
+                'group_price' => '100',
+                'tier_price_website' => $webSite,
+                'tier_price_customer_group' => $userGroup,
+                'tier_price_qty' => '2',
+                'tier_price' => '23',
+            ]
         ];
         $this->advancedPricing->expects($this->once())->method('getExportData')->willReturn($data);
-        $webSite = 'All Websites [USD]';
-        $userGroup = 'General';
-        $this->advancedPricing->expects($this->once())->method('_getWebsiteCode')->willReturn($webSite);
-        $this->advancedPricing->expects($this->once())->method('_getCustomerGroupById')->willReturn($userGroup);
         $exportData = [
             'sku' => 'simpletest',
             'group_price_website' => $webSite,
@@ -406,11 +422,18 @@ public function testExportCurPageEqualToLastBreakInternalCalls()
             'tier_price_qty' => '2',
             'tier_price' => '23',
         ];
+        $this->advancedPricing
+            ->expects($this->any())
+            ->method('correctExportData')
+            ->willReturn($exportData);
         $this->writer->expects($this->once())->method('writeRow')->with($exportData);
         $this->writer->expects($this->once())->method('getContents');
         $this->advancedPricing->export();
     }
 
+    /**
+     * tearDown
+     */
     protected function tearDown()
     {
         unset($this->object);