MAGETWO-38138: Stabilize story

 - fix unit test

Author: nsyvokonenko

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

@@ -335,7 +335,8 @@ protected function _getModelMockForTestImportDataWithCustomBehaviour()
             ],
         ];
         $updateResult = [
-            'entity_row' => $this->_customBehaviour['update_id'],
+            'entity_row_new' => [],
+            'entity_row_update' => $this->_customBehaviour['update_id'],
             'attributes' => [],
             'defaults' => [],
         ];
@@ -678,13 +679,15 @@ public function testImportDataWithCustomBehaviour()
     /**
      * Validation method for _saveAddressEntities (callback for _saveAddressEntities)
      *
-     * @param array $addUpdateRows
+     * @param array $addRows
+     * @param array $updateRows
      * @return Address|\PHPUnit_Framework_MockObject_MockObject
      */
-    public function validateSaveAddressEntities(array $addUpdateRows)
+    public function validateSaveAddressEntities(array $addRows, array $updateRows)
     {
-        $this->assertCount(1, $addUpdateRows);
-        $this->assertContains($this->_customBehaviour['update_id'], $addUpdateRows);
+        $this->assertCount(0, $addRows);
+        $this->assertCount(1, $updateRows);
+        $this->assertContains($this->_customBehaviour['update_id'], $updateRows);
         return $this->_model;
     }