Add test

Signed-off-by: Christopher Ng <chrng8@gmail.com>

1 files changed, 27 insertions, 0 deletions

diff --git a/tests/lib/AppFramework/Bootstrap/RegistrationContextTest.php b/tests/lib/AppFramework/Bootstrap/RegistrationContextTest.php
index 067e9a09673..81fac49348b 100644
--- a/tests/lib/AppFramework/Bootstrap/RegistrationContextTest.php
+++ b/tests/lib/AppFramework/Bootstrap/RegistrationContextTest.php
@@ -26,6 +26,7 @@ declare(strict_types=1);
 namespace lib\AppFramework\Bootstrap;
 
 use OC\AppFramework\Bootstrap\RegistrationContext;
+use OC\AppFramework\Bootstrap\ServiceRegistration;
 use OCP\AppFramework\App;
 use OCP\AppFramework\IAppContainer;
 use OCP\EventDispatcher\IEventDispatcher;
@@ -163,6 +164,32 @@ class RegistrationContextTest extends TestCase {
 		]);
 	}
 
+	public function testRegisterUserMigrator(): void {
+		$appIdA = 'myapp';
+		$migratorClassA = 'OCA\App\UserMigration\AppMigrator';
+
+		$appIdB = 'otherapp';
+		$migratorClassB = 'OCA\OtherApp\UserMigration\OtherAppMigrator';
+
+		$serviceRegistrationA = new ServiceRegistration($appIdA, $migratorClassA);
+		$serviceRegistrationB = new ServiceRegistration($appIdB, $migratorClassB);
+
+		$this->context
+			->for($appIdA)
+			->registerUserMigrator($migratorClassA);
+		$this->context
+			->for($appIdB)
+			->registerUserMigrator($migratorClassB);
+
+		$this->assertEquals(
+			[
+				$serviceRegistrationA,
+				$serviceRegistrationB,
+			],
+			$this->context->getUserMigrators(),
+		);
+	}
+
 	public function dataProvider_TrueFalse() {
 		return[
 			[true],