View source
<?php
namespace Drupal\Tests\Core\Database;
use Drupal\Tests\Core\Database\Stub\StubConnection;
use Drupal\Tests\UnitTestCase;
class ConnectionTest extends UnitTestCase {
public function providerPrefixRoundTrip() {
return array(
array(
array(
'' => 'test_',
),
'test_',
),
array(
array(
'fooTable' => 'foo_',
'barTable' => 'bar_',
),
array(
'fooTable' => 'foo_',
'barTable' => 'bar_',
),
),
);
}
public function testPrefixRoundTrip($expected, $prefix_info) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array());
$reflection = new \ReflectionClass('Drupal\\Tests\\Core\\Database\\Stub\\StubConnection');
$set_prefix = $reflection
->getMethod('setPrefix');
$set_prefix
->setAccessible(TRUE);
$set_prefix
->invokeArgs($connection, array(
$prefix_info,
));
foreach ($expected as $table => $prefix) {
$this
->assertEquals($prefix, $connection
->tablePrefix($table));
}
}
public function providerTestPrefixTables() {
return array(
array(
'SELECT * FROM test_table',
'test_',
'SELECT * FROM {table}',
),
array(
'SELECT * FROM first_table JOIN second_thingie',
array(
'table' => 'first_',
'thingie' => 'second_',
),
'SELECT * FROM {table} JOIN {thingie}',
),
);
}
public function testPrefixTables($expected, $prefix_info, $query) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array(
'prefix' => $prefix_info,
));
$this
->assertEquals($expected, $connection
->prefixTables($query));
}
public function providerEscapeMethods() {
return array(
array(
'thing',
'thing',
),
array(
'_item',
'_item',
),
array(
'item_',
'item_',
),
array(
'_item_',
'_item_',
),
array(
'',
'!@#$%^&*()-=+',
),
array(
'123',
'!1@2#3',
),
);
}
public function testEscapeMethods($expected, $name) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array());
$this
->assertEquals($expected, $connection
->escapeDatabase($name));
$this
->assertEquals($expected, $connection
->escapeTable($name));
$this
->assertEquals($expected, $connection
->escapeField($name));
$this
->assertEquals($expected, $connection
->escapeAlias($name));
}
public function providerGetDriverClass() {
return array(
array(
'nonexistent_class',
'stub',
'\\',
'nonexistent_class',
),
array(
'Drupal\\Core\\Database\\Driver\\mysql\\Select',
'mysql',
NULL,
'Select',
),
array(
'Drupal\\Tests\\Core\\Database\\Stub\\Driver\\Schema',
'stub',
'Drupal\\Tests\\Core\\Database\\Stub\\Driver',
'Schema',
),
);
}
public function testGetDriverClass($expected, $driver, $namespace, $class) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array(
'namespace' => $namespace,
));
$connection->driver = $driver;
$this
->assertEquals($expected, $connection
->getDriverClass($class));
}
public function providerSchema() {
return array(
array(
'Drupal\\Tests\\Core\\Database\\Stub\\Driver\\Schema',
'stub',
'Drupal\\Tests\\Core\\Database\\Stub\\Driver',
),
);
}
public function testSchema($expected, $driver, $namespace) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array(
'namespace' => $namespace,
));
$connection->driver = $driver;
$this
->assertInstanceOf($expected, $connection
->schema());
}
public function testDestroy() {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubConnection', NULL, array(
$mock_pdo,
array(
'namespace' => 'Drupal\\Tests\\Core\\Database\\Stub\\Driver',
),
));
$this
->assertInstanceOf('Drupal\\Tests\\Core\\Database\\Stub\\Driver\\Schema', $connection
->schema());
$connection
->destroy();
$this
->assertAttributeEquals(NULL, 'schema', $connection);
}
public function providerMakeComments() {
return array(
array(
'/* */ ',
array(
'',
),
),
array(
'/* Exploit * / DROP TABLE node. -- */ ',
array(
'Exploit * / DROP TABLE node; --',
),
),
array(
'/* Exploit * / DROP TABLE node. --. another comment */ ',
array(
'Exploit * / DROP TABLE node; --',
'another comment',
),
),
);
}
public function testMakeComments($expected, $comment_array) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array());
$this
->assertEquals($expected, $connection
->makeComment($comment_array));
}
public function providerFilterComments() {
return array(
array(
'',
'',
),
array(
'Exploit * / DROP TABLE node. --',
'Exploit * / DROP TABLE node; --',
),
array(
'Exploit * / DROP TABLE node. --',
'Exploit */ DROP TABLE node; --',
),
);
}
public function testFilterComments($expected, $comment) {
$mock_pdo = $this
->getMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
$connection = new StubConnection($mock_pdo, array());
$reflection = new \ReflectionClass('Drupal\\Tests\\Core\\Database\\Stub\\StubConnection');
$filter_comment = $reflection
->getMethod('filterComment');
$filter_comment
->setAccessible(TRUE);
$this
->assertEquals($expected, $filter_comment
->invokeArgs($connection, array(
$comment,
)));
}
}