Keep SET SCHEMA query on one line

More info about SET SCHEMA http://docs.oracle.com/javadb/10.8.3.0/ref/rrefsqlj32268.html

Author: Uku Pattak

src/languages/N1qlFormatter.js

@@ -19,9 +19,10 @@ const reservedWords = [
 ];
 
 const reservedToplevelWords = [
-    "SELECT", "FROM", "WHERE", "SET", "ORDER BY", "GROUP BY", "LIMIT", "VALUES", "EXPLAIN UPDATE", "UPDATE",
-    "HAVING", "EXPLAIN DELETE FROM", "DELETE FROM", "UNION ALL", "UNION", "EXCEPT ALL", "EXCEPT", "INTERSECT ALL", "INTERSECT",
-    "INFER", "EXPLAIN UPSERT", "UPSERT", "MERGE", "PREPARE", "LET", "USE KEYS", "UNNEST", "NEST", "INSERT INTO"
+    "SET SCHEMA", "SET CURRENT SCHEMA", "SELECT", "FROM", "WHERE", "SET", "ORDER BY", "GROUP BY", "LIMIT", "VALUES",
+    "EXPLAIN UPDATE", "UPDATE", "HAVING", "EXPLAIN DELETE FROM", "DELETE FROM", "UNION ALL", "UNION", "EXCEPT ALL",
+    "EXCEPT", "INTERSECT ALL", "INTERSECT", "INFER", "EXPLAIN UPSERT", "UPSERT", "MERGE", "PREPARE", "LET", "USE KEYS",
+    "UNNEST", "NEST", "INSERT INTO"
 ];
 
 const reservedNewlineWords = [

src/languages/StandardSqlFormatter.js

@@ -39,8 +39,9 @@ const reservedWords = [
 ];
 
 const reservedToplevelWords = [
-    "SELECT", "FROM", "WHERE", "SET", "ORDER BY", "GROUP BY", "LIMIT", "ALTER TABLE", "ALTER COLUMN", "INSERT INTO",
-    "VALUES", "UPDATE", "HAVING", "ADD", "AFTER", "DELETE FROM", "UNION ALL", "UNION", "EXCEPT", "INTERSECT", "MODIFY"
+    "SET SCHEMA", "SET CURRENT SCHEMA", "SELECT", "FROM", "WHERE", "SET", "ORDER BY", "GROUP BY", "LIMIT",
+    "ALTER TABLE", "ALTER COLUMN", "INSERT INTO", "VALUES", "UPDATE", "HAVING", "ADD", "AFTER", "DELETE FROM",
+    "UNION ALL", "UNION", "EXCEPT", "INTERSECT", "MODIFY"
 ];
 
 const reservedNewlineWords = [

test/behavesLikeSqlFormatter.js

@@ -24,6 +24,16 @@ export default function behavesLikeSqlFormatter(language) {
         return sqlFormatter.format(query, {language});
     }
 
+    it("formats simple SET SCHEMA queries", function() {
+        const result = format("SET SCHEMA tetrisdb; SET CURRENT SCHEMA bingodb;");
+        expect(result).toBe(
+            "SET SCHEMA\n" +
+            "  tetrisdb;\n" +
+            "SET CURRENT SCHEMA\n" +
+            "  bingodb;\n"
+        );
+    });
+
     it("formats simple SELECT query", function() {
         const result = format("SELECT count(*),Column1 FROM Table1;");
         expect(result).toBe(