<?php namespace yiiunit\framework\db\pgsql; use yii\behaviors\TimestampBehavior; use yii\db\pgsql\Schema; use yiiunit\data\ar\ActiveRecord; use yiiunit\framework\ar\ActiveRecordTestTrait; use yiiunit\framework\db\ActiveRecordTest; use yiiunit\TestCase; /** * @group db * @group pgsql */ class PostgreSQLActiveRecordTest extends ActiveRecordTest { protected $driverName = 'pgsql'; public function testBooleanAttribute() { /* @var $customerClass \yii\db\ActiveRecordInterface */ $customerClass = $this->getCustomerClass(); /* @var $this TestCase|ActiveRecordTestTrait */ $customer = new $customerClass(); $customer->name = 'boolean customer'; $customer->email = 'mail@example.com'; $customer->bool_status = false; $customer->save(false); $customer->refresh(); $this->assertSame(false, $customer->bool_status); $customer->bool_status = true; $customer->save(false); $customer->refresh(); $this->assertSame(true, $customer->bool_status); $customers = $customerClass::find()->where(['bool_status' => true])->all(); $this->assertEquals(3, count($customers)); $customers = $customerClass::find()->where(['bool_status' => false])->all(); $this->assertEquals(1, count($customers)); } public function testFindAsArray() { /* @var $customerClass \yii\db\ActiveRecordInterface */ $customerClass = $this->getCustomerClass(); // asArray $customer = $customerClass::find()->where(['id' => 2])->asArray()->one(); $this->assertEquals([ 'id' => 2, 'email' => 'user2@example.com', 'name' => 'user2', 'address' => 'address2', 'status' => 1, 'profile_id' => null, 'bool_status' => true, ], $customer); // find all asArray $customers = $customerClass::find()->asArray()->all(); $this->assertEquals(3, count($customers)); $this->assertArrayHasKey('id', $customers[0]); $this->assertArrayHasKey('name', $customers[0]); $this->assertArrayHasKey('email', $customers[0]); $this->assertArrayHasKey('address', $customers[0]); $this->assertArrayHasKey('status', $customers[0]); $this->assertArrayHasKey('bool_status', $customers[0]); $this->assertArrayHasKey('id', $customers[1]); $this->assertArrayHasKey('name', $customers[1]); $this->assertArrayHasKey('email', $customers[1]); $this->assertArrayHasKey('address', $customers[1]); $this->assertArrayHasKey('status', $customers[1]); $this->assertArrayHasKey('bool_status', $customers[1]); $this->assertArrayHasKey('id', $customers[2]); $this->assertArrayHasKey('name', $customers[2]); $this->assertArrayHasKey('email', $customers[2]); $this->assertArrayHasKey('address', $customers[2]); $this->assertArrayHasKey('status', $customers[2]); $this->assertArrayHasKey('bool_status', $customers[2]); } public function testBooleanValues() { $db = $this->getConnection(); $command = $db->createCommand(); $command->batchInsert('bool_values', ['bool_col'], [ [true], [false], ] )->execute(); $this->assertEquals(1, BoolAR::find()->where('bool_col = TRUE')->count('*', $db)); $this->assertEquals(1, BoolAR::find()->where('bool_col = FALSE')->count('*', $db)); $this->assertEquals(2, BoolAR::find()->where('bool_col IN (TRUE, FALSE)')->count('*', $db)); $this->assertEquals(1, BoolAR::find()->where(['bool_col' => true])->count('*', $db)); $this->assertEquals(1, BoolAR::find()->where(['bool_col' => false])->count('*', $db)); $this->assertEquals(2, BoolAR::find()->where(['bool_col' => [true, false]])->count('*', $db)); $this->assertEquals(1, BoolAR::find()->where('bool_col = :bool_col', ['bool_col' => true])->count('*', $db)); $this->assertEquals(1, BoolAR::find()->where('bool_col = :bool_col', ['bool_col' => false])->count('*', $db)); $this->assertSame(true, BoolAR::find()->where(['bool_col' => true])->one($db)->bool_col); $this->assertSame(false, BoolAR::find()->where(['bool_col' => false])->one($db)->bool_col); } /** * https://github.com/yiisoft/yii2/issues/4672 */ public function testBooleanValues2() { $db = $this->getConnection(); $db->charset = 'utf8'; $db->createCommand("DROP TABLE IF EXISTS bool_user;")->execute(); $db->createCommand()->createTable('bool_user', [ 'id' => Schema::TYPE_PK, 'username' => Schema::TYPE_STRING . ' NOT NULL', 'auth_key' => Schema::TYPE_STRING . '(32) NOT NULL', 'password_hash' => Schema::TYPE_STRING . ' NOT NULL', 'password_reset_token' => Schema::TYPE_STRING, 'email' => Schema::TYPE_STRING . ' NOT NULL', 'role' => Schema::TYPE_SMALLINT . ' NOT NULL DEFAULT 10', 'status' => Schema::TYPE_SMALLINT . ' NOT NULL DEFAULT 10', 'created_at' => Schema::TYPE_INTEGER . ' NOT NULL', 'updated_at' => Schema::TYPE_INTEGER . ' NOT NULL', ])->execute(); $db->createCommand()->addColumn('bool_user', 'is_deleted', Schema::TYPE_BOOLEAN . ' NOT NULL DEFAULT FALSE')->execute(); $user = new UserAR(); $user->username = 'test'; $user->auth_key = 'test'; $user->password_hash = 'test'; $user->email = 'test@example.com'; $user->save(false); $this->assertEquals(1, count(UserAR::find()->where(['is_deleted' => false])->all($db))); $this->assertEquals(0, count(UserAR::find()->where(['is_deleted' => true])->all($db))); $this->assertEquals(1, count(UserAR::find()->where(['is_deleted' => [true, false]])->all($db))); } public function testBooleanDefaultValues() { $model = new BoolAR(); $this->assertNull($model->bool_col); $this->assertNull($model->default_true); $this->assertNull($model->default_false); $model->loadDefaultValues(); $this->assertNull($model->bool_col); $this->assertSame(true, $model->default_true); $this->assertSame(false, $model->default_false); $this->assertTrue($model->save(false)); } } class BoolAR extends ActiveRecord { public static function tableName() { return 'bool_values'; } } class UserAR extends ActiveRecord { const STATUS_DELETED = 0; const STATUS_ACTIVE = 10; const ROLE_USER = 10; public static function tableName() { return '{{%bool_user}}'; } public function behaviors() { return [ TimestampBehavior::className(), ]; } }