class UpdateRegistryTest

class UpdateRegistryTest extends UnitTestCase {
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $settings = [];
    $settings['extension_discovery_scan_tests'] = TRUE;
    new Settings($settings);
  }
  
  /**
   * Sets up some extensions with some update functions.
   */
  protected function setupBasicExtensions() {
    $info_a = <<<'EOS'
type: module
name: Module A
core_version_requirement: '*'
EOS;
    $info_b = <<<'EOS'
type: module
name: Module B
core_version_requirement: '*'
EOS;
    $info_c = <<<'EOS'
type: module
name: Module C
core_version_requirement: '*'
EOS;
    $info_d = <<<'EOS'
type: theme
name: Theme D
EOS;
    $module_a = <<<'EOS'
<?php

/**
 * Module A update B.
 */
function module_a_post_update_b() {
}

/**
 * Module A update A.
 */
function module_a_post_update_a() {
}

EOS;
    $module_b = <<<'EOS'
<?php

/**
 * Module B update A.
 */
function module_b_post_update_a() {
}

/**
 * Implements hook_removed_post_updates().
 */
function module_b_removed_post_updates() {
  return [
    'module_b_post_update_b' => '8.9.0',
    'module_b_post_update_c' => '8.9.0',
  ];
}

EOS;
    $module_c = <<<'EOS'
<?php

/**
 * Module C update A.
 */
function module_c_post_update_a() {
}

/**
 * Module C update B.
 */
function module_c_post_update_b() {
}

/**
 * Implements hook_removed_post_updates().
 */
function module_c_removed_post_updates() {
  return [
    'module_c_post_update_b' => '8.9.0',
    'module_c_post_update_c' => '8.9.0',
  ];
}

EOS;
    $theme_d = <<<'EOS'
<?php

/**
 * Theme D update B.
 */
function theme_d_post_update_b() {
}

/**
 * Theme D update C.
 */
function theme_d_post_update_c() {
}

/**
 * Implements hook_removed_post_updates().
 */
function theme_d_removed_post_updates() {
  return [
    'theme_d_post_update_a' => '8.9.0',
  ];
}

EOS;
    vfsStream::setup('drupal');
    vfsStream::create([
      'sites' => [
        'default' => [
          'modules' => [
            'module_a' => [
              'module_a.post_update.php' => $module_a,
              'module_a.info.yml' => $info_a,
            ],
            'module_b' => [
              'module_b.post_update.php' => $module_b,
              'module_b.info.yml' => $info_b,
            ],
            'module_c' => [
              'module_c.post_update.php' => $module_c,
              'module_c.info.yml' => $info_c,
            ],
          ],
          'themes' => [
            'theme_d' => [
              'theme_d.post_update.php' => $theme_d,
              'theme_d.info.yml' => $info_d,
            ],
          ],
        ],
      ],
    ]);
  }
  
  /**
   * @covers ::getPendingUpdateFunctions
   */
  public function testGetPendingUpdateFunctionsNoExistingUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    // Confirm the updates are sorted alphabetically.
    $this->assertEquals([
      'module_a_post_update_a',
      'module_a_post_update_b',
      'module_b_post_update_a',
      'theme_d_post_update_b',
      'theme_d_post_update_c',
    ], $update_registry->getPendingUpdateFunctions());
  }
  
  /**
   * @covers ::getPendingUpdateFunctions
   */
  public function testGetPendingUpdateFunctionsWithLoadedModulesButNotEnabled() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([]);
    $theme_handler = $theme_handler->reveal();
    // Preload modules to ensure that ::getAvailableUpdateFunctions filters out
    // not enabled modules.
    include_once 'vfs://drupal/sites/default/modules/module_a/module_a.post_update.php';
    include_once 'vfs://drupal/sites/default/modules/module_b/module_b.post_update.php';
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    // Confirm the updates are sorted alphabetically.
    $this->assertEquals([
      'module_a_post_update_a',
      'module_a_post_update_b',
    ], $update_registry->getPendingUpdateFunctions());
  }
  
  /**
   * @covers ::getPendingUpdateFunctions
   */
  public function testGetPendingUpdateFunctionsExistingUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([
      'module_a_post_update_a',
      'theme_d_post_update_a',
      'theme_d_post_update_b',
    ]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    // Confirm the updates are sorted alphabetically.
    $this->assertEquals(array_values([
      'module_a_post_update_b',
      'module_b_post_update_a',
      'theme_d_post_update_c',
    ]), array_values($update_registry->getPendingUpdateFunctions()));
  }
  
  /**
   * @covers ::getPendingUpdateInformation
   */
  public function testGetPendingUpdateInformation() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    // Confirm the updates are sorted alphabetically.
    $expected = [];
    $expected['module_a']['pending']['a'] = 'Module A update A.';
    $expected['module_a']['pending']['b'] = 'Module A update B.';
    $expected['module_a']['start'] = 'a';
    $expected['module_b']['pending']['a'] = 'Module B update A.';
    $expected['module_b']['start'] = 'a';
    $expected['theme_d']['pending']['b'] = 'Theme D update B.';
    $expected['theme_d']['pending']['c'] = 'Theme D update C.';
    $expected['theme_d']['start'] = 'b';
    $this->assertEquals($expected, $update_registry->getPendingUpdateInformation());
  }
  
  /**
   * @covers ::getPendingUpdateInformation
   */
  public function testGetPendingUpdateInformationWithExistingUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([
      'module_a_post_update_a',
      'theme_d_post_update_a',
      'theme_d_post_update_b',
    ]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    // Confirm the updates are sorted alphabetically.
    $expected = [];
    $expected['module_a']['pending']['b'] = 'Module A update B.';
    $expected['module_a']['start'] = 'b';
    $expected['module_b']['pending']['a'] = 'Module B update A.';
    $expected['module_b']['start'] = 'a';
    $expected['theme_d']['pending']['c'] = 'Theme D update C.';
    $expected['theme_d']['start'] = 'c';
    $this->assertEquals($expected, $update_registry->getPendingUpdateInformation());
  }
  
  /**
   * @covers ::getPendingUpdateInformation
   */
  public function testGetPendingUpdateInformationWithRemovedUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([
      'module_a_post_update_a',
    ]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_c' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_c/module_c.info.yml',
        'filename' => 'module_c.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $this->expectException(RemovedPostUpdateNameException::class);
    $update_registry->getPendingUpdateInformation();
  }
  
  /**
   * @covers ::getUpdateFunctions
   */
  public function testGetUpdateFunctions() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class)
      ->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $this->assertEquals([
      'module_a_post_update_a',
      'module_a_post_update_b',
    ], array_values($update_registry->getUpdateFunctions('module_a')));
    $this->assertEquals([
      'module_b_post_update_a',
    ], array_values($update_registry->getUpdateFunctions('module_b')));
    $this->assertEquals([
      'theme_d_post_update_b',
      'theme_d_post_update_c',
    ], array_values($update_registry->getUpdateFunctions('theme_d')));
  }
  
  /**
   * @covers ::registerInvokedUpdates
   */
  public function testRegisterInvokedUpdatesWithoutExistingUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([])
      ->shouldBeCalledTimes(1);
    $key_value->set('existing_updates', [
      'module_a_post_update_a',
    ])
      ->willReturn(NULL)
      ->shouldBeCalledTimes(1);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $update_registry->registerInvokedUpdates([
      'module_a_post_update_a',
    ]);
  }
  
  /**
   * @covers ::registerInvokedUpdates
   */
  public function testRegisterInvokedUpdatesWithMultiple() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([])
      ->shouldBeCalledTimes(1);
    $key_value->set('existing_updates', [
      'module_a_post_update_a',
      'module_a_post_update_b',
      'theme_d_post_update_c',
    ])
      ->willReturn(NULL)
      ->shouldBeCalledTimes(1);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $update_registry->registerInvokedUpdates([
      'module_a_post_update_a',
      'module_a_post_update_b',
      'theme_d_post_update_c',
    ]);
  }
  
  /**
   * @covers ::registerInvokedUpdates
   */
  public function testRegisterInvokedUpdatesWithExistingUpdates() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([
      'module_a_post_update_b',
    ])
      ->shouldBeCalledTimes(1);
    $key_value->set('existing_updates', [
      'module_a_post_update_b',
      'module_a_post_update_a',
    ])
      ->willReturn(NULL)
      ->shouldBeCalledTimes(1);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $update_registry->registerInvokedUpdates([
      'module_a_post_update_a',
    ]);
  }
  
  /**
   * @covers ::filterOutInvokedUpdatesByExtension
   */
  public function testFilterOutInvokedUpdatesByExtension() : void {
    $this->setupBasicExtensions();
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([
      'module_a_post_update_b',
      'module_a_post_update_a',
      'module_b_post_update_a',
      'theme_d_post_update_c',
    ])
      ->shouldBeCalledTimes(1);
    $key_value->set('existing_updates', [
      'module_b_post_update_a',
      'theme_d_post_update_c',
    ])
      ->willReturn(NULL)
      ->shouldBeCalledTimes(1);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
      'module_b' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_b/module_b.info.yml',
        'filename' => 'module_b.module',
      ],
    ], $key_value, $theme_handler, 'post_update');
    $update_registry->filterOutInvokedUpdatesByExtension('module_a');
  }
  
  /**
   * @covers ::getPendingUpdateFunctions
   */
  public function testGetPendingCustomUpdateFunctions() : void {
    // Set up a simplified module structure with custom update hooks.
    $info_a = <<<'EOS'
type: module
name: Module A
core_version_requirement: '*'
EOS;
    $info_d = <<<'EOS'
type: theme
name: Theme D
EOS;
    $module_a = <<<'EOS'
<?php

/**
 * Module A update A.
 */
function module_a_custom_update_a() {
}

EOS;
    $theme_d = <<<'EOS'
<?php

/**
 * Theme D update B.
 */
function theme_d_custom_update_a() {
}

EOS;
    vfsStream::setup('drupal');
    vfsStream::create([
      'sites' => [
        'default' => [
          'modules' => [
            'module_a' => [
              'module_a.custom_update.php' => $module_a,
              'module_a.info.yml' => $info_a,
            ],
          ],
          'themes' => [
            'theme_d' => [
              'theme_d.custom_update.php' => $theme_d,
              'theme_d.info.yml' => $info_d,
            ],
          ],
        ],
      ],
    ]);
    $key_value = $this->prophesize(KeyValueStoreInterface::class);
    $key_value->get('existing_updates', [])
      ->willReturn([]);
    $key_value = $key_value->reveal();
    $theme_handler = $this->prophesize(ThemeHandlerInterface::class);
    $theme_handler->listInfo()
      ->willReturn([
      'theme_d' => [
        'type' => 'theme_d',
        'pathname' => 'core/themes/theme_d/theme_d.info.yml',
      ],
    ]);
    $theme_handler = $theme_handler->reveal();
    $update_registry = new UpdateRegistry('vfs://drupal', 'sites/default', [
      'module_a' => [
        'type' => 'module',
        'pathname' => 'core/modules/module_a/module_a.info.yml',
        'filename' => 'module_a.module',
      ],
    ], $key_value, $theme_handler, 'custom_update');
    // Themes are not supported.
    $this->assertEquals([
      'module_a_custom_update_a',
    ], $update_registry->getPendingUpdateFunctions());
  }

}