Add the Countable interface to RouteCollection.

Crell · Crell · commit c3509446c119 · 2012-06-23T12:00:40.000-05:00
diff --git a/src/Symfony/Component/Routing/RouteCollection.php b/src/Symfony/Component/Routing/RouteCollection.php
@@ -24,7 +24,7 @@
  *
  * @api
  */
-class RouteCollection implements \IteratorAggregate
+class RouteCollection implements \IteratorAggregate, \Countable
 {
     private $routes;
     private $resources;
@@ -88,6 +88,16 @@ public function getIterator()
         return new \ArrayIterator($this->routes);
     }
 
+    /**
+     * Gets the number of Routes in this collection.
+     *
+     * @return int The number of routes in this collection, including nested collections
+     */
+    public function count()
+    {
+        return count($this->routes->all());
+    }
+
     /**
      * Adds a route.
      *
diff --git a/src/Symfony/Component/Routing/Tests/RouteCollectionTest.php b/src/Symfony/Component/Routing/Tests/RouteCollectionTest.php
@@ -76,6 +76,18 @@ public function testIteratorWithOverridenRoutes()
         $this->assertEquals('/foo1', $this->getFirstNamedRoute($collection, 'foo')->getPattern());
     }
 
+    public function testCount()
+    {
+        $collection = new RouteCollection();
+        $collection->add('foo', new Route('/foo'));
+
+        $collection1 = new RouteCollection();
+        $collection->addCollection($collection1);
+        $collection1->add('foo', new Route('/foo1'));
+
+        $this->assertCount(2, $collection);
+    }
+
     protected function getFirstNamedRoute(RouteCollection $routeCollection, $name)
     {
         foreach ($routeCollection as $key => $route) {

Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`*`
`25`	`25`	`* @api`
`26`	`26`	`*/`
`27`		`-class RouteCollection implements \IteratorAggregate`
	`27`	`+class RouteCollection implements \IteratorAggregate, \Countable`
`28`	`28`	`{`
`29`	`29`	`private $routes;`
`30`	`30`	`private $resources;`
`@@ -88,6 +88,16 @@ public function getIterator()`
`88`	`88`	`return new \ArrayIterator($this->routes);`
`89`	`89`	`}`
`90`	`90`
	`91`	`+ /**`
	`92`	`+ * Gets the number of Routes in this collection.`
	`93`	`+ *`
	`94`	`+ * @return int The number of routes in this collection, including nested collections`
	`95`	`+ */`
	`96`	`+ public function count()`
	`97`	`+ {`
	`98`	`+ return count($this->routes->all());`
	`99`	`+ }`
	`100`	`+`
`91`	`101`	`/**`
`92`	`102`	`* Adds a route.`
`93`	`103`	`*`