Qiang Xue
12 years ago
2 changed files with 116 additions and 1 deletions
@ -0,0 +1,115 @@ |
|||||||
|
<?php |
||||||
|
|
||||||
|
namespace yiiunit\framework\behaviors; |
||||||
|
|
||||||
|
use Yii; |
||||||
|
use yiiunit\TestCase; |
||||||
|
use yii\db\Connection; |
||||||
|
use yii\db\ActiveRecord; |
||||||
|
use yii\behaviors\AutoTimestamp; |
||||||
|
|
||||||
|
/** |
||||||
|
* Unit test for [[\yii\behaviors\AutoTimestamp]]. |
||||||
|
* @see AutoTimestamp |
||||||
|
*/ |
||||||
|
class AutoTimestampTest extends TestCase |
||||||
|
{ |
||||||
|
/** |
||||||
|
* @var Connection test db connection |
||||||
|
*/ |
||||||
|
protected $dbConnection; |
||||||
|
|
||||||
|
public static function setUpBeforeClass() |
||||||
|
{ |
||||||
|
if (!extension_loaded('pdo') || !extension_loaded('pdo_sqlite')) { |
||||||
|
static::markTestSkipped('PDO and SQLite extensions are required.'); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public function setUp() { |
||||||
|
$this->mockApplication( |
||||||
|
array( |
||||||
|
'components' => array( |
||||||
|
'db' => array( |
||||||
|
'class' => '\yii\db\Connection', |
||||||
|
'dsn' => 'sqlite::memory:', |
||||||
|
) |
||||||
|
) |
||||||
|
) |
||||||
|
); |
||||||
|
|
||||||
|
$columns = array( |
||||||
|
'id' => 'pk', |
||||||
|
'create_time' => 'integer', |
||||||
|
'update_time' => 'integer', |
||||||
|
); |
||||||
|
Yii::$app->getDb()->createCommand()->createTable('test_auto_timestamp', $columns)->execute(); |
||||||
|
} |
||||||
|
|
||||||
|
public function tearDown() |
||||||
|
{ |
||||||
|
Yii::$app->getDb()->close(); |
||||||
|
parent::tearDown(); |
||||||
|
} |
||||||
|
|
||||||
|
// Tests : |
||||||
|
|
||||||
|
public function testNewRecord() |
||||||
|
{ |
||||||
|
$currentTime = time(); |
||||||
|
|
||||||
|
$model = new ActiveRecordAutoTimestamp(); |
||||||
|
$model->save(false); |
||||||
|
|
||||||
|
$this->assertTrue($model->create_time >= $currentTime); |
||||||
|
$this->assertTrue($model->update_time >= $currentTime); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @depends testNewRecord |
||||||
|
*/ |
||||||
|
public function testUpdateRecord() |
||||||
|
{ |
||||||
|
$currentTime = time(); |
||||||
|
|
||||||
|
$model = new ActiveRecordAutoTimestamp(); |
||||||
|
$model->save(false); |
||||||
|
|
||||||
|
$enforcedTime = $currentTime - 100; |
||||||
|
|
||||||
|
$model->create_time = $enforcedTime; |
||||||
|
$model->update_time = $enforcedTime; |
||||||
|
$model->save(false); |
||||||
|
|
||||||
|
$this->assertEquals($enforcedTime, $model->create_time, 'Create time has been set on update!'); |
||||||
|
$this->assertTrue($model->update_time >= $currentTime, 'Update time has NOT been set on update!'); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Test Active Record class with [[AutoTimestamp]] behavior attached. |
||||||
|
* |
||||||
|
* @property integer $id |
||||||
|
* @property integer $create_time |
||||||
|
* @property integer $update_time |
||||||
|
*/ |
||||||
|
class ActiveRecordAutoTimestamp extends ActiveRecord |
||||||
|
{ |
||||||
|
public function behaviors() |
||||||
|
{ |
||||||
|
return array( |
||||||
|
'timestamp' => array( |
||||||
|
'class' => AutoTimestamp::className(), |
||||||
|
'attributes' => array( |
||||||
|
static::EVENT_BEFORE_INSERT => array('create_time', 'update_time'), |
||||||
|
static::EVENT_BEFORE_UPDATE => 'update_time', |
||||||
|
), |
||||||
|
), |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
public static function tableName() |
||||||
|
{ |
||||||
|
return 'test_auto_timestamp'; |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue