1 files changed, 125 insertions, 0 deletions
diff --git a/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Mock/IteratorMockBuilder.php b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Mock/IteratorMockBuilder.php
new file mode 100644
index 00000000..721f4b36
--- /dev/null
+++ b/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Utils/Mock/IteratorMockBuilder.php
@@ -0,0 +1,125 @@
+<?php
+
+namespace SMW\Tests\Utils\Mock;
+
+use RuntimeException;
+
+/**
+ * Convenience mock builder for Iterator classes
+ * @group semantic-mediawiki
+ *
+ * @license GNU GPL v2+
+ * @since 2.0
+ *
+ * @author mwjames
+ */
+class IteratorMockBuilder extends \PHPUnit_Framework_TestCase {
+
+	private $iteratorClass;
+	private $items = [];
+	private $methods = [];
+	private $counter = 0;
+
+	/**
+	 * @since  2.0
+	 *
+	 * @param string $iteratorClass
+	 *
+	 * @return IteratorMockBuilder
+	 */
+	public function setClass( $iteratorClass ) {
+		$this->iteratorClass = $iteratorClass;
+		return $this;
+	}
+
+	/**
+	 * @since  2.0
+	 *
+	 * @param array $items
+	 *
+	 * @return IteratorMockBuilder
+	 */
+	public function with( array $items ) {
+		$this->items = $items;
+		return $this;
+	}
+
+	/**
+	 * @since  2.5
+	 *
+	 * @param array $methods
+	 *
+	 * @return IteratorMockBuilder
+	 */
+	public function setMethods( array $methods ) {
+		$this->methods = $methods;
+		return $this;
+	}
+
+	/**
+	 * @note When other methods called before the actual current/next then
+	 * set the counter to ensure the starting point matches the expected
+	 * InvokeCount.
+	 *
+	 * @since  2.5
+	 *
+	 * @param integer $num
+	 *
+	 * @return IteratorMockBuilder
+	 */
+	public function incrementInvokedCounterBy( $num ) {
+		$this->counter += $num;
+		return $this;
+	}
+
+	/**
+	 * @since  2.0
+	 *
+	 * @return PHPUnit_Framework_MockObject_MockObject
+	 * @throws RuntimeException
+	 */
+	public function getMockForIterator() {
+
+		$instance = $this->getMockBuilder( $this->iteratorClass )
+			->disableOriginalConstructor()
+			->setMethods( $this->methods )
+			->getMock();
+
+		if ( !$instance instanceof \Iterator ) {
+			throw new RuntimeException( "Instance is not an Iterator" );
+		}
+
+		$instance->expects( $this->at( $this->counter++ ) )
+			->method( 'rewind' );
+
+		foreach ( $this->items as $key => $value ) {
+
+			$instance->expects( $this->at( $this->counter++ ) )
+				->method( 'valid' )
+				->will( $this->returnValue( true ) );
+
+			$instance->expects( $this->at( $this->counter++ ) )
+				->method( 'current' )
+				->will( $this->returnValue( $value ) );
+
+			$instance->expects( $this->at( $this->counter++ ) )
+				->method( 'next' );
+		}
+
+		$instance->expects( $this->at( $this->counter++ ) )
+			->method( 'valid' )
+			->will( $this->returnValue( false ) );
+
+		return $instance;
+	}
+
+	/**
+	 * @since  2.0
+	 *
+	 * @return integer
+	 */
+	public function getLastCounter() {
+		return $this->counter;
+	}
+
+}