McrWriteBothSchemaOverride.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Tests\Revision;
 
use Wikimedia\Rdbms\IMaintainableDatabase;
use MediaWiki\DB\PatchFileLocation;
 
trait McrWriteBothSchemaOverride {
 
    use PatchFileLocation;
    use McrSchemaDetection;
 
    protected function getMcrMigrationStage() {
        return SCHEMA_COMPAT_WRITE_BOTH | SCHEMA_COMPAT_READ_OLD;
    }
 
    protected function getMcrTablesToReset() {
        return [ 'content', 'content_models', 'slots', 'slot_roles' ];
    }
 
    protected function getSchemaOverrides( IMaintainableDatabase $db ) {
        $overrides = [
            'scripts' => [],
            'drop' => [],
            'create' => [],
            'alter' => [],
        ];
 
        if ( !$this->hasMcrTables( $db ) ) {
            $overrides['create'] = [ 'slots', 'content', 'slot_roles', 'content_models', ];
            $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-slot_roles' );
            $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-content_models' );
            $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-content' );
            $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-slots' );
        }
 
        if ( !$this->hasPreMcrFields( $db ) ) {
            $overrides['alter'][] = 'revision';
            $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'create-pre-mcr-fields', __DIR__ );
        }
 
        return $overrides;
    }
 
}