diff --git a/tests/unit/framework/behaviors/AutoTimestampTest.php b/tests/unit/framework/behaviors/AutoTimestampTest.php index 51bbe2f..de96630 100644 --- a/tests/unit/framework/behaviors/AutoTimestampTest.php +++ b/tests/unit/framework/behaviors/AutoTimestampTest.php @@ -1,5 +1,8 @@ 'integer', 'update_time' => 'integer', ); - Yii::$app->getDb()->createCommand()->createTable(self::$testTableName, $columns)->execute(); - - $this->declareTestActiveRecordClass(); + Yii::$app->getDb()->createCommand()->createTable('test_auto_timestamp', $columns)->execute(); } public function tearDown() @@ -60,54 +52,13 @@ class AutoTimestampTest extends TestCase parent::tearDown(); } - /** - * Declares test Active Record class with auto timestamp behavior attached. - */ - protected function declareTestActiveRecordClass() - { - $className = static::$testActiveRecordClassName; - if (class_exists($className, false)) { - return true; - } - - $activeRecordClassName = ActiveRecord::className(); - $behaviorClassName = AutoTimestamp::className(); - $tableName = static::$testTableName; - - $classDefinitionCode = << array( - 'class' => '{$behaviorClassName}', - 'attributes' => array( - static::EVENT_BEFORE_INSERT => array('create_time', 'update_time'), - static::EVENT_BEFORE_UPDATE => 'update_time', - ), - ), - ); - } - - public static function tableName() - { - return '{$tableName}'; - } -} -EOL; - eval($classDefinitionCode); - return true; - } - // Tests : public function testNewRecord() { $currentTime = time(); - $className = static::$testActiveRecordClassName; - $model = new $className(); + $model = new ActiveRecordAutoTimestamp(); $model->save(false); $this->assertTrue($model->create_time >= $currentTime); @@ -121,8 +72,7 @@ EOL; { $currentTime = time(); - $className = static::$testActiveRecordClassName; - $model = new $className(); + $model = new ActiveRecordAutoTimestamp(); $model->save(false); $enforcedTime = $currentTime - 100; @@ -134,4 +84,32 @@ EOL; $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'; + } } \ No newline at end of file