4 files changed, 8 insertions, 1 deletions
diff --git a/tests/parser/DbTestRecorder.php b/tests/parser/DbTestRecorder.php
index 04a2a8d44722..f68f5953d1f6 100644
--- a/tests/parser/DbTestRecorder.php
+++ b/tests/parser/DbTestRecorder.php
@@ -19,11 +19,14 @@
  * @ingroup Testing
  */
 
+use Wikimedia\Rdbms\IMaintainableDatabase;
+
 class DbTestRecorder extends TestRecorder {
 	public $version;
+	/** @var Database */
 	private $db;
 
-	public function __construct( IDatabase $db ) {
+	public function __construct( IMaintainableDatabase $db ) {
 		$this->db = $db;
 	}
 
diff --git a/tests/parser/ParserTestRunner.php b/tests/parser/ParserTestRunner.php
index 35c2480d0221..f100411f253c 100644
--- a/tests/parser/ParserTestRunner.php
+++ b/tests/parser/ParserTestRunner.php
@@ -25,6 +25,7 @@
  * @file
  * @ingroup Testing
  */
+use Wikimedia\Rdbms\IDatabase;
 use MediaWiki\MediaWikiServices;
 use Wikimedia\ScopedCallback;
 
diff --git a/tests/phpunit/includes/db/DatabaseTest.php b/tests/phpunit/includes/db/DatabaseTest.php
index 606a20989cc5..45791e2424f1 100644
--- a/tests/phpunit/includes/db/DatabaseTest.php
+++ b/tests/phpunit/includes/db/DatabaseTest.php
@@ -1,5 +1,7 @@
 <?php
 
+use Wikimedia\Rdbms\IDatabase;
+
 /**
  * @group Database
  * @group Database
diff --git a/tests/phpunit/includes/resourceloader/ResourceLoaderWikiModuleTest.php b/tests/phpunit/includes/resourceloader/ResourceLoaderWikiModuleTest.php
index 4048ffe6d47c..5cab8e2ff6d3 100644
--- a/tests/phpunit/includes/resourceloader/ResourceLoaderWikiModuleTest.php
+++ b/tests/phpunit/includes/resourceloader/ResourceLoaderWikiModuleTest.php
@@ -1,6 +1,7 @@
 <?php
 
 use MediaWiki\MediaWikiServices;
+use Wikimedia\Rdbms\IDatabase;
 
 class ResourceLoaderWikiModuleTest extends ResourceLoaderTestCase {