← Back to branch summary

~ubuntu-branches/ubuntu/natty/phpunit/natty

« back to all changes in this revision

Viewing changes to PHPUnit-3.4.14/PHPUnit/Extensions/Database/DB/MetaData/PgSQL.php

  • Committer: Package Import Robot
  • Author(s): Ivan Borzenkov
  • Date: 2010-12-11 18:19:39 UTC
  • mfrom: (0.11.1) (1.5.3) (12.1.7 sid)
  • Revision ID: package-import@ubuntu.com-20101211181939-8650nbu08hf2z9v1
Tags: 3.5.5-2
fix doc-base-file-references-missing-file

Show diffs side-by-side

added added

removed removed

Lines of Context:
PHPUnit-3.4.14/PHPUnit/Extensions/Database/DB/MetaData/PgSQL.php
1
 
<?php
2
 
/**
3
 
 * PHPUnit
4
 
 *
5
 
 * Copyright (c) 2002-2010, Sebastian Bergmann <sb@sebastian-bergmann.de>.
6
 
 * All rights reserved.
7
 
 *
8
 
 * Redistribution and use in source and binary forms, with or without
9
 
 * modification, are permitted provided that the following conditions
10
 
 * are met:
11
 
 *
12
 
 *   * Redistributions of source code must retain the above copyright
13
 
 *     notice, this list of conditions and the following disclaimer.
14
 
 *
15
 
 *   * Redistributions in binary form must reproduce the above copyright
16
 
 *     notice, this list of conditions and the following disclaimer in
17
 
 *     the documentation and/or other materials provided with the
18
 
 *     distribution.
19
 
 *
20
 
 *   * Neither the name of Sebastian Bergmann nor the names of his
21
 
 *     contributors may be used to endorse or promote products derived
22
 
 *     from this software without specific prior written permission.
23
 
 *
24
 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25
 
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26
 
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
27
 
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
28
 
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
29
 
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
30
 
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
31
 
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
32
 
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
33
 
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
34
 
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35
 
 * POSSIBILITY OF SUCH DAMAGE.
36
 
 *
37
 
 * @category   Testing
38
 
 * @package    PHPUnit
39
 
 * @author     Mike Lively <m@digitalsandwich.com>
40
 
 * @copyright  2002-2010 Sebastian Bergmann <sb@sebastian-bergmann.de>
41
 
 * @license    http://www.opensource.org/licenses/bsd-license.php  BSD License
42
 
 * @link       http://www.phpunit.de/
43
 
 * @since      File available since Release 3.2.0
44
 
 */
45
 
 
46
 
require_once 'PHPUnit/Framework.php';
47
 
require_once 'PHPUnit/Util/Filter.php';
48
 
require_once 'PHPUnit/Extensions/Database/DB/MetaData/InformationSchema.php';
49
 
 
50
 
PHPUnit_Util_Filter::addFileToFilter(__FILE__, 'PHPUNIT');
51
 
 
52
 
/**
53
 
 * Provides functionality to retrieve meta data from a PostgreSQL database.
54
 
 *
55
 
 * @category   Testing
56
 
 * @package    PHPUnit
57
 
 * @author     Mike Lively <m@digitalsandwich.com>
58
 
 * @copyright  2002-2010 Sebastian Bergmann <sb@sebastian-bergmann.de>
59
 
 * @license    http://www.opensource.org/licenses/bsd-license.php  BSD License
60
 
 * @version    Release: 3.4.14
61
 
 * @link       http://www.phpunit.de/
62
 
 * @since      Class available since Release 3.2.0
63
 
 */
64
 
class PHPUnit_Extensions_Database_DB_MetaData_PgSQL extends PHPUnit_Extensions_Database_DB_MetaData
65
 
{
66
 
 
67
 
    /**
68
 
     * Returns an array containing the names of all the tables in the database.
69
 
     *
70
 
     * @return array
71
 
     */
72
 
    public function getTableNames()
73
 
    {
74
 
        $query = "
75
 
            SELECT DISTINCT
76
 
                TABLE_NAME
77
 
            FROM INFORMATION_SCHEMA.TABLES
78
 
            WHERE
79
 
                TABLE_TYPE='BASE TABLE' AND
80
 
                TABLE_SCHEMA = ?
81
 
            ORDER BY TABLE_NAME
82
 
        ";
83
 
 
84
 
        $statement = $this->pdo->prepare($query);
85
 
        $statement->execute(array($this->getSchema()));
86
 
 
87
 
        $tableNames = array();
88
 
        while ($tableName = $statement->fetchColumn(0)) {
89
 
            $tableNames[] = $tableName;
90
 
        }
91
 
 
92
 
        return $tableNames;
93
 
    }
94
 
 
95
 
        /**
96
 
         * Returns an array containing the names of all the columns in the
97
 
         * $tableName table,
98
 
         *
99
 
         * @param string $tableName
100
 
         * @return array
101
 
         */
102
 
        public function getTableColumns($tableName)
103
 
        {
104
 
                if (!isset($this->columns[$tableName])) {
105
 
                        $this->loadColumnInfo($tableName);
106
 
                }
107
 
 
108
 
                return $this->columns[$tableName];
109
 
        }
110
 
 
111
 
        /**
112
 
         * Returns an array containing the names of all the primary key columns in
113
 
         * the $tableName table.
114
 
         *
115
 
         * @param string $tableName
116
 
         * @return array
117
 
         */
118
 
        public function getTablePrimaryKeys($tableName)
119
 
        {
120
 
                if (!isset($this->keys[$tableName])) {
121
 
                        $this->loadColumnInfo($tableName);
122
 
                }
123
 
                
124
 
                return $this->keys[$tableName];
125
 
        }
126
 
 
127
 
        /**
128
 
         * Loads column info from a database table.
129
 
     *
130
 
     * @param string $tableName
131
 
     */
132
 
    protected function loadColumnInfo($tableName)
133
 
    {
134
 
        $this->columns[$tableName] = array();
135
 
        $this->keys[$tableName] = array();
136
 
 
137
 
        $columnQuery = "
138
 
            SELECT DISTINCT
139
 
                COLUMN_NAME, ORDINAL_POSITION
140
 
            FROM INFORMATION_SCHEMA.COLUMNS
141
 
            WHERE
142
 
                TABLE_NAME = ? AND
143
 
                TABLE_SCHEMA = ?
144
 
            ORDER BY ORDINAL_POSITION
145
 
        ";
146
 
 
147
 
        $columnStatement = $this->pdo->prepare($columnQuery);
148
 
        $columnStatement->execute(array($tableName, $this->getSchema()));
149
 
 
150
 
        while ($columName = $columnStatement->fetchColumn(0)) {
151
 
            $this->columns[$tableName][] = $columName;
152
 
        }
153
 
 
154
 
        $keyQuery = "
155
 
                        SELECT
156
 
                                KCU.COLUMN_NAME,
157
 
                                KCU.ORDINAL_POSITION
158
 
                        FROM
159
 
                                INFORMATION_SCHEMA.KEY_COLUMN_USAGE as KCU
160
 
                        LEFT JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS as TC
161
 
                                ON TC.TABLE_NAME = KCU.TABLE_NAME
162
 
                        WHERE
163
 
                                TC.CONSTRAINT_TYPE = 'PRIMARY KEY' AND
164
 
                                TC.TABLE_NAME = ? AND
165
 
                                TC.TABLE_SCHEMA = ?
166
 
                        ORDER BY
167
 
                                KCU.ORDINAL_POSITION ASC
168
 
        ";
169
 
 
170
 
        $keyStatement = $this->pdo->prepare($keyQuery);
171
 
        $keyStatement->execute(array($tableName, $this->getSchema()));
172
 
 
173
 
        while ($columName = $keyStatement->fetchColumn(0)) {
174
 
            $this->keys[$tableName][] = $columName;
175
 
        }
176
 
    }
177
 
 
178
 
        /**
179
 
         * Returns the schema for the connection.
180
 
         *
181
 
         * @return string
182
 
         */
183
 
        public function getSchema()
184
 
        {
185
 
                if (empty($this->schema)) {
186
 
                        return 'public';
187
 
                } else {
188
 
                        return $this->schema;
189
 
                }
190
 
        }
191
 
 
192
 
    /**
193
 
     * Returns true if the rdbms allows cascading
194
 
     *
195
 
     * @return bool
196
 
     */
197
 
    public function allowsCascading()
198
 
    {
199
 
        return TRUE;
200
 
    }
201
 
}
202
 
?>
 
 
b'\\ No newline at end of file'