Have StatementByGuidMap implement IteratorAggregate

Author: JeroenDeDauw

src/Statement/StatementByGuidMap.php

@@ -2,7 +2,11 @@
 
 namespace Wikibase\DataModel\Statement;
 
+use ArrayIterator;
+use Countable;
 use InvalidArgumentException;
+use IteratorAggregate;
+use Traversable;
 
 /**
  * Ordered and unique collection of Statement objects.
@@ -14,7 +18,7 @@
  * @author Jeroen De Dauw < jeroendedauw@gmail.com >
  * @author Kai Nissen < kai.nissen@wikimedia.de >
  */
-class StatementByGuidMap implements \Countable {
+class StatementByGuidMap implements IteratorAggregate, Countable {
 
 	private $statements = array();
 
@@ -86,4 +90,12 @@ public function count() {
 		return count( $this->statements );
 	}
 
+	/**
+	 * @see IteratorAggregate::getIterator
+	 * @return Traversable
+	 */
+	public function getIterator() {
+		return new ArrayIterator( $this->statements );
+	}
+
 }

tests/unit/Statement/StatementByGuidMapTest.php

@@ -141,4 +141,24 @@ public function testMapCanBePassedToCount() {
 		$this->assertSame( 2, count( $statements ) );
 	}
 
+	public function testMapCanBeIteratedOver() {
+		$statement1 = $this->newStatement( 1, 'some guid' );
+		$statement2 = $this->newStatement( 2, 'other guid' );
+
+		$statementMap = new StatementByGuidMap( array( $statement1, $statement2 ) );
+
+		$iteratedStatements = array();
+
+		foreach ( $statementMap as $guid => $statement ) {
+			$iteratedStatements[$guid] = $statement;
+		}
+
+		$expectedStatements = array(
+			'some guid' => $statement1,
+			'other guid' => $statement2
+		);
+
+		$this->assertEquals( $expectedStatements, $iteratedStatements );
+	}
+
 }