Add support for end options in SET statement

* Support `[{ COLLATE expr | DEFAULT }] in `SET NAMES` statement

Signed-off-by: Deven Bansod <devenbansod.bits@gmail.com>

1 files changed, 28 insertions, 6 deletions

diff --git a/tests/Builder/SetStatementTest.php b/tests/Builder/SetStatementTest.php
index f6f669a..746e92f 100644
--- a/tests/Builder/SetStatementTest.php
+++ b/tests/Builder/SetStatementTest.php
@@ -10,35 +10,57 @@ class SetStatementTest extends TestCase
     public function testBuilderView()
     {
         /* Assertion 1 */
-        $query = 'SET CHARACTER SET \'utf8\';';
+        $query = 'SET CHARACTER SET \'utf8\'';
 
         $parser = new Parser($query);
         $stmt = $parser->statements[0];
 
         $this->assertEquals(
-            'SET CHARACTER SET \'utf8\' ',
+            $query,
             $stmt->build()
         );
 
         /* Assertion 2 */
-        $query = 'SET CHARSET \'utf8\';';
+        $query = 'SET CHARSET \'utf8\'';
 
         $parser = new Parser($query);
         $stmt = $parser->statements[0];
 
         $this->assertEquals(
-            'SET CHARSET \'utf8\' ',
+            $query,
             $stmt->build()
         );
 
         /* Assertion 3 */
-        $query = 'SET NAMES \'utf8\';';
+        $query = 'SET NAMES \'utf8\'';
 
         $parser = new Parser($query);
         $stmt = $parser->statements[0];
 
         $this->assertEquals(
-            'SET NAMES \'utf8\' ',
+            $query,
+            $stmt->build()
+        );
+
+        /* Assertion 4 */
+        $query = 'SET NAMES \'utf8\' COLLATE \'utf8_general_ci\'';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'SET NAMES \'utf8\'  COLLATE \'utf8_general_ci\'',
+            $stmt->build()
+        );
+
+        /* Assertion 5 */
+        $query = 'SET NAMES \'utf8\' DEFAULT';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'SET NAMES \'utf8\'  DEFAULT',
             $stmt->build()
         );
     }