added \Countable interface to pagerfanta

whiteoctober · Nov 28, 2011 · 931bebe · 931bebe
1 parent bbbf256
commit 931bebe
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/Pagerfanta/Pagerfanta.php b/src/Pagerfanta/Pagerfanta.php
@@ -26,7 +26,7 @@
  *
  * @api
  */
-class Pagerfanta implements PagerfantaInterface, \IteratorAggregate
+class Pagerfanta implements PagerfantaInterface, \Countable, \IteratorAggregate
 {
     private $adapter;
     private $maxPerPage;
@@ -230,6 +230,18 @@ public function getNextPage()
         return $this->currentPage + 1;
     }
 
+    /**
+     * Implements the \Countable interface.
+     *
+     * Return integer The number of results.
+     *
+     * @api
+     */
+    public function count()
+    {
+        return $this->getNbResults();
+    }
+
     /**
      * Implements the \IteratorAggregate interface.
      *

diff --git a/tests/Pagerfanta/Tests/PagerfantaTest.php b/tests/Pagerfanta/Tests/PagerfantaTest.php
@@ -258,4 +258,15 @@ public function testHasGetPreviousNextPage()
             $this->assertInstanceOf('Pagerfanta\Exception\LogicException', $e);
         }
     }
+
+    public function testCountableInterface()
+    {
+        $this->adapter
+            ->expects($this->any())
+            ->method('getNbResults')
+            ->will($this->returnValue(100))
+        ;
+
+        $this->assertSame(100, count($this->pagerfanta));
+    }
 }