MAGETWO-54211: Downloadable product has direct link in category after update applied

Author: rganin

app/code/Magento/CatalogUrlRewrite/Test/Unit/Observer/ProductProcessUrlRewriteSavingObserverTest.php

@@ -1,5 +1,4 @@
 <?php
-
 /**
  * Copyright © 2016 Magento. All rights reserved.
  * See COPYING.txt for license details.
@@ -188,11 +187,12 @@ public function testExecuteUrlKey(
 
         $this->product->expects($this->any())
             ->method('dataHasChangedFor')
-            ->will($this->returnValueMap([
-                    ['visibility', $isChangedVisibility],
-                    ['url_key', $isChangedUrlKey]
-                ])
-            );
+            ->will($this->returnValueMap(
+                    [
+                        ['visibility', $isChangedVisibility],
+                        ['url_key', $isChangedUrlKey]
+                    ]
+                ));
 
         $this->product->expects($this->any())
             ->method('getIsChangedWebsites')
@@ -207,8 +207,15 @@ public function testExecuteUrlKey(
             UrlRewrite::ENTITY_TYPE => ProductUrlRewriteGenerator::ENTITY_TYPE,
             UrlRewrite::REDIRECT_TYPE => 0,
         ]);
-        $this->product->expects($this->any())->method('isVisibleInSiteVisibility')->will($this->returnValue($visibilityResult));
-        $this->urlPersist->expects($this->exactly($expectedReplaceCount))->method('replace')->with([3 => 'rewrite']);
+
+        $this->product->expects($this->any())
+            ->method('isVisibleInSiteVisibility')
+            ->will($this->returnValue($visibilityResult));
+
+        $this->urlPersist->expects($this->exactly($expectedReplaceCount))
+            ->method('replace')
+            ->with([3 => 'rewrite']);
+
         $this->model->execute($this->observer);
     }
 }