TestUtil.php 4.84 KB
Newer Older
1
<?php
romanb's avatar
romanb committed
2

3
namespace Doctrine\Tests;
romanb's avatar
romanb committed
4

romanb's avatar
romanb committed
5 6
/**
 * TestUtil is a class with static utility methods used during tests.
7
 *
romanb's avatar
romanb committed
8 9
 * @author robo
 */
10
class TestUtil
11
{
romanb's avatar
romanb committed
12 13 14
    /**
     * Gets a <b>real</b> database connection using the following parameters
     * of the $GLOBALS array:
15
     *
romanb's avatar
romanb committed
16 17 18 19 20 21
     * 'db_type' : The name of the Doctrine DBAL database driver to use.
     * 'db_username' : The username to use for connecting.
     * 'db_password' : The password to use for connecting.
     * 'db_host' : The hostname of the database to connect to.
     * 'db_name' : The name of the database to connect to.
     * 'db_port' : The port of the database to connect to.
22
     *
romanb's avatar
romanb committed
23 24 25
     * Usually these variables of the $GLOBALS array are filled by PHPUnit based
     * on an XML configuration file. If no such parameters exist, an SQLite
     * in-memory database is used.
26
     *
romanb's avatar
romanb committed
27 28 29
     * IMPORTANT:
     * 1) Each invocation of this method returns a NEW database connection.
     * 2) The database is dropped and recreated to ensure it's clean.
30
     *
romanb's avatar
romanb committed
31 32
     * @return Doctrine\DBAL\Connection The database connection instance.
     */
romanb's avatar
romanb committed
33 34 35
    public static function getConnection()
    {
        if (isset($GLOBALS['db_type'], $GLOBALS['db_username'], $GLOBALS['db_password'],
36 37 38 39
                $GLOBALS['db_host'], $GLOBALS['db_name'], $GLOBALS['db_port']) &&
           isset($GLOBALS['tmpdb_type'], $GLOBALS['tmpdb_username'], $GLOBALS['tmpdb_password'],
                $GLOBALS['tmpdb_host'], $GLOBALS['tmpdb_name'], $GLOBALS['tmpdb_port'])) {
            $realDbParams = array(
40 41 42 43
                'driver' => $GLOBALS['db_type'],
                'user' => $GLOBALS['db_username'],
                'password' => $GLOBALS['db_password'],
                'host' => $GLOBALS['db_host'],
44 45
                'dbname' => $GLOBALS['db_name'],
                'port' => $GLOBALS['db_port']
46
            );
47 48 49 50 51 52 53 54
            $tmpDbParams = array(
                'driver' => $GLOBALS['tmpdb_type'],
                'user' => $GLOBALS['tmpdb_username'],
                'password' => $GLOBALS['tmpdb_password'],
                'host' => $GLOBALS['tmpdb_host'],
                'dbname' => $GLOBALS['tmpdb_name'],
                'port' => $GLOBALS['tmpdb_port']
            );
55

56 57 58 59 60 61 62 63
            if (isset($GLOBALS['db_unix_socket'])) {
                $realDbParams['unix_socket'] = $GLOBALS['db_unix_socket'];
            }

            if (isset($GLOBALS['tmpdb_unix_socket'])) {
                $tmpDbParams['unix_socket'] = $GLOBALS['tmpdb_unix_socket'];
            }

64
            $realConn = \Doctrine\DBAL\DriverManager::getConnection($realDbParams);
65

66 67 68 69 70 71 72 73 74 75 76 77 78
            $platform  = $realConn->getDatabasePlatform();

            if ($platform->supportsCreateDropDatabase()) {
                $dbname = $realConn->getDatabase();
                // Connect to tmpdb in order to drop and create the real test db.
                $tmpConn = \Doctrine\DBAL\DriverManager::getConnection($tmpDbParams);
                $realConn->close();

                $tmpConn->getSchemaManager()->dropDatabase($dbname);
                $tmpConn->getSchemaManager()->createDatabase($dbname);

                $tmpConn->close();
            } else {
79 80
                $sm = $realConn->getSchemaManager();

81 82 83
                /* @var $schema Schema */
                $schema = $sm->createSchema();
                $stmts = $schema->toDropSql($realConn->getDatabasePlatform());
84

85 86
                foreach ($stmts AS $stmt) {
                    $realConn->exec($stmt);
87 88
                }
            }
89

90
            $conn = \Doctrine\DBAL\DriverManager::getConnection($realDbParams, null, null);
romanb's avatar
romanb committed
91
        } else {
92
            $params = array(
romanb's avatar
romanb committed
93
                'driver' => 'pdo_sqlite',
94
                'memory' => true
95
            );
96 97 98 99
            if (isset($GLOBALS['db_path'])) {
                $params['path'] = $GLOBALS['db_path'];
                unlink($GLOBALS['db_path']);
            }
100
            $conn = \Doctrine\DBAL\DriverManager::getConnection($params);
101
        }
102

103 104 105 106 107 108 109 110
        if (isset($GLOBALS['db_event_subscribers'])) {
            $evm = $conn->getEventManager();
            foreach (explode(",", $GLOBALS['db_event_subscribers']) AS $subscriberClass) {
                $subscriberInstance = new $subscriberClass();
                $evm->addEventSubscriber($subscriberInstance);
            }
        }

111
        return $conn;
romanb's avatar
romanb committed
112
    }
113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130

    /**
     * @return \Doctrine\DBAL\Connection
     */
    public static function getTempConnection()
    {
        $tmpDbParams = array(
            'driver' => $GLOBALS['tmpdb_type'],
            'user' => $GLOBALS['tmpdb_username'],
            'password' => $GLOBALS['tmpdb_password'],
            'host' => $GLOBALS['tmpdb_host'],
            'dbname' => $GLOBALS['tmpdb_name'],
            'port' => $GLOBALS['tmpdb_port']
        );

        // Connect to tmpdb in order to drop and create the real test db.
        return \Doctrine\DBAL\DriverManager::getConnection($tmpDbParams);
    }
131
}