add unionSelectAll() method and tests

Author: co0lc0der

src/QueryBuilder.php

@@ -1028,6 +1028,22 @@ public function unionSelect($table, bool $unionAll = false): QueryBuilder
         return $this;
     }
 
+    /**
+     * @param string|array $table
+     * @return QueryBuilder
+     */
+    public function unionSelectAll($table): QueryBuilder
+    {
+        if (empty($table)) {
+            $this->setError('Empty $table in ' . __METHOD__);
+            return $this;
+        }
+
+        $this->unionSelect($table, true);
+
+        return $this;
+    }
+
     /**
      * @return QueryBuilder
      */

tests/SqlSelectExtTest.php

@@ -248,22 +248,22 @@ public function testUnionSelectEmptyTable()
         $this->assertSame($result->getErrorMessage(), 'Empty $table in QueryBuilder::unionSelect');
     }
 
-    public function testUnionSelectIncorrectTable()
+    public function testUnionSelectAllMethodEmptyTable()
     {
-        $result = $this->qb->select('clients', ['name', 'age'])->unionSelect(2);
+        $result = $this->qb->select('clients', ['name', 'age'])->unionSelectAll('');
 
         $this->assertSame($this->qb, $result);
         $this->assertSame(true, $result->hasError());
-        $this->assertSame($result->getErrorMessage(), 'Incorrect type of $table in QueryBuilder::unionSelect. $table must be a string or an array');
+        $this->assertSame($result->getErrorMessage(), 'Empty $table in QueryBuilder::unionSelectAll');
     }
 
-    public function testUnionSelectDoubleUnion()
+    public function testUnionSelectIncorrectTable()
     {
-        $result = $this->qb->select('clients', ['name', 'age'])->union()->unionSelect('clients');
+        $result = $this->qb->select('clients', ['name', 'age'])->unionSelect(2);
 
         $this->assertSame($this->qb, $result);
         $this->assertSame(true, $result->hasError());
-        $this->assertSame($result->getErrorMessage(), 'SQL has already UNION in QueryBuilder::unionSelect');
+        $this->assertSame($result->getErrorMessage(), 'Incorrect type of $table in QueryBuilder::unionSelect. $table must be a string or an array');
     }
 
     public function testUnionSelectWhere()
@@ -288,6 +288,17 @@ public function testUnionAllSelectWhere()
         $this->assertSame([10], $result->getParams());
     }
 
+    public function testUnionSelectAllMethodWhere()
+    {
+        $result = $this->qb->select('cabs', ['id', 'name'])
+                ->unionSelectAll('printer_models')->where([['id', '<', 10]]);
+
+        $this->assertSame($this->qb, $result);
+        $this->assertSame(false, $result->hasError());
+        $this->assertSame("SELECT `id`, `name` FROM `cabs` UNION ALL SELECT `id`, `name` FROM `printer_models` WHERE (`id` < 10)", $result->getSql());
+        $this->assertSame([10], $result->getParams());
+    }
+
     public function testSelectExceptsWhere()
     {
         $result = $this->qb->select('contacts', ['contact_id', 'last_name', 'first_name'])