ConnectionTest.php

Same filename in this branch
  1. 11.x core/modules/sqlite/tests/src/Unit/ConnectionTest.php
  2. 11.x core/modules/mysql/tests/src/Unit/ConnectionTest.php
  3. 11.x core/modules/mysql/tests/src/Kernel/mysql/ConnectionTest.php
  4. 11.x core/tests/Drupal/KernelTests/Core/Database/ConnectionTest.php
  5. 11.x core/modules/mysqli/tests/src/Kernel/mysqli/ConnectionTest.php
Same filename and directory in other branches
  1. 9 core/modules/sqlite/tests/src/Unit/ConnectionTest.php
  2. 9 core/modules/mysql/tests/src/Unit/ConnectionTest.php
  3. 9 core/modules/mysql/tests/src/Kernel/mysql/ConnectionTest.php
  4. 9 core/tests/Drupal/KernelTests/Core/Database/ConnectionTest.php
  5. 9 core/tests/Drupal/Tests/Core/Database/ConnectionTest.php
  6. 8.9.x core/tests/Drupal/KernelTests/Core/Database/ConnectionTest.php
  7. 8.9.x core/tests/Drupal/Tests/Core/Database/Driver/sqlite/ConnectionTest.php
  8. 8.9.x core/tests/Drupal/Tests/Core/Database/ConnectionTest.php
  9. 10 core/modules/sqlite/tests/src/Unit/ConnectionTest.php
  10. 10 core/modules/sqlite/tests/src/Kernel/sqlite/ConnectionTest.php
  11. 10 core/modules/mysql/tests/src/Unit/ConnectionTest.php
  12. 10 core/modules/mysql/tests/src/Kernel/mysql/ConnectionTest.php
  13. 10 core/tests/Drupal/KernelTests/Core/Database/ConnectionTest.php
  14. 10 core/tests/Drupal/Tests/Core/Database/ConnectionTest.php

Namespace

Drupal\Tests\Core\Database

File

core/tests/Drupal/Tests/Core/Database/ConnectionTest.php

View source
<?php

declare (strict_types=1);
namespace Drupal\Tests\Core\Database;

use Composer\Autoload\ClassLoader;
use Drupal\Core\Database\Database;
use Drupal\Core\Database\Statement\FetchAs;
use Drupal\Core\Database\StatementPrefetchIterator;
use Drupal\Tests\Core\Database\Stub\StubConnection;
use Drupal\Tests\Core\Database\Stub\StubPDO;
use Drupal\Tests\UnitTestCase;
use PHPUnit\Framework\Attributes\DataProvider;
use PHPUnit\Framework\Attributes\IgnoreDeprecations;

/**
 * Tests the Connection class.
 *
 * @coversDefaultClass \Drupal\Core\Database\Connection
 * @group Database
 */
class ConnectionTest extends UnitTestCase {
  
  /**
   * Data provider for testPrefixRoundTrip().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Arguments to pass to Connection::setPrefix().
   *   - Expected result from Connection::getPrefix().
   */
  public static function providerPrefixRoundTrip() {
    return [
      [
        [
          '' => 'test_',
        ],
        'test_',
      ],
      [
        [
          'fooTable' => 'foo_',
          'barTable' => 'foo_',
        ],
        'foo_',
      ],
    ];
  }
  
  /**
   * Exercise setPrefix() and getPrefix().
   *
   * @dataProvider providerPrefixRoundTrip
   */
  public function testPrefixRoundTrip($expected, $prefix_info) : void {
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, []);
    // setPrefix() is protected, so we make it accessible with reflection.
    $reflection = new \ReflectionClass('Drupal\\Tests\\Core\\Database\\Stub\\StubConnection');
    $set_prefix = $reflection->getMethod('setPrefix');
    // Set the prefix data.
    $set_prefix->invokeArgs($connection, [
      $prefix_info,
    ]);
    // Check the round-trip.
    foreach ($expected as $prefix) {
      $this->assertEquals($prefix, $connection->getPrefix());
    }
  }
  
  /**
   * Data provider for testPrefixTables().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected result.
   *   - Table prefix.
   *   - Query to be prefixed.
   *   - Quote identifier.
   */
  public static function providerTestPrefixTables() {
    return [
      [
        'SELECT * FROM test_table',
        'test_',
        'SELECT * FROM {table}',
        [
          '',
          '',
        ],
      ],
      [
        'SELECT * FROM "test_table"',
        'test_',
        'SELECT * FROM {table}',
        [
          '"',
          '"',
        ],
      ],
      [
        "SELECT * FROM 'test_table'",
        'test_',
        'SELECT * FROM {table}',
        [
          "'",
          "'",
        ],
      ],
      [
        "SELECT * FROM [test_table]",
        'test_',
        'SELECT * FROM {table}',
        [
          '[',
          ']',
        ],
      ],
    ];
  }
  
  /**
   * Exercise the prefixTables() method.
   *
   * @dataProvider providerTestPrefixTables
   */
  public function testPrefixTables($expected, $prefix_info, $query, array $quote_identifier = [
    '"',
    '"',
  ]) : void {
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, [
      'prefix' => $prefix_info,
    ], $quote_identifier);
    $this->assertEquals($expected, $connection->prefixTables($query));
  }
  
  /**
   * Data provider for testGetDriverClass().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected namespaced class name.
   *   - Namespace.
   *   - Class name without namespace.
   */
  public static function providerGetDriverClass() {
    return [
      [
        'nonexistent_class',
        '\\',
        'nonexistent_class',
      ],
      [
        'Drupal\\Tests\\Core\\Database\\Stub\\Select',
        NULL,
        'Select',
      ],
      // Tests with the CoreFake database driver. This driver has no custom
      // driver classes.
[
        'Drupal\\Core\\Database\\Query\\Condition',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Condition',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Delete',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Delete',
      ],
      [
        'Drupal\\Core\\Database\\ExceptionHandler',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'ExceptionHandler',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Insert',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Insert',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Merge',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Merge',
      ],
      [
        'PagerSelectExtender',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'PagerSelectExtender',
      ],
      [
        'Drupal\\Core\\Database\\Schema',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Schema',
      ],
      [
        'SearchQuery',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'SearchQuery',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Select',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Select',
      ],
      [
        'Drupal\\Core\\Database\\Transaction',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Transaction',
      ],
      [
        'TableSortExtender',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'TableSortExtender',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Truncate',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Truncate',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Update',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Update',
      ],
      [
        'Drupal\\Core\\Database\\Query\\Upsert',
        'Drupal\\CoreFake\\Driver\\Database\\CoreFake',
        'Upsert',
      ],
      // Tests with the CoreFakeWithAllCustomClasses database driver. This
      // driver has custom driver classes for all classes.
[
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Condition',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Condition',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Delete',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Delete',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\ExceptionHandler',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'ExceptionHandler',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Insert',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Insert',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Merge',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Merge',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\PagerSelectExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'PagerSelectExtender',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Schema',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Schema',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\SearchQuery',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'SearchQuery',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Select',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Select',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\TableSortExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'TableSortExtender',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Transaction',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Transaction',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Truncate',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Truncate',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Update',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Update',
      ],
      [
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\Upsert',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Upsert',
      ],
      [
        'Drupal\\Core\\Database\\Query\\PagerSelectExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Drupal\\Core\\Database\\Query\\PagerSelectExtender',
      ],
      [
        '\\Drupal\\Core\\Database\\Query\\PagerSelectExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        '\\Drupal\\Core\\Database\\Query\\PagerSelectExtender',
      ],
      [
        'Drupal\\Core\\Database\\Query\\TableSortExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Drupal\\Core\\Database\\Query\\TableSortExtender',
      ],
      [
        '\\Drupal\\Core\\Database\\Query\\TableSortExtender',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        '\\Drupal\\Core\\Database\\Query\\TableSortExtender',
      ],
      [
        'Drupal\\search\\SearchQuery',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        'Drupal\\search\\SearchQuery',
      ],
      [
        '\\Drupal\\search\\SearchQuery',
        'Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses',
        '\\Drupal\\search\\SearchQuery',
      ],
    ];
  }
  
  /**
   * @covers ::getDriverClass
   * @dataProvider providerGetDriverClass
   * @group legacy
   */
  public function testGetDriverClass($expected, $namespace, $class) : void {
    $additional_class_loader = new ClassLoader();
    $additional_class_loader->addPsr4("Drupal\\core_fake\\Driver\\Database\\CoreFake\\", __DIR__ . "/../../../../../tests/fixtures/database_drivers/module/core_fake/src/Driver/Database/CoreFake");
    $additional_class_loader->addPsr4("Drupal\\core_fake\\Driver\\Database\\CoreFakeWithAllCustomClasses\\", __DIR__ . "/../../../../../tests/fixtures/database_drivers/module/core_fake/src/Driver/Database/CoreFakeWithAllCustomClasses");
    $additional_class_loader->register(TRUE);
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, [
      'namespace' => $namespace,
    ]);
    match ($class) {  'Install\\Tasks', 'ExceptionHandler', 'Select', 'Insert', 'Merge', 'Upsert', 'Update', 'Delete', 'Truncate', 'Schema', 'Condition', 'Transaction' => $this->expectExceptionMessage('Calling Drupal\\Core\\Database\\Connection::getDriverClass() for \'' . $class . '\' is not supported. Use standard autoloading in the methods that return database operations. See https://www.drupal.org/node/3217534'),
      default => NULL,
    
    };
    $this->assertEquals($expected, $connection->getDriverClass($class));
  }
  
  /**
   * Data provider for testSchema().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected namespaced class of schema object.
   *   - Driver for PDO connection.
   *   - Namespace for connection.
   */
  public static function providerSchema() {
    return [
      [
        'Drupal\\Tests\\Core\\Database\\Stub\\Driver\\Schema',
        'stub',
        'Drupal\\Tests\\Core\\Database\\Stub\\Driver',
      ],
    ];
  }
  
  /**
   * Tests Connection::schema().
   *
   * @dataProvider providerSchema
   */
  public function testSchema($expected, $driver, $namespace) : void {
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, [
      'namespace' => $namespace,
    ]);
    $connection->driver = $driver;
    $this->assertInstanceOf($expected, $connection->schema());
  }
  
  /**
   * Data provider for testMakeComments().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected filtered comment.
   *   - Arguments for Connection::makeComment().
   */
  public static function providerMakeComments() {
    return [
      [
        '/*  */ ',
        [
          '',
        ],
      ],
      [
        '/* Exploit  *  / DROP TABLE node. -- */ ',
        [
          'Exploit * / DROP TABLE node; --',
        ],
      ],
      [
        '/* Exploit  *  / DROP TABLE node. --. another comment */ ',
        [
          'Exploit * / DROP TABLE node; --',
          'another comment',
        ],
      ],
    ];
  }
  
  /**
   * Tests Connection::makeComments().
   *
   * @dataProvider providerMakeComments
   */
  public function testMakeComments($expected, $comment_array) : void {
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, []);
    $this->assertEquals($expected, $connection->makeComment($comment_array));
  }
  
  /**
   * Data provider for testFilterComments().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected filtered comment.
   *   - Comment to filter.
   */
  public static function providerFilterComments() {
    return [
      [
        '',
        '',
      ],
      [
        'Exploit  *  / DROP TABLE node. --',
        'Exploit * / DROP TABLE node; --',
      ],
      [
        'Exploit  * / DROP TABLE node. --',
        'Exploit */ DROP TABLE node; --',
      ],
    ];
  }
  
  /**
   * Tests Connection::filterComments().
   *
   * @dataProvider providerFilterComments
   */
  public function testFilterComments($expected, $comment) : void {
    $mock_pdo = $this->createMock('Drupal\\Tests\\Core\\Database\\Stub\\StubPDO');
    $connection = new StubConnection($mock_pdo, []);
    // filterComment() is protected, so we make it accessible with reflection.
    $reflection = new \ReflectionClass('Drupal\\Tests\\Core\\Database\\Stub\\StubConnection');
    $filter_comment = $reflection->getMethod('filterComment');
    $this->assertEquals($expected, $filter_comment->invokeArgs($connection, [
      $comment,
    ]));
  }
  
  /**
   * Data provider for testEscapeTable.
   *
   * @return array
   *   An indexed array of where each value is an array of arguments to pass to
   *   testEscapeField. The first value is the expected value, and the second
   *   value is the value to test.
   */
  public static function providerEscapeTables() {
    return [
      [
        'nocase',
        'nocase',
      ],
      [
        'camelCase',
        'camelCase',
      ],
      [
        'backtick',
        '`backtick`',
        [
          '`',
          '`',
        ],
      ],
      [
        'brackets',
        '[brackets]',
        [
          '[',
          ']',
        ],
      ],
      [
        'camelCase',
        '"camelCase"',
      ],
      [
        'camelCase',
        'camel/Case',
      ],
      // Sometimes, table names are following the pattern database.schema.table.
[
        'camelCase.nocase.nocase',
        'camelCase.nocase.nocase',
      ],
      [
        'nocase.camelCase.nocase',
        'nocase.camelCase.nocase',
      ],
      [
        'nocase.nocase.camelCase',
        'nocase.nocase.camelCase',
      ],
      [
        'camelCase.camelCase.camelCase',
        'camelCase.camelCase.camelCase',
      ],
    ];
  }
  
  /**
   * @covers ::escapeTable
   * @dataProvider providerEscapeTables
   */
  public function testEscapeTable($expected, $name, array $identifier_quote = [
    '"',
    '"',
  ]) : void {
    $mock_pdo = $this->createMock(StubPDO::class);
    $connection = new StubConnection($mock_pdo, [], $identifier_quote);
    $this->assertEquals($expected, $connection->escapeTable($name));
  }
  
  /**
   * Data provider for testEscapeAlias.
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected escaped string.
   *   - String to escape.
   */
  public static function providerEscapeAlias() {
    return [
      [
        '!nocase!',
        'nocase',
        [
          '!',
          '!',
        ],
      ],
      [
        '`backtick`',
        'backtick',
        [
          '`',
          '`',
        ],
      ],
      [
        'nocase',
        'nocase',
        [
          '',
          '',
        ],
      ],
      [
        '[brackets]',
        'brackets',
        [
          '[',
          ']',
        ],
      ],
      [
        '"camelCase"',
        '"camelCase"',
      ],
      [
        '"camelCase"',
        'camelCase',
      ],
      [
        '"camelCase"',
        'camel.Case',
      ],
    ];
  }
  
  /**
   * @covers ::escapeAlias
   * @dataProvider providerEscapeAlias
   */
  public function testEscapeAlias($expected, $name, array $identifier_quote = [
    '"',
    '"',
  ]) : void {
    $mock_pdo = $this->createMock(StubPDO::class);
    $connection = new StubConnection($mock_pdo, [], $identifier_quote);
    $this->assertEquals($expected, $connection->escapeAlias($name));
  }
  
  /**
   * Data provider for testEscapeField.
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected escaped string.
   *   - String to escape.
   */
  public static function providerEscapeFields() {
    return [
      [
        '/title/',
        'title',
        [
          '/',
          '/',
        ],
      ],
      [
        '`backtick`',
        'backtick',
        [
          '`',
          '`',
        ],
      ],
      [
        'test.title',
        'test.title',
        [
          '',
          '',
        ],
      ],
      [
        '"isDefaultRevision"',
        'isDefaultRevision',
      ],
      [
        '"isDefaultRevision"',
        '"isDefaultRevision"',
      ],
      [
        '"entity_test"."isDefaultRevision"',
        'entity_test.isDefaultRevision',
      ],
      [
        '"entity_test"."isDefaultRevision"',
        '"entity_test"."isDefaultRevision"',
      ],
      [
        '"entityTest"."isDefaultRevision"',
        '"entityTest"."isDefaultRevision"',
      ],
      [
        '"entityTest"."isDefaultRevision"',
        'entityTest.isDefaultRevision',
      ],
      [
        '[entityTest].[isDefaultRevision]',
        'entityTest.isDefaultRevision',
        [
          '[',
          ']',
        ],
      ],
    ];
  }
  
  /**
   * @covers ::escapeField
   * @dataProvider providerEscapeFields
   */
  public function testEscapeField($expected, $name, array $identifier_quote = [
    '"',
    '"',
  ]) : void {
    $mock_pdo = $this->createMock(StubPDO::class);
    $connection = new StubConnection($mock_pdo, [], $identifier_quote);
    $this->assertEquals($expected, $connection->escapeField($name));
  }
  
  /**
   * Data provider for testEscapeDatabase.
   *
   * @return array
   *   An indexed array of where each value is an array of arguments to pass to
   *   testEscapeField. The first value is the expected value, and the second
   *   value is the value to test.
   */
  public static function providerEscapeDatabase() {
    return [
      [
        '/name/',
        'name',
        [
          '/',
          '/',
        ],
      ],
      [
        '`backtick`',
        'backtick',
        [
          '`',
          '`',
        ],
      ],
      [
        'anything',
        'any.thing',
        [
          '',
          '',
        ],
      ],
      [
        '"name"',
        'name',
      ],
      [
        '[name]',
        'name',
        [
          '[',
          ']',
        ],
      ],
    ];
  }
  
  /**
   * @covers ::escapeDatabase
   * @dataProvider providerEscapeDatabase
   */
  public function testEscapeDatabase($expected, $name, array $identifier_quote = [
    '"',
    '"',
  ]) : void {
    $mock_pdo = $this->createMock(StubPDO::class);
    $connection = new StubConnection($mock_pdo, [], $identifier_quote);
    $this->assertEquals($expected, $connection->escapeDatabase($name));
  }
  
  /**
   * @covers ::__construct
   */
  public function testIdentifierQuotesAssertCount() : void {
    $this->expectException(\AssertionError::class);
    $this->expectExceptionMessage('\\Drupal\\Core\\Database\\Connection::$identifierQuotes must contain 2 string values');
    $mock_pdo = $this->createMock(StubPDO::class);
    new StubConnection($mock_pdo, [], [
      '"',
    ]);
  }
  
  /**
   * @covers ::__construct
   */
  public function testIdentifierQuotesAssertString() : void {
    $this->expectException(\AssertionError::class);
    $this->expectExceptionMessage('\\Drupal\\Core\\Database\\Connection::$identifierQuotes must contain 2 string values');
    $mock_pdo = $this->createMock(StubPDO::class);
    new StubConnection($mock_pdo, [], [
      0,
      '1',
    ]);
  }
  
  /**
   * @covers ::__construct
   */
  public function testNamespaceDefault() : void {
    $mock_pdo = $this->createMock(StubPDO::class);
    $connection = new StubConnection($mock_pdo, []);
    $this->assertSame('Drupal\\Tests\\Core\\Database\\Stub', $connection->getConnectionOptions()['namespace']);
  }
  
  /**
   * Test rtrim() of query strings.
   *
   * @dataProvider provideQueriesToTrim
   */
  public function testQueryTrim($expected, $query, $options) : void {
    $mock_pdo = $this->getMockBuilder(StubPdo::class)
      ->getMock();
    $connection = new StubConnection($mock_pdo, []);
    $preprocess_method = new \ReflectionMethod($connection, 'preprocessStatement');
    $this->assertSame($expected, $preprocess_method->invoke($connection, $query, $options));
  }
  
  /**
   * Data provider for testQueryTrim().
   *
   * @return array
   *   Array of arrays with the following elements:
   *   - Expected trimmed query.
   *   - Padded query.
   *   - Query options.
   */
  public static function provideQueriesToTrim() {
    return [
      'remove_non_breaking_space' => [
        'SELECT * FROM test',
        "SELECT * FROM test\xa0",
        [],
      ],
      'remove_ordinary_space' => [
        'SELECT * FROM test',
        'SELECT * FROM test ',
        [],
      ],
      'remove_semicolon' => [
        'SELECT * FROM test',
        'SELECT * FROM test;',
        [],
      ],
      'keep_trailing_semicolon' => [
        'SELECT * FROM test;',
        'SELECT * FROM test;',
        [
          'allow_delimiter_in_query' => TRUE,
        ],
      ],
      'remove_semicolon_with_whitespace' => [
        'SELECT * FROM test',
        'SELECT * FROM test; ',
        [],
      ],
      'keep_trailing_semicolon_with_whitespace' => [
        'SELECT * FROM test;',
        'SELECT * FROM test; ',
        [
          'allow_delimiter_in_query' => TRUE,
        ],
      ],
    ];
  }
  
  /**
   * Tests that the proper caller is retrieved from the backtrace.
   *
   * @covers ::findCallerFromDebugBacktrace
   * @covers ::removeDatabaseEntriesFromDebugBacktrace
   * @covers ::getDebugBacktrace
   */
  public function testFindCallerFromDebugBacktrace() : void {
    Database::addConnectionInfo('default', 'default', [
      'driver' => 'test',
      'namespace' => 'Drupal\\Tests\\Core\\Database\\Stub',
    ]);
    $connection = new StubConnection($this->createMock(StubPDO::class), []);
    $result = $connection->findCallerFromDebugBacktrace();
    $this->assertSame([
      'file' => __FILE__,
      'line' => __LINE__ - 3,
      'function' => 'testFindCallerFromDebugBacktrace',
      'class' => 'Drupal\\Tests\\Core\\Database\\ConnectionTest',
      'type' => '->',
      'args' => [],
    ], $result);
  }
  
  /**
   * Tests that a log called by a custom database driver returns proper caller.
   *
   * @param string $driver_namespace
   *   The driver namespace to be tested.
   * @param array $stack
   *   A test debug_backtrace stack.
   * @param array $expected_entry
   *   The expected stack entry.
   *
   * @covers ::findCallerFromDebugBacktrace
   * @covers ::removeDatabaseEntriesFromDebugBacktrace
   *
   * @dataProvider providerMockedBacktrace
   *
   * @group legacy
   */
  public function testFindCallerFromDebugBacktraceWithMockedBacktrace(string $driver_namespace, array $stack, array $expected_entry) : void {
    $mock_builder = $this->getMockBuilder(StubConnection::class);
    $connection = $mock_builder->onlyMethods([
      'getDebugBacktrace',
      'getConnectionOptions',
    ])
      ->setConstructorArgs([
      $this->createMock(StubPDO::class),
      [],
    ])
      ->getMock();
    $connection->expects($this->once())
      ->method('getConnectionOptions')
      ->willReturn([
      'driver' => 'test',
      'namespace' => $driver_namespace,
    ]);
    $connection->expects($this->once())
      ->method('getDebugBacktrace')
      ->willReturn($stack);
    $result = $connection->findCallerFromDebugBacktrace();
    $this->assertEquals($expected_entry, $result);
  }
  
  /**
   * Provides data for testFindCallerFromDebugBacktraceWithMockedBacktrace.
   *
   * @return array[]
   *   A associative array of simple arrays, each having the following elements:
   *   - the contrib driver PHP namespace
   *   - a test debug_backtrace stack
   *   - the stack entry expected to be returned.
   *
   * @see ::testFindCallerFromDebugBacktraceWithMockedBacktrace()
   */
  public static function providerMockedBacktrace() : array {
    $stack = [
      [
        'file' => '/var/www/core/lib/Drupal/Core/Database/Log.php',
        'line' => 125,
        'function' => 'findCaller',
        'class' => 'Drupal\\Core\\Database\\Log',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/libraries/test/lib/Statement.php',
        'line' => 264,
        'function' => 'log',
        'class' => 'Drupal\\Core\\Database\\Log',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/libraries/test/lib/Connection.php',
        'line' => 213,
        'function' => 'execute',
        'class' => 'Drupal\\Driver\\Database\\dbal\\Statement',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/core/tests/Drupal/KernelTests/Core/Database/LoggingTest.php',
        'line' => 23,
        'function' => 'query',
        'class' => 'Drupal\\Driver\\Database\\dbal\\Connection',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/vendor/phpunit/phpunit/src/Framework/TestCase.php',
        'line' => 1154,
        'function' => 'testEnableLogging',
        'class' => 'Drupal\\KernelTests\\Core\\Database\\LoggingTest',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/vendor/phpunit/phpunit/src/Framework/TestCase.php',
        'line' => 842,
        'function' => 'runTest',
        'class' => 'PHPUnit\\Framework\\TestCase',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/vendor/phpunit/phpunit/src/Framework/TestResult.php',
        'line' => 693,
        'function' => 'runBare',
        'class' => 'PHPUnit\\Framework\\TestCase',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => '/var/www/vendor/phpunit/phpunit/src/Framework/TestCase.php',
        'line' => 796,
        'function' => 'run',
        'class' => 'PHPUnit\\Framework\\TestResult',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => 'Standard input code',
        'line' => 57,
        'function' => 'run',
        'class' => 'PHPUnit\\Framework\\TestCase',
        'object' => 'test',
        'type' => '->',
        'args' => [
          0 => 'test',
        ],
      ],
      [
        'file' => 'Standard input code',
        'line' => 111,
        'function' => '__phpunit_run_isolated_test',
        'args' => [
          0 => 'test',
        ],
      ],
    ];
    return [
      // Test that if the driver namespace is in the stack trace, the first
      // non-database entry is returned.
'contrib driver namespace' => [
        'Drupal\\Driver\\Database\\dbal',
        $stack,
        [
          'class' => 'Drupal\\KernelTests\\Core\\Database\\LoggingTest',
          'function' => 'testEnableLogging',
          'file' => '/var/www/core/tests/Drupal/KernelTests/Core/Database/LoggingTest.php',
          'line' => 23,
          'type' => '->',
          'args' => [
            0 => 'test',
          ],
        ],
      ],
      // Extreme case, should not happen at normal runtime - if the driver
      // namespace is not in the stack trace, the first entry to a method
      // in core database namespace is returned.
'missing driver namespace' => [
        'Drupal\\Driver\\Database\\fake',
        $stack,
        [
          'class' => 'Drupal\\Driver\\Database\\dbal\\Statement',
          'function' => 'execute',
          'file' => '/var/www/libraries/test/lib/Statement.php',
          'line' => 264,
          'type' => '->',
          'args' => [
            0 => 'test',
          ],
        ],
      ],
    ];
  }
  
  /**
   * Provides data for testSupportedFetchModes.
   *
   * @return array
   *   An associative array of simple arrays, each having the following
   *   elements:
   *   - a PDO fetch mode.
   */
  public static function providerSupportedLegacyFetchModes() : array {
    return [
      'FETCH_ASSOC' => [
        \PDO::FETCH_ASSOC,
      ],
      'FETCH_CLASS' => [
        \PDO::FETCH_CLASS,
      ],
      'FETCH_CLASS | FETCH_PROPS_LATE' => [
        \PDO::FETCH_CLASS | \PDO::FETCH_PROPS_LATE,
      ],
      'FETCH_COLUMN' => [
        \PDO::FETCH_COLUMN,
      ],
      'FETCH_NUM' => [
        \PDO::FETCH_NUM,
      ],
      'FETCH_OBJ' => [
        \PDO::FETCH_OBJ,
      ],
    ];
  }
  
  /**
   * Tests supported fetch modes.
   */
  public function testSupportedLegacyFetchModes(int $mode) : void {
    $this->expectDeprecation("Passing the \$mode argument as an integer to setFetchMode() is deprecated in drupal:11.2.0 and is removed from drupal:12.0.0. Use a case of \\Drupal\\Core\\Database\\Statement\\FetchAs enum instead. See https://www.drupal.org/node/3488338");
    $mockPdo = $this->createMock(StubPDO::class);
    $mockConnection = new StubConnection($mockPdo, []);
    $statement = new StatementPrefetchIterator($mockPdo, $mockConnection, '');
    $this->assertInstanceOf(StatementPrefetchIterator::class, $statement);
    $statement->setFetchMode($mode);
  }
  
  /**
   * Provides data for testSupportedFetchModes.
   *
   * @return array<string,array<\Drupal\Core\Database\Statement\FetchAs>>
   *   The FetchAs cases.
   */
  public static function providerSupportedFetchModes() : array {
    return [
      'Associative array' => [
        FetchAs::Associative,
      ],
      'Classed object' => [
        FetchAs::ClassObject,
      ],
      'Single column' => [
        FetchAs::Column,
      ],
      'Simple array' => [
        FetchAs::List,
      ],
      'Standard object' => [
        FetchAs::Object,
      ],
    ];
  }
  
  /**
   * Tests supported fetch modes.
   *
   * @dataProvider providerSupportedFetchModes
   */
  public function testSupportedFetchModes(FetchAs $mode) : void {
    $mockPdo = $this->createMock(StubPDO::class);
    $mockConnection = new StubConnection($mockPdo, []);
    $statement = new StatementPrefetchIterator($mockPdo, $mockConnection, '');
    $this->assertInstanceOf(StatementPrefetchIterator::class, $statement);
    $statement->setFetchMode($mode);
  }
  
  /**
   * Provides data for testUnsupportedFetchModes.
   *
   * @return array
   *   An associative array of simple arrays, each having the following
   *   elements:
   *   - a PDO fetch mode.
   */
  public static function providerUnsupportedFetchModes() : array {
    return [
      'FETCH_DEFAULT' => [
        \PDO::FETCH_DEFAULT,
      ],
      'FETCH_LAZY' => [
        \PDO::FETCH_LAZY,
      ],
      'FETCH_BOTH' => [
        \PDO::FETCH_BOTH,
      ],
      'FETCH_BOUND' => [
        \PDO::FETCH_BOUND,
      ],
      'FETCH_INTO' => [
        \PDO::FETCH_INTO,
      ],
      'FETCH_FUNC' => [
        \PDO::FETCH_FUNC,
      ],
      'FETCH_NAMED' => [
        \PDO::FETCH_NAMED,
      ],
      'FETCH_KEY_PAIR' => [
        \PDO::FETCH_KEY_PAIR,
      ],
      'FETCH_CLASS | FETCH_CLASSTYPE' => [
        \PDO::FETCH_CLASS | \PDO::FETCH_CLASSTYPE,
      ],
    ];
  }
  
  /**
   * Tests unsupported legacy fetch modes.
   */
  public function testUnsupportedFetchModes(int $mode) : void {
    $this->expectDeprecation("Passing the \$mode argument as an integer to setFetchMode() is deprecated in drupal:11.2.0 and is removed from drupal:12.0.0. Use a case of \\Drupal\\Core\\Database\\Statement\\FetchAs enum instead. See https://www.drupal.org/node/3488338");
    $this->expectException(\RuntimeException::class);
    $this->expectExceptionMessageMatches("/^Fetch mode FETCH_.* is not supported\\. Use supported modes only/");
    $mockPdo = $this->createMock(StubPDO::class);
    $mockConnection = new StubConnection($mockPdo, []);
    $statement = new StatementPrefetchIterator($mockPdo, $mockConnection, '');
    $this->assertInstanceOf(StatementPrefetchIterator::class, $statement);
    $statement->setFetchMode($mode);
  }
  
  /**
   * {@inheritdoc}
   */
  public function tearDown() : void {
    parent::tearDown();
    // Removes the default connection added by the
    // testFindCallerFromDebugBacktrace test.
    Database::removeConnection('default');
  }

}

Classes

Title Deprecated Summary
ConnectionTest Tests the Connection class.

Buggy or inaccurate documentation? Please file an issue. Need support? Need help programming? Connect with the Drupal community.