Merge pull request #75 from devenbansod/fix_51_74

Add correct parsing of SET CHARACTER SET, CHARSET , NAMES statements

1 files changed, 47 insertions, 0 deletions

diff --git a/tests/Builder/SetStatementTest.php b/tests/Builder/SetStatementTest.php
new file mode 100644
index 0000000..2f75c0f
--- /dev/null
+++ b/tests/Builder/SetStatementTest.php
@@ -0,0 +1,47 @@
+<?php
+
+namespace SqlParser\Tests\Builder;
+
+use SqlParser\Parser;
+
+use SqlParser\Tests\TestCase;
+
+class SetStatementTest extends TestCase
+{
+
+    public function testBuilderView()
+    {
+        /* Assertion 1 */
+        $query = 'SET CHARACTER SET \'utf8\';';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'SET CHARACTER SET \'utf8\' ',
+            $stmt->build()
+        );
+
+        /* Assertion 2 */
+        $query = 'SET CHARSET \'utf8\';';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'SET CHARSET \'utf8\' ',
+            $stmt->build()
+        );
+
+        /* Assertion 3 */
+        $query = 'SET NAMES \'utf8\';';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'SET NAMES \'utf8\' ',
+            $stmt->build()
+        );
+    }
+}