YII 类库手册

public function checkIntegrity($check)
{
    foreach($this->schemas as $schema)
        $this->getDbConnection()->getSchema()->checkIntegrity($check,$schema);
}

public function getDbConnection()
{
    if($this->_db===null)
    {
        $this->_db=Yii::app()->getComponent($this->connectionID);
        if(!$this->_db instanceof CDbConnection)
            throw new CException(Yii::t('yii','CDbTestFixture.connectionID "{id}" is invalid. Please make sure it refers to the ID of a CDbConnection application component.',
                array('{id}'=>$this->connectionID)));
    }
    return $this->_db;
}

public function getFixtures()
{
    if($this->_fixtures===null)
    {
        $this->_fixtures=array();
        $schema=$this->getDbConnection()->getSchema();
        $folder=opendir($this->basePath);
        $suffixLen=strlen($this->initScriptSuffix);
        while($file=readdir($folder))
        {
            if($file==='.' || $file==='..' || $file===$this->initScript)
                continue;
            $path=$this->basePath.DIRECTORY_SEPARATOR.$file;
            if(substr($file,-4)==='.php' && is_file($path) && substr($file,-$suffixLen)!==$this->initScriptSuffix)
            {
                $tableName=substr($file,0,-4);
                if($schema->getTable($tableName)!==null)
                    $this->_fixtures[$tableName]=$path;
            }
        }
        closedir($folder);
    }
    return $this->_fixtures;
}

public function getRecord($name,$alias)
{
    if(isset($this->_records[$name][$alias]))
    {
        if(is_string($this->_records[$name][$alias]))
        {
            $row=$this->_rows[$name][$alias];
            $model=CActiveRecord::model($this->_records[$name][$alias]);
            $key=$model->getTableSchema()->primaryKey;
            if(is_string($key))
                $pk=$row[$key];
            else
            {
                foreach($key as $k)
                    $pk[$k]=$row[$k];
            }
            $this->_records[$name][$alias]=$model->findByPk($pk);
        }
        return $this->_records[$name][$alias];
    }
    else
        return false;
}

public function getRows($name)
{
    if(isset($this->_rows[$name]))
        return $this->_rows[$name];
    else
        return false;
}

public function init()
{
    parent::init();
    if($this->basePath===null)
        $this->basePath=Yii::getPathOfAlias('application.tests.fixtures');
    $this->prepare();
}

public function load($fixtures)
{
    $schema=$this->getDbConnection()->getSchema();
    $schema->checkIntegrity(false);

    $this->_rows=array();
    $this->_records=array();
    foreach($fixtures as $fixtureName=>$tableName)
    {
        if($tableName[0]===':')
        {
            $tableName=substr($tableName,1);
            unset($modelClass);
        }
        else
        {
            $modelClass=Yii::import($tableName,true);
            $tableName=CActiveRecord::model($modelClass)->tableName();
            if(($prefix=$this->getDbConnection()->tablePrefix)!==null)
                $tableName=preg_replace('/{{(.*?)}}/',$prefix.'\1',$tableName);
        }
        $this->resetTable($tableName);
        $rows=$this->loadFixture($tableName);
        if(is_array($rows) && is_string($fixtureName))
        {
            $this->_rows[$fixtureName]=$rows;
            if(isset($modelClass))
            {
                foreach(array_keys($rows) as $alias)
                    $this->_records[$fixtureName][$alias]=$modelClass;
            }
        }
    }

    $schema->checkIntegrity(true);
}

public function loadFixture($tableName)
{
    $fileName=$this->basePath.DIRECTORY_SEPARATOR.$tableName.'.php';
    if(!is_file($fileName))
        return false;

    $rows=array();
    $schema=$this->getDbConnection()->getSchema();
    $builder=$schema->getCommandBuilder();
    $table=$schema->getTable($tableName);

    foreach(require($fileName) as $alias=>$row)
    {
        $builder->createInsertCommand($table,$row)->execute();
        $primaryKey=$table->primaryKey;
        if($table->sequenceName!==null)
        {
            if(is_string($primaryKey) && !isset($row[$primaryKey]))
                $row[$primaryKey]=$builder->getLastInsertID($table);
            else if(is_array($primaryKey))
            {
                foreach($primaryKey as $pk)
                {
                    if(!isset($row[$pk]))
                    {
                        $row[$pk]=$builder->getLastInsertID($table);
                        break;
                    }
                }
            }
        }
        $rows[$alias]=$row;
    }
    return $rows;
}

public function prepare()
{
    $initFile=$this->basePath . DIRECTORY_SEPARATOR . $this->initScript;

    $this->checkIntegrity(false);

    if(is_file($initFile))
        require($initFile);
    else
    {
        foreach($this->getFixtures() as $tableName=>$fixturePath)
        {
            $this->resetTable($tableName);
            $this->loadFixture($tableName);
        }
    }
    $this->checkIntegrity(true);
}

public function resetTable($tableName)
{
    $initFile=$this->basePath . DIRECTORY_SEPARATOR . $tableName . $this->initScriptSuffix;
    if(is_file($initFile))
        require($initFile);
    else
        $this->truncateTable($tableName);
}

public function truncateTable($tableName)
{
    $db=$this->getDbConnection();
    $schema=$db->getSchema();
    if(($table=$schema->getTable($tableName))!==null)
    {
        $db->createCommand('DELETE FROM '.$table->rawName)->execute();
        $schema->resetSequence($table,1);
    }
    else
        throw new CException("Table '$tableName' does not exist.");
}

public function truncateTables($schema='')
{
    $tableNames=$this->getDbConnection()->getSchema()->getTableNames($schema);
    foreach($tableNames as $tableName)
        $this->truncateTable($tableName);
}