DatabaseTestBase.php

<?php

namespace Drupal\KernelTests\Core\Database;

use Drupal\Core\Database\Database;
use Drupal\KernelTests\KernelTestBase;

/**
 * Base class for databases database tests.
 *
 * Because all database tests share the same test data, we can centralize that
 * here.
 */
abstract class DatabaseTestBase extends KernelTestBase {
    public static $modules = [
        'database_test',
    ];
    
    /**
     * The database connection for testing.
     *
     * @var \Drupal\Core\Database\Connection
     */
    protected $connection;
    protected function setUp() {
        parent::setUp();
        $this->connection = Database::getConnection();
        $this->installSchema('database_test', [
            'test',
            'test_classtype',
            'test_people',
            'test_people_copy',
            'test_one_blob',
            'test_two_blobs',
            'test_task',
            'test_null',
            'test_serialized',
            'test_special_columns',
            'TEST_UPPERCASE',
            'virtual',
        ]);
        self::addSampleData();
    }
    
    /**
     * Sets up tables for NULL handling.
     */
    public function ensureSampleDataNull() {
        $this->connection
            ->insert('test_null')
            ->fields([
            'name',
            'age',
        ])
            ->values([
            'name' => 'Kermit',
            'age' => 25,
        ])
            ->values([
            'name' => 'Fozzie',
            'age' => NULL,
        ])
            ->values([
            'name' => 'Gonzo',
            'age' => 27,
        ])
            ->execute();
    }
    
    /**
     * Sets up our sample data.
     */
    public static function addSampleData() {
        $connection = Database::getConnection();
        // We need the IDs, so we can't use a multi-insert here.
        $john = $connection->insert('test')
            ->fields([
            'name' => 'John',
            'age' => 25,
            'job' => 'Singer',
        ])
            ->execute();
        $george = $connection->insert('test')
            ->fields([
            'name' => 'George',
            'age' => 27,
            'job' => 'Singer',
        ])
            ->execute();
        $connection->insert('test')
            ->fields([
            'name' => 'Ringo',
            'age' => 28,
            'job' => 'Drummer',
        ])
            ->execute();
        $paul = $connection->insert('test')
            ->fields([
            'name' => 'Paul',
            'age' => 26,
            'job' => 'Songwriter',
        ])
            ->execute();
        $connection->insert('test_classtype')
            ->fields([
            'classname' => 'Drupal\\Tests\\system\\Functional\\Database\\FakeRecord',
            'name' => 'Kay',
            'age' => 26,
            'job' => 'Web Developer',
        ])
            ->execute();
        $connection->insert('test_people')
            ->fields([
            'name' => 'Meredith',
            'age' => 30,
            'job' => 'Speaker',
        ])
            ->execute();
        $connection->insert('test_task')
            ->fields([
            'pid',
            'task',
            'priority',
        ])
            ->values([
            'pid' => $john,
            'task' => 'eat',
            'priority' => 3,
        ])
            ->values([
            'pid' => $john,
            'task' => 'sleep',
            'priority' => 4,
        ])
            ->values([
            'pid' => $john,
            'task' => 'code',
            'priority' => 1,
        ])
            ->values([
            'pid' => $george,
            'task' => 'sing',
            'priority' => 2,
        ])
            ->values([
            'pid' => $george,
            'task' => 'sleep',
            'priority' => 2,
        ])
            ->values([
            'pid' => $paul,
            'task' => 'found new band',
            'priority' => 1,
        ])
            ->values([
            'pid' => $paul,
            'task' => 'perform at superbowl',
            'priority' => 3,
        ])
            ->execute();
        $connection->insert('test_special_columns')
            ->fields([
            'id' => 1,
            'offset' => 'Offset value 1',
            'function' => 'Function value 1',
        ])
            ->execute();
        $connection->insert('virtual')
            ->fields([
            'id' => 1,
            'function' => 'Function value 1',
        ])
            ->execute();
    }

}