Update forced query type tests

Author: alex268

jdbc/src/test/java/tech/ydb/jdbc/query/QueryLexerTest.java

@@ -108,4 +108,68 @@ public void mixQueryExceptionTest() throws SQLException {
                 "SELECT * FROM test_table;\n\tCREATE TABLE test_table (id int, value text);"
         );
     }
+
+    @Test
+    public void forsedTypeTest() throws SQLException {
+        YdbQueryOptions opts = new YdbQueryOptions(false, true, false, false, false, false, QueryType.SCHEME_QUERY);
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "CREATE TABLE test_table (id int, value text)"
+        ));
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "\tcreate TABLE test_table2 (id int, value text);"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                " drop TABLE test_table1 (id int, value text);" +
+                "ALTER TABLE test_table2 (id int, value text);"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SELECT id, value FROM test_table"
+        ));
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "UPSERT INTO test_table VALUES (?, ?)"
+        ));
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "DELETE FROM test_table"
+        ));
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SELECT id, value FROM test_table;\n" +
+                "UPSERT INTO test_table VALUES (?, ?);" +
+                "DELETE FROM test_table"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SELECT id, value FROM test_table;\n" +
+                "UPDATE test_table SET value = ? WHERE id = ?;" +
+                "SELECT id, value FROM test_table WHERE id=CREATE"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SCAN SELECT id, value FROM test_table"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "EXPLAIN SELECT id, value FROM test_table"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "CREATE TABLE test_table (id int, value text);" +
+                "SELECT * FROM test_table;"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "DROP TABLE test_table (id int, value text);SELECT * FROM test_table;"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SELECT * FROM test_table;CREATE TABLE test_table (id int, value text);"
+        ));
+
+        Assertions.assertEquals(QueryType.SCHEME_QUERY, parseQueryType(opts,
+                "SELECT * FROM test_table;\n\tCREATE TABLE test_table (id int, value text);"
+        ));
+    }
+
 }