Fixed AutoRouteManagerTest

Author: dantleech

Tests/Unit/AutoRoute/AutoRouteManagerTest.php

@@ -9,7 +9,7 @@ class AutoRouteManagerTest extends \PHPUnit_Framework_TestCase
 {
     public function setUp()
     {
-        $this->driver = $this->getMock('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\Driver\DriverInterface');
+        $this->driver = $this->getMock('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\Adapter\AdapterInterface');
         $this->urlGenerator = $this->getMock('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\UrlGeneratorInterface');
         $this->defunctRouteHandler = $this->getMock('Symfony\Cmf\Bundle\RoutingAutoBundle\AutoRoute\DefunctRouteHandlerInterface');
         $this->autoRouteManager = new AutoRouteManager(
@@ -58,26 +58,27 @@ public function testBuildUrlContextCollection($params)
         $document = new \stdClass;
 
         for ($i = 0; $i < $localesCount; $i++) {
-            $expectedRoutes[] = $this->getMock('Symfony\Cmf\Component\Routing\RouteObjectInterface');
+            $expectedRoutes[] = $this->getMock('Symfony\Cmf\Bundle\RoutingAutoBundle\Model\AutoRouteInterface');
 
             $this->urlGenerator->expects($this->exactly($localesCount))
                 ->method('generateUrl')
-                ->with($document)
                 ->will($this->returnCallback(function () use ($i, $indexedUrls) {
                     return $indexedUrls[$i];
                 }));
-
-            $this->driver->expects($this->exactly($localesCount))
-                ->method('createAutoRoute')
-                ->will($this->returnCallback(function ($url, $document) use ($i, $expectedRoutes) {
-                    return $expectedRoutes[$i];
-                }));
         }
 
-        $urlContextCollection = new UrlContextCollection();
-        $this->autoRouteManager->buildUrlContextCollection($urlContextCollection, $document);
+        $this->driver->expects($this->exactly($localesCount))
+            ->method('createAutoRoute')
+            ->will($this->returnCallback(function ($url, $document) use ($expectedRoutes) {
+                static $i = 0;
+                return $expectedRoutes[$i++];
+            }));
 
-        $res = $urlContextCollection->getPersistStack();
-        $this->assertEquals($expectedRoutes, $res);
+        $urlContextCollection = new UrlContextCollection($document);
+        $this->autoRouteManager->buildUrlContextCollection($urlContextCollection);
+
+        foreach ($expectedRoutes as $expectedRoute) {
+            $this->assertTrue($urlContextCollection->containsAutoRoute($expectedRoute), 'URL context collection contains route: ' . spl_object_hash($expectedRoute));
+        }
     }
 }