DbTestHelperTrait.php

Go to the documentation of this file.

<?php
 
namespace Sabre\DAV;
 
use PDO;
use PDOException;
 
class DbCache {
 
    static $cache = [];
 
}
 
trait DbTestHelperTrait {
 
    public $driver = null;
 
    function getDb() {
 
        if (!$this->driver) {
            throw new \Exception('You must set the $driver public property');
        }
 
        if (array_key_exists($this->driver, DbCache::$cache)) {
            $pdo = DbCache::$cache[$this->driver];
            if ($pdo === null) {
                $this->markTestSkipped($this->driver . ' was not enabled, not correctly configured or of the wrong version');
            }
            return $pdo;
        }
 
        try {
 
            switch ($this->driver) {
 
                case 'mysql' :
                    $pdo = new PDO(SABRE_MYSQLDSN, SABRE_MYSQLUSER, SABRE_MYSQLPASS);
                    break;
                case 'sqlite' :
                    $pdo = new \PDO('sqlite:' . SABRE_TEMPDIR . '/testdb');
                    break;
                case 'pgsql' :
                    $pdo = new \PDO(SABRE_PGSQLDSN);
                    $version = $pdo->query('SELECT VERSION()')->fetchColumn();
                    preg_match('|([0-9\.]){5,}|', $version, $matches);
                    $version = $matches[0];
                    if (version_compare($version, '9.5.0', '<')) {
                        DbCache::$cache[$this->driver] = null;
                        $this->markTestSkipped('We require at least Postgres 9.5. This server is running ' . $version);
                    }
                    break;
 
 
 
            }
            $pdo->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
 
        } catch (PDOException $e) {
 
            $this->markTestSkipped($this->driver . ' was not enabled or not correctly configured. Error message: ' . $e->getMessage());
 
        }
 
        DbCache::$cache[$this->driver] = $pdo;
        return $pdo;
 
    }
 
    function getPDO() {
 
        return $this->getDb();
 
    }
 
    function createSchema($schemaName) {
 
        $db = $this->getDb();
 
        $queries = file_get_contents(
            __DIR__ . '/../../../examples/sql/' . $this->driver . '.' . $schemaName . '.sql'
        );
 
        foreach (explode(';', $queries) as $query) {
 
            if (trim($query) === '') {
                continue;
            }
 
            $db->exec($query);
 
        }
 
    }
 
    function dropTables($tableNames) {
 
        $tableNames = (array)$tableNames;
        $db = $this->getDb();
        foreach ($tableNames as $tableName) {
            $db->exec('DROP TABLE IF EXISTS ' . $tableName);
        }
 
 
    }
 
    function tearDown() {
 
        switch ($this->driver) {
 
            case 'sqlite' :
                // Recreating sqlite, just in case
                unset(DbCache::$cache[$this->driver]);
                unlink(SABRE_TEMPDIR . '/testdb');
        }
 
    }
 
}