1 files changed, 214 insertions, 0 deletions
diff --git a/bin/reevotech/vendor/addwiki/mediawiki-api-base/tests/Unit/Guzzle/MiddlewareFactoryTest.php b/bin/reevotech/vendor/addwiki/mediawiki-api-base/tests/Unit/Guzzle/MiddlewareFactoryTest.php
new file mode 100644
index 00000000..1cf7270a
--- /dev/null
+++ b/bin/reevotech/vendor/addwiki/mediawiki-api-base/tests/Unit/Guzzle/MiddlewareFactoryTest.php
@@ -0,0 +1,214 @@
+<?php
+
+namespace Mediawiki\Api\Test\Unit\Guzzle;
+
+use GuzzleHttp\Client;
+use GuzzleHttp\Exception\ConnectException;
+use GuzzleHttp\Handler\MockHandler;
+use GuzzleHttp\HandlerStack;
+use GuzzleHttp\Psr7\Request;
+use GuzzleHttp\Psr7\Response;
+use Mediawiki\Api\Guzzle\MiddlewareFactory;
+
+/**
+ * @author Addshore
+ *
+ * @todo test interaction with logger
+ *
+ * @covers Mediawiki\Api\Guzzle\MiddlewareFactory
+ */
+class MiddlewareFactoryTest extends \PHPUnit_Framework_TestCase {
+
+	public function testRetriesConnectException() {
+		$queue = [
+			new ConnectException( 'Error 1', new Request( 'GET', 'test' ) ),
+			new Response( 200, [ 'X-Foo' => 'Bar' ] ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( [ 1000 ], $delays );
+	}
+
+	public function testRetries500Errors() {
+		$queue = [
+			new Response( 500 ),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( [ 1000 ], $delays );
+	}
+
+	public function testRetriesSomeMediawikiApiErrorHeaders() {
+		$queue = [
+			new Response( 200, [ 'mediawiki-api-error' => 'ratelimited' ] ),
+			new Response( 200, [ 'mediawiki-api-error' => 'maxlag' ] ),
+			new Response( 200, [ 'mediawiki-api-error' => 'readonly' ] ),
+			new Response( 200, [ 'mediawiki-api-error' => 'internal_api_error_DBQueryError' ] ),
+			new Response( 200, [ 'mediawiki-api-error' => 'DoNotRetryThisHeader' ] ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals(
+			[ 'DoNotRetryThisHeader' ],
+			$response->getHeader( 'mediawiki-api-error' )
+		);
+		$this->assertEquals( [ 1000, 2000, 3000, 4000 ], $delays );
+	}
+
+	public function testRetryAntiAbuseMeasure() {
+		$antiAbusejson = json_encode(
+			[
+				'error' => [
+					'info' => 'anti-abuse measure'
+				]
+			]
+		);
+
+		$queue = [
+			new Response( 200, [ 'mediawiki-api-error' => 'failed-save' ], $antiAbusejson ),
+			new Response( 200, [ 'mediawiki-api-error' => 'DoNotRetryThisHeader' ] ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( 'DoNotRetryThisHeader', $response->getHeaderLine( 'mediawiki-api-error' ) );
+	}
+
+	public function testRetryLimit() {
+		$queue = [
+			new ConnectException( 'Error 1', new Request( 'GET', 'test' ) ),
+			new ConnectException( 'Error 2', new Request( 'GET', 'test' ) ),
+			new ConnectException( 'Error 3', new Request( 'GET', 'test' ) ),
+			new ConnectException( 'Error 4', new Request( 'GET', 'test' ) ),
+			new ConnectException( 'Error 5', new Request( 'GET', 'test' ) ),
+			new ConnectException( 'Error 6', new Request( 'GET', 'test' ) ),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue );
+
+		$this->setExpectedException(
+			'GuzzleHttp\Exception\ConnectException',
+			'Error 6'
+		);
+
+		$client->request( 'GET', '/' );
+	}
+
+	public function testConnectExceptionRetryDelay() {
+		$queue = [
+			new ConnectException( '+1 second delay', new Request( 'GET', 'test' ) ),
+			new ConnectException( '+2 second delay', new Request( 'GET', 'test' ) ),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( [ 1000, 2000 ], $delays );
+	}
+
+	public function testServerErrorRetryDelay() {
+		$queue = [
+			new Response( 500 ),
+			new Response( 503 ),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( [ 1000, 2000 ], $delays );
+	}
+
+	public function testRelativeRetryDelayHeaderRetryDelay() {
+		$queue = [
+			new Response( 200, [ 'mediawiki-api-error' => 'maxlag', 'retry-after' => 10 ] ),
+			new Response( 200 ),
+		];
+
+		$this->getClient( $queue, $delays )->request( 'GET', '/' );
+
+		$this->assertEquals( [ 10000 ], $delays );
+	}
+
+	public function testAbsoluteRetryDelayHeaderRetryDelay() {
+		$queue = [
+			new Response(
+				200,
+				[
+					'mediawiki-api-error' => 'maxlag',
+					'retry-after' => gmdate( DATE_RFC1123, time() + 600 ),
+				]
+			),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertCount( 1, $delays );
+		// Allow 5 second delay while running this test.
+		$this->assertGreaterThan( 600000 - 5000, $delays[0] );
+	}
+
+	public function testPastRetryDelayHeaderRetryDelay() {
+		$queue = [
+			new Response(
+				200,
+				[
+					'mediawiki-api-error' => 'maxlag',
+					'retry-after' => 'Fri, 31 Dec 1999 23:59:59 GMT',
+				]
+			),
+			new Response( 200 ),
+		];
+
+		$client = $this->getClient( $queue, $delays );
+		$response = $client->request( 'GET', '/' );
+
+		$this->assertEquals( 200, $response->getStatusCode() );
+		$this->assertEquals( [ 1000 ], $delays );
+	}
+
+	private function getClient( array $queue, &$delays = null ) {
+		$mock = new MockHandler( $queue );
+
+		$handler = HandlerStack::create( $mock );
+
+		$middlewareFactory = new MiddlewareFactory();
+		$handler->push( $middlewareFactory->retry() );
+
+		$delayMocker = $this->getDelayMocker( $delays );
+		$handler->push( $delayMocker );
+
+		return new Client( [ 'handler' => $handler ] );
+	}
+
+	private function getDelayMocker( &$delays ) {
+		return function ( callable $handler ) use ( &$delays ) {
+			return function ( $request, array $options ) use ( $handler, &$delays ) {
+				if ( isset( $options['delay'] ) ) {
+					$delays[] = $options['delay'];
+					unset( $options['delay'] );
+				}
+				return $handler( $request, $options );
+			};
+		};
+	}
+}