class MigrateShortcutSetTest

class MigrateShortcutSetTest extends MigrateDrupal7TestBase {
    
    /**
     * Modules to enable.
     *
     * @var array
     */
    protected static $modules = [
        'link',
        'field',
        'shortcut',
        'menu_link_content',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->installEntitySchema('shortcut');
        $this->installEntitySchema('menu_link_content');
        $this->executeMigration('d7_shortcut_set');
        $this->executeMigration('d7_menu');
        $this->executeMigration('d7_shortcut');
    }
    
    /**
     * Tests the shortcut set migration.
     */
    public function testShortcutSetMigration() {
        $this->assertEntity('default', 'Default', 2);
        $this->assertEntity('shortcut_set_2', 'Alternative shortcut set', 2);
    }
    
    /**
     * Asserts various aspects of a shortcut set entity.
     *
     * @param string $id
     *   The expected shortcut set ID.
     * @param string $label
     *   The expected shortcut set label.
     * @param int $expected_size
     *   The number of shortcuts expected to be in the set.
     *
     * @internal
     */
    protected function assertEntity(string $id, string $label, int $expected_size) : void {
        $shortcut_set = ShortcutSet::load($id);
        $this->assertInstanceOf(ShortcutSetInterface::class, $shortcut_set);
        
        /** @var \Drupal\shortcut\ShortcutSetInterface $shortcut_set */
        $this->assertSame($id, $shortcut_set->id());
        $this->assertSame($label, $shortcut_set->label());
        // Check the number of shortcuts in the set.
        $shortcuts = $shortcut_set->getShortcuts();
        $this->assertCount($expected_size, $shortcuts);
    }

}