Add recognizer for SET Statements

Fix #157

Add Tests for SET statement flag

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

2 files changed, 9 insertions, 0 deletions

diff --git a/src/Utils/Query.php b/src/Utils/Query.php
index bbd1d3b..0d53920 100644
--- a/src/Utils/Query.php
+++ b/src/Utils/Query.php
@@ -26,6 +26,7 @@ use PhpMyAdmin\SqlParser\Statements\RenameStatement;
 use PhpMyAdmin\SqlParser\Statements\RepairStatement;
 use PhpMyAdmin\SqlParser\Statements\ReplaceStatement;
 use PhpMyAdmin\SqlParser\Statements\SelectStatement;
+use PhpMyAdmin\SqlParser\Statements\SetStatement;
 use PhpMyAdmin\SqlParser\Statements\ShowStatement;
 use PhpMyAdmin\SqlParser\Statements\TruncateStatement;
 use PhpMyAdmin\SqlParser\Statements\UpdateStatement;
@@ -358,6 +359,8 @@ class Query
         } elseif ($statement instanceof UpdateStatement) {
             $flags['querytype'] = 'UPDATE';
             $flags['is_affected'] = true;
+        } elseif ($statement instanceof SetStatement) {
+            $flags['querytype'] = 'SET';
         }
 
         if (($statement instanceof SelectStatement)
diff --git a/tests/Utils/QueryTest.php b/tests/Utils/QueryTest.php
index 9f54b7b..124b161 100644
--- a/tests/Utils/QueryTest.php
+++ b/tests/Utils/QueryTest.php
@@ -263,6 +263,12 @@ class QueryTest extends TestCase
                     'select_from' => true,
                 ),
             ),
+            array(
+                'SET NAMES \'latin\'',
+                array(
+                    'querytype' => 'SET',
+                ),
+            ),
         );
     }