1 files changed, 74 insertions, 0 deletions
diff --git a/CheckUser/tests/phpunit/TokenQueryManagerTest.php b/CheckUser/tests/phpunit/TokenQueryManagerTest.php
new file mode 100644
index 00000000..762daed0
--- /dev/null
+++ b/CheckUser/tests/phpunit/TokenQueryManagerTest.php
@@ -0,0 +1,74 @@
+<?php
+
+namespace MediaWiki\CheckUser\Tests;
+
+use FauxRequest;
+use MediaWiki\CheckUser\TokenManager;
+use MediaWiki\CheckUser\TokenQueryManager;
+use MediaWikiIntegrationTestCase;
+
+/**
+ * Test class for TokenQueryManager class
+ *
+ * @group CheckUser
+ *
+ * @covers \MediaWiki\CheckUser\TokenQueryManager
+ */
+class TokenQueryManagerTest extends MediaWikiIntegrationTestCase {
+
+	/**
+	 * @return MockObject|TokenManager
+	 */
+	private function getMockTokenManager() {
+		return $this->getMockBuilder( TokenManager::class )
+			->disableOriginalConstructor()->getMock();
+	}
+
+	public function testUpdateToken() {
+		$tokenManager = $this->getMockTokenManager();
+		$tokenQueryManager = $this->getMockBuilder( TokenQueryManager::class )
+			->setConstructorArgs( [ $tokenManager ] )
+			->setMethods( [ 'getDataFromRequest' ] )
+			->getMock();
+
+		$tokenData = [ 'foo' => true, 'bar' => false,  'baz' => 'test' ];
+
+		$tokenQueryManager->method( 'getDataFromRequest' )->willReturn( $tokenData );
+		$tokenManager->expects( $this->once() )
+			->method( 'encode' )
+			->with( $this->anything(), [ 'bar' => true, 'baz' => 'test' ] );
+
+		$tokenQueryManager->updateToken( new FauxRequest(), [ 'foo' => null, 'bar' => true ] );
+	}
+
+	public function testGetDataFromRequest() {
+		$request = new FauxRequest( [ 'token' => 'token' ] );
+
+		$tokenManager = $this->getMockTokenManager();
+		$tokenManager->expects( $this->once() )->method( 'decode' )->with( $this->anything(), 'token' );
+
+		$tokenQueryManager = new TokenQueryManager( $tokenManager );
+		$tokenQueryManager->getDataFromRequest( $request );
+	}
+
+	public function testGetDataFromRequestWithNoToken() {
+		$request = new FauxRequest();
+
+		$tokenManager = $this->getMockTokenManager();
+		$tokenQueryManager = new TokenQueryManager( $tokenManager );
+		$data = $tokenQueryManager->getDataFromRequest( $request );
+
+		$this->assertSame( [], $data );
+	}
+
+	public function testGetDataFromRequestHandlesDecodeException() {
+		$tokenManager = $this->getMockTokenManager();
+		$tokenManager->method( 'decode' )->willThrowException( new \Exception() );
+
+		$tokenQueryManager = new TokenQueryManager( $tokenManager );
+		$request = new FauxRequest( [ 'token' => 'token' ] );
+		$data = $tokenQueryManager->getDataFromRequest( $request );
+
+		$this->assertSame( [], $data );
+	}
+}