3 namespace MediaWiki\Tests\Revision
;
5 use Wikimedia\Rdbms\IMaintainableDatabase
;
6 use MediaWiki\DB\PatchFileLocation
;
9 * Trait providing schema overrides that allow tests to run against the intermediate MCR database
10 * schema for use during schema migration.
12 trait McrWriteBothSchemaOverride
{
14 use PatchFileLocation
;
15 use McrSchemaDetection
;
20 protected function getMcrMigrationStage() {
21 return SCHEMA_COMPAT_WRITE_BOTH | SCHEMA_COMPAT_READ_OLD
;
27 protected function getMcrTablesToReset() {
28 return [ 'content', 'content_models', 'slots', 'slot_roles' ];
34 protected function getSchemaOverrides( IMaintainableDatabase
$db ) {
42 if ( !$this->hasMcrTables( $db ) ) {
43 $overrides['create'] = [ 'slots', 'content', 'slot_roles', 'content_models', ];
44 $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-slot_roles' );
45 $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-content_models' );
46 $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-content' );
47 $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'patch-slots' );
50 if ( !$this->hasPreMcrFields( $db ) ) {
51 $overrides['alter'][] = 'revision';
52 $overrides['scripts'][] = $this->getSqlPatchPath( $db, 'create-pre-mcr-fields', __DIR__
);