Re-adding the ability to add a resource to the RouteCollectionBuilder

weaverryan · weaverryan · commit 549a3e271838 · 2015-10-27T20:34:35.000-04:00
We need this because when you import a RouteCollection and this has a resource
on it, we need to pass that resource forward to the RouteCollectionBuilder
diff --git a/src/Symfony/Component/Routing/RouteCollectionBuilder.php b/src/Symfony/Component/Routing/RouteCollectionBuilder.php
@@ -13,6 +13,7 @@
 
 use Symfony\Component\Config\Exception\FileLoaderLoadException;
 use Symfony\Component\Config\Loader\LoaderInterface;
+use Symfony\Component\Config\Resource\ResourceInterface;
 
 /**
  * Helps add and import routes into a RouteCollection.
@@ -35,6 +36,7 @@ class RouteCollectionBuilder
     private $options = array();
     private $schemes;
     private $methods;
+    private $resources = array();
 
     /**
      * @param LoaderInterface $loader
@@ -237,6 +239,20 @@ public function setMethods($methods)
         return $this;
     }
 
+    /**
+     * Adds a resource for this collection.
+     *
+     * @param ResourceInterface $resource
+     *
+     * @return $this
+     */
+    private function addResource(ResourceInterface $resource)
+    {
+        $this->resources[] = $resource;
+
+        return $this;
+    }
+
     /**
      * Creates the final ArrayCollection, returns it, and clears everything.
      *
@@ -291,6 +307,10 @@ public function build()
 
                 $routeCollection->addCollection($subCollection);
             }
+
+            foreach ($this->resources as $resource) {
+                $routeCollection->addResource($resource);
+            }
         }
 
         return $routeCollection;
diff --git a/src/Symfony/Component/Routing/Tests/RouteCollectionBuilderTest.php b/src/Symfony/Component/Routing/Tests/RouteCollectionBuilderTest.php
@@ -11,6 +11,7 @@
 
 namespace Symfony\Component\Routing\Tests;
 
+use Symfony\Component\Config\Resource\FileResource;
 use Symfony\Component\Routing\Route;
 use Symfony\Component\Routing\RouteCollection;
 use Symfony\Component\Routing\RouteCollectionBuilder;
@@ -29,6 +30,7 @@ public function testImport()
         $originalRoute = new Route('/foo/path');
         $expectedCollection = new RouteCollection();
         $expectedCollection->add('one_test_route', $originalRoute);
+        $expectedCollection->addResource(new FileResource('file_resource.yml'));
 
         $resolvedLoader
             ->expects($this->once())
@@ -52,6 +54,8 @@ public function testImport()
         $addedCollection = $importedRoutes->build();
         $route = $addedCollection->get('one_test_route');
         $this->assertSame($originalRoute, $route);
+        // should return file_resource.yml, which is in the original collection
+        $this->assertCount(1, $addedCollection->getResources());
     }
 
     /**
