* @copyright 2012 - 2023 Timothy J. Warren * @license http://www.opensource.org/licenses/mit-license.html MIT License * @link https://git.timshomepage.net/aviat/Query * @version 4.1.0 */ namespace Query\Tests\Drivers\PgSQL; use PDO; use Query\Drivers\Pgsql\Driver; use Query\Tests\BaseDriverTest; use TypeError; use UnitTestCase; /** * PgTest class. * * @extends DBTest * @requires extension pdo_pgsql */ class PgSQLDriverTest extends BaseDriverTest { protected function setUp(): void { // If the database isn't installed, skip the tests if ( ! class_exists(Driver::class)) { $this->markTestSkipped('Postgres extension for PDO not loaded'); } } public static function setUpBeforeClass(): void { $params = get_json_config(); if ($var = getenv('TRAVIS')) { self::$db = new Driver('host=127.0.0.1;port=5432;dbname=test', 'postgres'); } // Attempt to connect, if there is a test config file elseif ($params !== FALSE) { $params = $params->pgsql; self::$db = new Driver("pgsql:host={$params->host};dbname={$params->database};port=5432", $params->user, $params->pass); } self::$db->setTablePrefix('create_'); } public function testExists(): void { $drivers = PDO::getAvailableDrivers(); $this->assertTrue(in_array('pgsql', $drivers, TRUE)); } public function testConnection(): void { if (empty(self::$db)) { return; } $this->assertIsA(self::$db, Driver::class); } public function testCreateTable(): void { // self::$db->exec(file_get_contents(QTEST_DIR.'/db_files/pgsql.sql')); // Drop the table(s) if they exist $sql = 'DROP TABLE IF EXISTS "create_test"'; self::$db->query($sql); $sql = 'DROP TABLE IF EXISTS "create_join"'; self::$db->query($sql); //Attempt to create the table $sql = self::$db->getUtil()->createTable( 'create_test', [ 'id' => 'integer', 'key' => 'TEXT', 'val' => 'TEXT', ], [ 'id' => 'PRIMARY KEY', ] ); self::$db->query($sql); //Attempt to create the table $sql = self::$db->getUtil()->createTable( 'create_join', [ 'id' => 'integer', 'key' => 'TEXT', 'val' => 'TEXT', ], [ 'id' => 'PRIMARY KEY', ] ); self::$db->query($sql); //echo $sql.'
'; //Reset //unset(self::$db); //$this->setUp(); //Check $dbs = self::$db->getTables(); $this->assertTrue(in_array('create_test', $dbs, TRUE)); } public function testTruncate(): void { self::$db->truncate('test'); $this->assertEquals(0, self::$db->countAll('test')); self::$db->truncate('join'); $this->assertEquals(0, self::$db->countAll('join')); } public function testPreparedStatements(): void { $sql = <<<'SQL' INSERT INTO "create_test" ("id", "key", "val") VALUES (?,?,?) SQL; $statement = self::$db->prepareQuery($sql, [1, 'boogers', 'Gross']); $statement->execute(); $res = self::$db->query('SELECT * FROM "create_test" WHERE "id"=1') ->fetch(PDO::FETCH_ASSOC); $this->assertEquals([ 'id' => 1, 'key' => 'boogers', 'val' => 'Gross', ], $res); } public function testBadPreparedStatement(): void { if (is_a($this, UnitTestCase::class)) { $this->markTestSkipped(); return; } $this->expectException(TypeError::class); $sql = <<<'SQL' INSERT INTO "create_test" ("id", "key", "val") VALUES (?,?,?) SQL; self::$db->prepareQuery($sql, 'foo'); } public function testPrepareExecute(): void { if (empty(self::$db)) { return; } $sql = <<<'SQL' INSERT INTO "create_test" ("id", "key", "val") VALUES (?,?,?) SQL; self::$db->prepareExecute($sql, [ 2, 'works', 'also?', ]); $res = self::$db->query('SELECT * FROM "create_test" WHERE "id"=2') ->fetch(PDO::FETCH_ASSOC); $this->assertEquals([ 'id' => 2, 'key' => 'works', 'val' => 'also?', ], $res); } public function testCommitTransaction(): void { if (empty(self::$db)) { return; } self::$db->beginTransaction(); $sql = 'INSERT INTO "create_test" ("id", "key", "val") VALUES (10, 12, 14)'; self::$db->query($sql); $res = self::$db->commit(); $this->assertTrue($res); } public function testRollbackTransaction(): void { if (empty(self::$db)) { return; } self::$db->beginTransaction(); $sql = 'INSERT INTO "create_test" ("id", "key", "val") VALUES (182, 96, 43)'; self::$db->query($sql); $res = self::$db->rollback(); $this->assertTrue($res); } public function testGetSchemas(): void { $this->assertIsArray(self::$db->getSchemas()); } public function testGetDBs(): void { $this->assertIsArray(self::$db->getDbs()); } public function testGetFunctions(): void { $this->assertNull(self::$db->getFunctions()); } }