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
|
<?php
abstract class ApiFormatTestBase extends MediaWikiTestCase {
/**
* Name of the formatter being tested
* @var string
*/
protected $printerName;
/**
* Return general data to be encoded for testing
* @return array See self::testGeneralEncoding
* @throws BadMethodCallException
*/
public static function provideGeneralEncoding() {
throw new BadMethodCallException( static::class . ' must implement ' . __METHOD__ );
}
/**
* Get the formatter output for the given input data
* @param array $params Query parameters
* @param array $data Data to encode
* @param array $options Options. If passed a string, the string is treated
* as the 'class' option.
* - name: Format name, rather than $this->printerName
* - class: If set, register 'name' with this class (and 'factory', if that's set)
* - factory: Used with 'class' to register at runtime
* - returnPrinter: Return the printer object
* @param callable|null $factory Factory to use instead of the normal one
* @return string|array The string if $options['returnPrinter'] isn't set, or an array if it is:
* - text: Output text string
* - printer: ApiFormatBase
* @throws Exception
*/
protected function encodeData( array $params, array $data, $options = [] ) {
if ( is_string( $options ) ) {
$options = [ 'class' => $options ];
}
$printerName = isset( $options['name'] ) ? $options['name'] : $this->printerName;
$context = new RequestContext;
$context->setRequest( new FauxRequest( $params, true ) );
$main = new ApiMain( $context );
if ( isset( $options['class'] ) ) {
$factory = isset( $options['factory'] ) ? $options['factory'] : null;
$main->getModuleManager()->addModule( $printerName, 'format', $options['class'], $factory );
}
$result = $main->getResult();
$result->addArrayType( null, 'default' );
foreach ( $data as $k => $v ) {
$result->addValue( null, $k, $v );
}
$ret = [];
$printer = $main->createPrinterByName( $printerName );
$printer->initPrinter();
$printer->execute();
ob_start();
try {
$printer->closePrinter();
$ret['text'] = ob_get_clean();
} catch ( Exception $ex ) {
ob_end_clean();
throw $ex;
}
if ( !empty( $options['returnPrinter'] ) ) {
$ret['printer'] = $printer;
}
return count( $ret ) === 1 ? $ret['text'] : $ret;
}
/**
* @dataProvider provideGeneralEncoding
* @param array $data Data to be encoded
* @param string|Exception $expect String to expect, or exception expected to be thrown
* @param array $params Query parameters to set in the FauxRequest
* @param array $options Options to pass to self::encodeData()
*/
public function testGeneralEncoding(
array $data, $expect, array $params = [], array $options = []
) {
if ( $expect instanceof Exception ) {
$this->setExpectedException( get_class( $expect ), $expect->getMessage() );
$this->encodeData( $params, $data, $options ); // Should throw
} else {
$this->assertSame( $expect, $this->encodeData( $params, $data, $options ) );
}
}
}
|
