MAGETWO-48933: use cookies to identify preview version

Author: Dmytro Voskoboinikov

lib/internal/Magento/Framework/Test/Unit/UrlTest.php

@@ -19,6 +19,11 @@ class UrlTest extends \PHPUnit_Framework_TestCase
      */
     protected $routeParamsResolverMock;
 
+    /**
+     * @var \Magento\Framework\Url\RouteParamsPreprocessorInterface|\PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $routeParamsPreprocessorMock;
+
     /**
      * @var \Magento\Framework\Url\ScopeResolverInterface|\PHPUnit_Framework_MockObject_MockObject
      */
@@ -58,6 +63,17 @@ protected function setUp()
             '',
             false
         );
+
+        $this->routeParamsPreprocessorMock = $this->getMockForAbstractClass(
+            'Magento\Framework\Url\RouteParamsPreprocessorInterface',
+            [],
+            '',
+            false,
+            true,
+            true,
+            []
+        );
+
         $this->scopeResolverMock = $this->getMock('Magento\Framework\Url\ScopeResolverInterface');
         $this->scopeMock = $this->getMock('Magento\Framework\Url\ScopeInterface');
         $this->queryParamsResolverMock = $this->getMock(
@@ -190,7 +206,9 @@ public function testGetUrl($query, $queryResult, $returnUri)
                 'scopeResolver' => $this->scopeResolverMock,
                 'routeParamsResolverFactory' => $this->getRouteParamsResolverFactory(),
                 'queryParamsResolver' => $this->queryParamsResolverMock,
-                'request' => $requestMock, 'routeConfig' => $routeConfigMock,
+                'request' => $requestMock,
+                'routeConfig' => $routeConfigMock,
+                'routeParamsPreprocessor' => $this->routeParamsPreprocessorMock
             ]
         );
 
@@ -204,6 +222,11 @@ public function testGetUrl($query, $queryResult, $returnUri)
         $this->routeParamsResolverMock->expects($this->any())->method('getType')->will($this->returnValue($urlType));
         $this->routeParamsResolverMock->expects($this->any())->method('getRouteParams')
             ->will($this->returnValue(['id' => 100]));
+
+        $this->routeParamsPreprocessorMock->expects($this->once())
+            ->method('execute')
+            ->willReturnArgument(2);
+
         $requestMock->expects($this->once())->method('isDirectAccessFrontendName')->will($this->returnValue(true));
         $routeConfigMock->expects($this->once())->method('getRouteFrontName')->will($this->returnValue('catalog'));
         $this->queryParamsResolverMock->expects($this->once())->method('getQuery')
@@ -272,12 +295,18 @@ public function testGetUrlRouteUseRewrite()
     {
         $this->routeParamsResolverMock->expects($this->any())->method('getRouteParams')
             ->will($this->returnValue(['foo' => 'bar']));
+
+        $this->routeParamsPreprocessorMock->expects($this->once())
+            ->method('execute')
+            ->willReturnArgument(2);
+
         $request = $this->getRequestMock();
         $request->expects($this->once())->method('getAlias')->will($this->returnValue('/catalog/product/view/'));
         $model = $this->getUrlModel([
             'scopeResolver' => $this->scopeResolverMock,
             'routeParamsResolverFactory' => $this->getRouteParamsResolverFactory(),
             'request' => $request,
+            'routeParamsPreprocessor' => $this->routeParamsPreprocessorMock
         ]);
 
         $this->scopeResolverMock->expects($this->any())
@@ -345,7 +374,9 @@ public function testGetDirectUrl()
                 'scopeResolver' => $this->scopeResolverMock,
                 'routeParamsResolverFactory' => $this->getRouteParamsResolverFactory(),
                 'queryParamsResolver' => $this->queryParamsResolverMock,
-                'request' => $requestMock, 'routeConfig' => $routeConfigMock,
+                'request' => $requestMock,
+                'routeConfig' => $routeConfigMock,
+                'routeParamsPreprocessor' => $this->routeParamsPreprocessorMock
             ]
         );
 
@@ -357,6 +388,11 @@ public function testGetDirectUrl()
             ->method('getScope')
             ->will($this->returnValue($this->scopeMock));
         $this->routeParamsResolverMock->expects($this->any())->method('getType')->will($this->returnValue($urlType));
+
+        $this->routeParamsPreprocessorMock->expects($this->once())
+            ->method('execute')
+            ->willReturnArgument(2);
+
         $requestMock->expects($this->once())->method('isDirectAccessFrontendName')->will($this->returnValue(true));
 
         $url = $model->getDirectUrl('direct-url');