From 8cd3b859b8948ce295e35bdf8c93939b2539666b Mon Sep 17 00:00:00 2001
From: addshore <addshorewiki@gmail.com>
Date: Mon, 2 May 2016 14:56:53 +0100
Subject: [PATCH] Add tests for MediawikiTestCase::setService stuff

Change-Id: If689d85f42ae26e786d599f59e76d3ebdfa22212
---
 tests/phpunit/tests/MediaWikiTestCaseTest.php | 31 +++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/tests/phpunit/tests/MediaWikiTestCaseTest.php b/tests/phpunit/tests/MediaWikiTestCaseTest.php
index 955dd2fb67..5d2f37e6a6 100644
--- a/tests/phpunit/tests/MediaWikiTestCaseTest.php
+++ b/tests/phpunit/tests/MediaWikiTestCaseTest.php
@@ -1,5 +1,6 @@
 <?php
 use MediaWiki\Logger\LoggerFactory;
+use MediaWiki\MediaWikiServices;
 use Psr\Log\LoggerInterface;
 
 /**
@@ -99,6 +100,36 @@ class MediaWikiTestCaseTest extends MediaWikiTestCase {
 		$this->stashMwGlobals( self::GLOBAL_KEY_NONEXISTING );
 	}
 
+	public function testOverrideMwServices() {
+		$initialServices = MediaWikiServices::getInstance();
+
+		$this->overrideMwServices();
+		$this->assertNotSame( $initialServices, MediaWikiServices::getInstance() );
+
+		$this->tearDown();
+		$this->assertSame( $initialServices, MediaWikiServices::getInstance() );
+	}
+
+	public function testSetService() {
+		$initialServices = MediaWikiServices::getInstance();
+		$initialService = $initialServices->getDBLoadBalancer();
+		$mockService = $this->getMockBuilder( LoadBalancer::class )
+			->disableOriginalConstructor()->getMock();
+
+		$this->setService( 'DBLoadBalancer', $mockService );
+		$this->assertNotSame( $initialServices, MediaWikiServices::getInstance() );
+		$this->assertNotSame(
+			$initialService,
+			MediaWikiServices::getInstance()->getDBLoadBalancer()
+		);
+		$this->assertSame( $mockService, MediaWikiServices::getInstance()->getDBLoadBalancer() );
+
+		$this->tearDown();
+		$this->assertSame( $initialServices, MediaWikiServices::getInstance() );
+		$this->assertNotSame( $mockService, MediaWikiServices::getInstance()->getDBLoadBalancer() );
+		$this->assertSame( $initialService, MediaWikiServices::getInstance()->getDBLoadBalancer() );
+	}
+
 	/**
 	 * @covers MediaWikiTestCase::setLogger
 	 * @covers MediaWikiTestCase::restoreLogger
-- 
2.20.1