[refactor] add Db class

- to avoid side effect in test, create Db class and seperate test
  case

src/YiiFactoryGirl/Db.php

@@ -0,0 +1,158 @@
+<?php
+/**
+ * YiiFactoryGirl\Db
+ *
+ * @author Seiji Amashige <tenjuu99@gmail.com>
+ * @package YiiFactoryGirl
+ */
+
+namespace YiiFactoryGirl;
+
+/**
+ * Db
+ *
+ * @property $connectionID
+ * @property $schemas
+ * @property $connection
+ * @property $lastInsertID
+ */
+class Db
+{
+    /**
+     * connectionID
+     *
+     * @var string
+     */
+    private $connectionID;
+
+    /**
+     * schemas
+     *
+     * @var array
+     */
+    private $schemas;
+
+    /**
+     * db
+     *
+     * @var \CDbConnection
+     */
+    private $connection;
+
+    /**
+     * lastInsertID
+     *
+     * @var array
+     */
+    private $lastInsertID;
+
+    /**
+     * __construct
+     *
+     * @param string $connectionID
+     * @param array $schemas
+     * @return void
+     */
+    public function __construct($connectionID, $schemas = array())
+    {
+        $this->connectionID = $connectionID;
+        $this->schemas = $schemas;
+    }
+
+    /**
+     * Returns the database connection used to load factories.
+     * @throws \CException if {@link connectionID} application component is invalid
+     * @return \CDbConnection the database connection
+     */
+    public function getConnection()
+    {
+        if (!$this->connection) {
+            $connection = \Yii::app()->getComponent($this->connectionID);
+            if (!$connection instanceof \CDbConnection) {
+                throw new \CException(\Yii::t(Factory::LOG_CATEGORY, '\YiiFactoryGirl\Db.connectionID "{id}" is invalid. Please make sure it refers to the ID of a CDbConnection application component.',
+                    array('{id}' => $this->connectionID)));
+            }
+            $this->connection = $connection;
+        }
+        return $this->connection;
+    }
+
+    /**
+     * Enables or disables database integrity check.
+     * This method may be used to temporarily turn off foreign constraints check.
+     * @param boolean $check whether to enable database integrity check
+     */
+    public function checkIntegrity($check)
+    {
+        foreach ($this->schemas as $schema) {
+            $this->getConnection()->getSchema()->checkIntegrity($check, $schema);
+        }
+    }
+
+    /**
+     * Removes all rows from the specified table and resets its primary key sequence, if any.
+     * You may need to call {@link checkIntegrity} to turn off integrity check temporarily
+     * before you call this method.
+     * @param string $tableName the table name
+     * @throws \CException if given table does not exist
+     */
+    public function truncateTable($tableName)
+    {
+        $schema = $this->getConnection()->getSchema();
+        if (($table = $schema->getTable($tableName)) !== null) {
+            $this->getConnection()->createCommand()->truncateTable($table->name);
+            if (isset($this->lastInsertID[$tableName])) {
+                $this->lastInsertID[$tableName] = null;
+            }
+        } else {
+            throw new \CException("Table '$tableName' does not exist.");
+        }
+    }
+
+    /**
+     * Truncates all tables in the specified schema.
+     * You may need to call {@link checkIntegrity} to turn off integrity check temporarily
+     * before you call this method.
+     * @param string $schema the schema name. Defaults to empty string, meaning the default database schema.
+     * @see truncateTable
+     */
+    public function truncateTables($schema = '')
+    {
+        $tableNames = $this->getConnection()->getSchema()->getTableNames($schema);
+        foreach ($tableNames as $tableName) {
+            $this->truncateTable($tableName);
+        }
+    }
+
+    /**
+     * insert
+     *
+     * @param mixed $table
+     * @param mixed $attributes
+     * @return void
+     */
+    public function insert($table, $attributes)
+    {
+        $schema = $this->getConnection()->getSchema();
+        $builder = $schema->getCommandBuilder();
+        $table = $schema->getTable($table);
+        $schema->checkIntegrity(false);
+        $result = $builder->createInsertCommand($table, $attributes)->execute();
+        if ($lastInsertID = $builder->getLastInsertID($table->name)) {
+            $this->lastInsertID[$table->name] = $lastInsertID;
+        }
+        $schema->checkIntegrity(true);
+        return $result;
+    }
+
+    /**
+     * getLastInsertID
+     *
+     * @param mixed $table
+     * @return void
+     */
+    public function getLastInsertID($table)
+    {
+        return isset($this->lastInsertID[$table]) ? $this->lastInsertID[$table] : null;
+    }
+}

tests/YiiFactoryGirl/DbTest.php

@@ -0,0 +1,47 @@
+<?php
+
+use YiiFactoryGirl\Db;
+
+/**
+ * DbTest
+ *
+ * @coversDefaultClass YiiFactoryGirl\Db
+ */
+class DbTest extends YiiFactoryGirl\UnitTestCase
+{
+    /**
+     * testGetDbConnectionSuccess
+     *
+     * @return array
+     */
+    public function testGetDbConnectionSuccess()
+    {
+        return array(
+            array(
+                'assert' => 'InstanceOf',
+                'result' => function() {
+                    return (new Db('db'))->getConnection();
+                },
+                'expect' => 'CDbConnection'
+            )
+        );
+    }
+
+    /**
+     * testGetDbConnectionFail
+     *
+     * @return array
+     */
+    public function testGetDbConnectionFail()
+    {
+        return array(
+            array(
+                'exception' => array('CException', '\YiiFactoryGirl\Db.connectionID "migrate" is invalid'),
+                'callback'  => function() {
+                    $db = new Db('migrate');
+                    $db->getConnection();
+                }
+            )
+        );
+    }
+}