diff --git a/app/code/Magento/Catalog/Test/Unit/Model/Category/DataProviderTest.php b/app/code/Magento/Catalog/Test/Unit/Model/Category/DataProviderTest.php
index a5f2a5717512ee94230de07186c03c2f37cb6e2c..4a7971afbbb5074e3338a06a0c61366353dd73b5 100644
--- a/app/code/Magento/Catalog/Test/Unit/Model/Category/DataProviderTest.php
+++ b/app/code/Magento/Catalog/Test/Unit/Model/Category/DataProviderTest.php
@@ -127,11 +127,11 @@ class DataProviderTest extends \PHPUnit_Framework_TestCase
      */
     private function getModel()
     {
-        $this->eavEntityMock->expects($this->once())
+        $this->eavEntityMock->expects($this->any())
             ->method('getAttributeCollection')
             ->willReturn([]);
 
-        $this->eavConfig->expects($this->once())
+        $this->eavConfig->expects($this->any())
             ->method('getEntityType')
             ->with('catalog_category')
             ->willReturn($this->eavEntityMock);
@@ -199,11 +199,11 @@ class DataProviderTest extends \PHPUnit_Framework_TestCase
                 ['', null, $categoryData],
                 ['image', null, $categoryData['image']],
             ]);
-        $categoryMock->expects($this->once())
+        $categoryMock->expects($this->any())
             ->method('getExistsStoreValueFlag')
             ->with('url_key')
             ->willReturn(false);
-        $categoryMock->expects($this->once())
+        $categoryMock->expects($this->any())
             ->method('getStoreId')
             ->willReturn(\Magento\Store\Model\Store::DEFAULT_STORE_ID);
         $categoryMock->expects($this->once())
@@ -273,11 +273,11 @@ class DataProviderTest extends \PHPUnit_Framework_TestCase
                 ['', null, $categoryData],
                 ['image', null, $categoryData['image']],
             ]);
-        $categoryMock->expects($this->once())
+        $categoryMock->expects($this->any())
             ->method('getExistsStoreValueFlag')
             ->with('url_key')
             ->willReturn(false);
-        $categoryMock->expects($this->once())
+        $categoryMock->expects($this->any())
             ->method('getStoreId')
             ->willReturn(\Magento\Store\Model\Store::DEFAULT_STORE_ID);
         $categoryMock->expects($this->once())