1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
<?php
use MediaWiki\Config\HashConfig;
use MediaWiki\MainConfigNames;
/**
* Import failure test.
*
* @group Database
* @covers \WikiImporter
*/
class ImportFailureTest extends MediaWikiLangTestCase {
protected function setUp(): void {
parent::setUp();
$slotRoleRegistry = $this->getServiceContainer()->getSlotRoleRegistry();
if ( !$slotRoleRegistry->isDefinedRole( 'ImportFailureTest' ) ) {
$slotRoleRegistry->defineRoleWithModel( 'ImportFailureTest', CONTENT_MODEL_WIKITEXT );
}
}
/**
* @param ImportSource $source
* @return WikiImporter
*/
private function getImporter( ImportSource $source ) {
$config = new HashConfig( [
MainConfigNames::MaxArticleSize => 2048,
] );
$services = $this->getServiceContainer();
$importer = new WikiImporter(
$source,
$this->getTestSysop()->getAuthority(),
$config,
$services->getHookContainer(),
$services->getContentLanguage(),
$services->getNamespaceInfo(),
$services->getTitleFactory(),
$services->getWikiPageFactory(),
$services->getWikiRevisionUploadImporter(),
$services->getContentHandlerFactory(),
$services->getSlotRoleRegistry()
);
return $importer;
}
/**
* @param string $testName
*
* @return string
*/
private function getFileToImport( string $testName ) {
return __DIR__ . "/../../data/import/$testName.xml";
}
/**
* @param string $prefix
* @param string[] $keys
*
* @return string[]
*/
private function getUniqueNames( string $prefix, array $keys ) {
$names = [];
foreach ( $keys as $k ) {
$names[$k] = "$prefix-$k-" . wfRandomString( 6 );
}
return $names;
}
/**
* @param string $xmlData
* @param string[] $pageTitles
*
* @return string
*/
private function injectPageTitles( string $xmlData, array $pageTitles ) {
$keys = array_map( static function ( $name ) {
return "{{{$name}_title}}";
}, array_keys( $pageTitles ) );
return str_replace(
$keys,
array_values( $pageTitles ),
$xmlData
);
}
public static function provideImportFailure() {
yield [ 'BadXML', RuntimeException::class, '/^XML error at line 3: Opening and ending tag mismatch:.*$/' ];
yield [ 'MissingMediaWikiTag', UnexpectedValueException::class, "/^Expected '<mediawiki>' tag, got .*$/" ];
yield [ 'MissingMainTextField', InvalidArgumentException::class, '/^Missing text field in import.$/' ];
yield [ 'MissingSlotTextField', InvalidArgumentException::class, '/^Missing text field in import.$/' ];
yield [ 'MissingSlotRole', RuntimeException::class, '/^Missing role for imported slot.$/' ];
yield [ 'UndefinedSlotRole', RuntimeException::class, '/^Undefined slot role .*$/' ];
yield [ 'UndefinedContentModel', MWUnknownContentModelException::class, '/not registered on this wiki/' ];
}
/**
* @dataProvider provideImportFailure
*/
public function testImportFailure( $testName, $exceptionName, $exceptionMessage ) {
$fileName = $this->getFileToImport( $testName );
$pageKeys = [ 'page1', 'page2', 'page3', 'page4', ];
$pageTitles = $this->getUniqueNames( $testName, $pageKeys );
$xmlData = file_get_contents( $fileName );
$xmlData = $this->injectPageTitles( $xmlData, $pageTitles );
$source = new ImportStringSource( $xmlData );
$importer = $this->getImporter( $source );
$this->expectException( $exceptionName );
$this->expectExceptionMessageMatches( $exceptionMessage );
$importer->doImport();
}
}
|
