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
121
122
123
|
<?php
/**
* @group Database
* @group DatabaseBase
*/
class DatabaseTest extends MediaWikiTestCase {
var $db, $functionTest = false;
function setUp() {
$this->db = wfGetDB( DB_MASTER );
}
function tearDown() {
if ( $this->functionTest ) {
$this->dropFunctions();
$this->functionTest = false;
}
}
function testAddQuotesNull() {
$check = "NULL";
if ( $this->db->getType() === 'sqlite' || $this->db->getType() === 'oracle' ) {
$check = "''";
}
$this->assertEquals( $check, $this->db->addQuotes( null ) );
}
function testAddQuotesInt() {
# returning just "1234" should be ok too, though...
# maybe
$this->assertEquals(
"'1234'",
$this->db->addQuotes( 1234 ) );
}
function testAddQuotesFloat() {
# returning just "1234.5678" would be ok too, though
$this->assertEquals(
"'1234.5678'",
$this->db->addQuotes( 1234.5678 ) );
}
function testAddQuotesString() {
$this->assertEquals(
"'string'",
$this->db->addQuotes( 'string' ) );
}
function testAddQuotesStringQuote() {
$check = "'string''s cause trouble'";
if ( $this->db->getType() === 'mysql' ) {
$check = "'string\'s cause trouble'";
}
$this->assertEquals(
$check,
$this->db->addQuotes( "string's cause trouble" ) );
}
function testFillPreparedEmpty() {
$sql = $this->db->fillPrepared(
'SELECT * FROM interwiki', array() );
$this->assertEquals(
"SELECT * FROM interwiki",
$sql );
}
function testFillPreparedQuestion() {
$sql = $this->db->fillPrepared(
'SELECT * FROM cur WHERE cur_namespace=? AND cur_title=?',
array( 4, "Snicker's_paradox" ) );
$check = "SELECT * FROM cur WHERE cur_namespace='4' AND cur_title='Snicker''s_paradox'";
if ( $this->db->getType() === 'mysql' ) {
$check = "SELECT * FROM cur WHERE cur_namespace='4' AND cur_title='Snicker\'s_paradox'";
}
$this->assertEquals( $check, $sql );
}
function testFillPreparedBang() {
$sql = $this->db->fillPrepared(
'SELECT user_id FROM ! WHERE user_name=?',
array( '"user"', "Slash's Dot" ) );
$check = "SELECT user_id FROM \"user\" WHERE user_name='Slash''s Dot'";
if ( $this->db->getType() === 'mysql' ) {
$check = "SELECT user_id FROM \"user\" WHERE user_name='Slash\'s Dot'";
}
$this->assertEquals( $check, $sql );
}
function testFillPreparedRaw() {
$sql = $this->db->fillPrepared(
"SELECT * FROM cur WHERE cur_title='This_\\&_that,_WTF\\?\\!'",
array( '"user"', "Slash's Dot" ) );
$this->assertEquals(
"SELECT * FROM cur WHERE cur_title='This_&_that,_WTF?!'",
$sql );
}
/**
* @group Broken
*/
function testStoredFunctions() {
if ( !in_array( wfGetDB( DB_MASTER )->getType(), array( 'mysql', 'postgres' ) ) ) {
$this->markTestSkipped( 'MySQL or Postgres required' );
}
global $IP;
$this->dropFunctions();
$this->functionTest = true;
$this->assertTrue( $this->db->sourceFile( "$IP/tests/phpunit/data/db/{$this->db->getType()}/functions.sql" ) );
$res = $this->db->query( 'SELECT mw_test_function() AS test', __METHOD__ );
$this->assertEquals( 42, $res->fetchObject()->test );
}
private function dropFunctions() {
$this->db->query( 'DROP FUNCTION IF EXISTS mw_test_function'
. ( $this->db->getType() == 'postgres' ? '()' : '' )
);
}
}
|
