264 lines
7.8 KiB
PHP
Executable file
264 lines
7.8 KiB
PHP
Executable file
<?php
|
|
|
|
namespace Tests\PHPCensor;
|
|
|
|
use PHPCensor\Config;
|
|
use PHPCensor\Database;
|
|
use PHPCensor\Store\Factory;
|
|
use PHPCensor\Model\Project;
|
|
use PHPCensor\Model\ProjectGroup;
|
|
|
|
class StorePostgresqlTest extends \PHPUnit_Extensions_Database_TestCase
|
|
{
|
|
/**
|
|
* @var \PHPUnit_Extensions_Database_DB_DefaultDatabaseConnection|null
|
|
*/
|
|
protected $connection = null;
|
|
|
|
/**
|
|
* @param string $name
|
|
* @param array $data
|
|
* @param string $dataName
|
|
*/
|
|
public function __construct($name = null, array $data = [], $dataName = '')
|
|
{
|
|
parent::__construct($name, $data, $dataName);
|
|
|
|
if (extension_loaded('pgsql')) {
|
|
if (null === $this->connection) {
|
|
try {
|
|
$pdo = new \PDO(
|
|
'pgsql:host=localhost;dbname=' . POSTGRESQL_DBNAME,
|
|
POSTGRESQL_USER,
|
|
POSTGRESQL_PASSWORD
|
|
);
|
|
|
|
$this->connection = $this->createDefaultDBConnection($pdo, POSTGRESQL_DBNAME);
|
|
|
|
$this->connection->getConnection()->query('
|
|
CREATE TABLE IF NOT EXISTS "project_group" (
|
|
"id" SERIAL,
|
|
"title" character varying(100) NOT NULL,
|
|
"create_date" timestamp without time zone,
|
|
"user_id" integer NOT NULL DEFAULT 0,
|
|
PRIMARY KEY ("id")
|
|
)
|
|
');
|
|
} catch (\PDOException $ex) {
|
|
$this->connection = null;
|
|
}
|
|
}
|
|
} else {
|
|
$this->connection = null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @return \PHPUnit_Extensions_Database_DB_IDatabaseConnection
|
|
*/
|
|
protected function getConnection()
|
|
{
|
|
if (null === $this->connection) {
|
|
$this->markTestSkipped('Test skipped because PostgreSQL database/user/extension doesn`t exist.');
|
|
}
|
|
|
|
return $this->connection;
|
|
}
|
|
|
|
/**
|
|
* @return \PHPUnit_Extensions_Database_DataSet_IDataSet
|
|
*/
|
|
protected function getDataSet()
|
|
{
|
|
return $this->createArrayDataSet([
|
|
'project_group' => [[
|
|
'id' => 1,
|
|
'title' => 'group 1',
|
|
'create_date' => null,
|
|
'user_id' => 0,
|
|
], [
|
|
'id' => 2,
|
|
'title' => 'group 2',
|
|
'create_date' => null,
|
|
'user_id' => 0,
|
|
], [
|
|
'id' => 3,
|
|
'title' => 'group 3',
|
|
'create_date' => null,
|
|
'user_id' => 1,
|
|
], [
|
|
'id' => 4,
|
|
'title' => 'group 4',
|
|
'create_date' => null,
|
|
'user_id' => 1,
|
|
], [
|
|
'id' => 5,
|
|
'title' => 'group 5',
|
|
'create_date' => '2018-01-01 01:01:00',
|
|
'user_id' => 2,
|
|
], [
|
|
'id' => 6,
|
|
'title' => 'group 6',
|
|
'create_date' => '2018-02-01 01:01:00',
|
|
'user_id' => 3,
|
|
], [
|
|
'id' => 7,
|
|
'title' => 'group 7',
|
|
'create_date' => '2018-03-01 01:01:00',
|
|
'user_id' => 4,
|
|
]],
|
|
]);
|
|
}
|
|
|
|
protected function setUp()
|
|
{
|
|
parent::setUp();
|
|
|
|
new Config([
|
|
'b8' => [
|
|
'database' => [
|
|
'servers' => [
|
|
'read' => [
|
|
['host' => 'localhost'],
|
|
],
|
|
'write' => [
|
|
['host' => 'localhost'],
|
|
],
|
|
],
|
|
'type' => Database::POSTGRESQL_TYPE,
|
|
'name' => POSTGRESQL_DBNAME,
|
|
'username' => POSTGRESQL_USER,
|
|
'password' => POSTGRESQL_PASSWORD,
|
|
],
|
|
],
|
|
]);
|
|
Database::reset();
|
|
}
|
|
|
|
/**
|
|
* @expectedException \RuntimeException
|
|
*/
|
|
public function testConstruct()
|
|
{
|
|
$store = new WrongStore();
|
|
}
|
|
|
|
public function testGetWhere()
|
|
{
|
|
$testStore = Factory::getStore('ProjectGroup');
|
|
|
|
$data = $testStore->getWhere([], 3, 1, ['id' => 'DESC']);
|
|
self::assertEquals(7, $data['count']);
|
|
self::assertEquals(3, count($data['items']));
|
|
|
|
self::assertEquals(6, $data['items'][0]->getId());
|
|
self::assertEquals(5, $data['items'][1]->getId());
|
|
self::assertEquals(4, $data['items'][2]->getId());
|
|
|
|
$data = $testStore->getWhere(['project_group.user_id' => 0], 100, 0, ['id' => 'ASC']);
|
|
self::assertEquals(2, $data['count']);
|
|
self::assertEquals(2, count($data['items']));
|
|
|
|
self::assertEquals(1, $data['items'][0]->getId());
|
|
self::assertEquals(2, $data['items'][1]->getId());
|
|
|
|
try {
|
|
$data = $testStore->getWhere(['' => 0], 100, 0, ['id' => 'ASC']);
|
|
} catch (\InvalidArgumentException $e) {
|
|
self::assertEquals('You cannot have an empty field name.', $e->getMessage());
|
|
}
|
|
|
|
try {
|
|
$data = $testStore->getWhere([], '; SELECT', 0, ['id' => 'ASC']);
|
|
} catch (\PDOException $e) {
|
|
self::assertInstanceOf('\PDOException', $e);
|
|
}
|
|
}
|
|
|
|
public function testSaveByInsert()
|
|
{
|
|
$this->connection->getConnection()->query('
|
|
ALTER SEQUENCE "project_group_id_seq" RESTART WITH 8;
|
|
');
|
|
|
|
$testStore = Factory::getStore('ProjectGroup');
|
|
$model = new ProjectGroup();
|
|
|
|
$model->setTitle('group 8');
|
|
|
|
$testStore->save($model);
|
|
|
|
$newModel = $testStore->getByPrimaryKey(8);
|
|
|
|
self::assertEquals(8, $newModel->getId());
|
|
self::assertEquals('group 8', $newModel->getTitle());
|
|
}
|
|
|
|
public function testSaveByUpdate()
|
|
{
|
|
$this->connection->getConnection()->query('
|
|
ALTER SEQUENCE "project_group_id_seq" RESTART WITH 8;
|
|
');
|
|
|
|
$testStore = Factory::getStore('ProjectGroup');
|
|
$model = $testStore->getByPrimaryKey(7);
|
|
|
|
$model->setTitle('group 100');
|
|
|
|
$testStore->save($model);
|
|
|
|
$newModel = $testStore->getByPrimaryKey(7);
|
|
|
|
self::assertEquals(7, $newModel->getId());
|
|
self::assertEquals('group 100', $newModel->getTitle());
|
|
|
|
// Without changes
|
|
$model = $testStore->getByPrimaryKey(6);
|
|
|
|
$testStore->save($model);
|
|
|
|
$newModel = $testStore->getByPrimaryKey(6);
|
|
|
|
self::assertEquals(6, $newModel->getId());
|
|
self::assertEquals('group 6', $newModel->getTitle());
|
|
|
|
// Wrong Model
|
|
try {
|
|
$model = new Project();
|
|
$model->setId(1);
|
|
|
|
$testStore->save($model);
|
|
} catch (\InvalidArgumentException $e) {
|
|
self::assertEquals(
|
|
'PHPCensor\Model\Project is an invalid model type for this store.',
|
|
$e->getMessage()
|
|
);
|
|
}
|
|
}
|
|
|
|
public function testDelete()
|
|
{
|
|
$testStore = Factory::getStore('ProjectGroup');
|
|
$model = $testStore->getByPrimaryKey(5);
|
|
|
|
$testStore->delete($model);
|
|
|
|
$newModel = $testStore->getByPrimaryKey(5);
|
|
|
|
self::assertEquals(null, $newModel);
|
|
|
|
// Wrong Model
|
|
try {
|
|
$model = new Project();
|
|
$model->setId(1);
|
|
|
|
$testStore->delete($model);
|
|
} catch (\InvalidArgumentException $e) {
|
|
self::assertEquals(
|
|
'PHPCensor\Model\Project is an invalid model type for this store.',
|
|
$e->getMessage()
|
|
);
|
|
}
|
|
}
|
|
}
|