Add Wildfire::setQuery

rougin · Mar 24, 2016 · a81c04a · a81c04a
1 parent a8fd1bd
commit a81c04a
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 3 deletions.
diff --git a/src/Wildfire.php b/src/Wildfire.php
@@ -27,7 +27,7 @@ class Wildfire
     protected $describe;
 
     /**
-     * @var CI_DB
+     * @var CI_DB_result
      */
     protected $query;
 
@@ -42,8 +42,8 @@ class Wildfire
     protected $tables = [];
 
     /**
-     * @param CI_DB|null $database
-     * @param CI_DB|null $query
+     * @param CI_DB|null        $database
+     * @param CI_DB_result|null $query
      */
     public function __construct($database = null, $query = null)
     {
@@ -186,6 +186,19 @@ public function setDatabase($database)
         return $this;
     }
 
+    /**
+     * Sets the query result.
+     * 
+     * @param  CI_DB_result $query
+     * @return self
+     */
+    public function setQuery($query)
+    {
+        $this->query = $query;
+
+        return $this;
+    }
+
     /**
      * Create an object from the specified data
      *

diff --git a/tests/WildfireTest.php b/tests/WildfireTest.php
@@ -133,4 +133,21 @@ public function testWildfireWithoutConstructor()
 
         $this->assertCount(5, $posts);
     }
+
+    /**
+     * Tests Wildfire::set_query.
+     * 
+     * @return void
+     */
+    public function testSetQueryMethod()
+    {
+        $query = $this->ci->db->query('SELECT * FROM ' . $this->table);
+
+        $wildfire = new Wildfire($this->ci->db);
+        $wildfire->set_query($query);
+
+        $posts = $wildfire->result();
+
+        $this->assertCount($this->expectedRows, $posts);
+    }
 }