MAGETWO-90808: [Performance] Customer Import check data does not complete

Author: zakdma

app/code/Magento/CustomerImportExport/Test/Unit/Model/Import/CustomerCompositeTest.php

@@ -232,6 +232,7 @@ protected function _getModelMockForImportData($isDeleteBehavior, $customerImport
         if ($isDeleteBehavior || !$customerImport) {
             $addressEntity->expects($this->never())->method('importData');
         } else {
+            $addressEntity->expects($this->once())->method('setCustomerAttributes')->will($this->returnSelf());
             $addressEntity->expects($this->once())->method('importData')->will($this->returnValue($addressImport));
         }
 
@@ -654,7 +655,6 @@ public function dataProviderTestImportData()
      */
     public function testImportData($behavior, $customerImport, $addressImport, $result)
     {
-        return;
         $isDeleteBehavior = $behavior == Import::BEHAVIOR_DELETE;
         $entityMock = $this->_getModelMockForImportData($isDeleteBehavior, $customerImport, $addressImport);
         $entityMock->setParameters(['behavior' => $behavior]);