class WatchedItemQueryServiceUnitTest extends PHPUnit_Framework_TestCase {
/**
- * @return PHPUnit_Framework_MockObject_MockObject|DatabaseBase
+ * @return PHPUnit_Framework_MockObject_MockObject|Database
*/
private function getMockDb() {
- $mock = $this->getMockBuilder( DatabaseBase::class )
+ $mock = $this->getMockBuilder( Database::class )
->disableOriginalConstructor()
->getMock();
->disableOriginalConstructor()
->getMock();
$mock->expects( $this->any() )
- ->method( 'getConnection' )
+ ->method( 'getConnectionRef' )
->with( DB_SLAVE )
->will( $this->returnValue( $mockDb ) );
return $mock;