[ignite] 01/02: Add tests

[am...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

amashenkov pushed a commit to branch ignite-17870
in repository https://gitbox.apache.org/repos/asf/ignite.git

commit a38999a03bcf7ac8cc2d0afe82c34cf2d4714ae7
Author: amashenkov <an...@gmail.com>
AuthorDate: Tue Oct 11 14:34:34 2022 +0300

    Add tests
---
 .../sql/SqlParserMultiStatementSelfTest.java       | 35 ++++++++++++++++++++++
 .../query/MultipleStatementsSqlQuerySelfTest.java  |  8 +++--
 2 files changed, 41 insertions(+), 2 deletions(-)

diff --git a/modules/core/src/test/java/org/apache/ignite/internal/sql/SqlParserMultiStatementSelfTest.java b/modules/core/src/test/java/org/apache/ignite/internal/sql/SqlParserMultiStatementSelfTest.java
index b511cdc188d..4207990ae13 100644
--- a/modules/core/src/test/java/org/apache/ignite/internal/sql/SqlParserMultiStatementSelfTest.java
+++ b/modules/core/src/test/java/org/apache/ignite/internal/sql/SqlParserMultiStatementSelfTest.java
@@ -63,6 +63,41 @@ public class SqlParserMultiStatementSelfTest extends SqlParserAbstractSelfTest {
      */
     @Test
     public void testComments() {
+        String sql = " -- Creating new index \n" +
+            " CREATE INDEX IDX1 on TABLE1(id); \n" +
+            " -- Creating one more index \n" +
+            " CREATE INDEX IDX2 on TABLE2(id); \n" +
+            " -- All done.";
+
+        SqlParser parser = new SqlParser("schema", sql);
+
+        // Haven't parse anything yet.
+        assertEquals(null, parser.lastCommandSql());
+        assertEquals(sql, parser.remainingSql());
+
+        SqlCommand cmd = parser.nextCommand();
+
+        assertTrue(cmd instanceof SqlCreateIndexCommand);
+        assertEquals("CREATE INDEX IDX1 on TABLE1(id)", parser.lastCommandSql());
+        assertEquals(" \n -- Creating one more index \n" +
+            " CREATE INDEX IDX2 on TABLE2(id); \n" +
+            " -- All done.", parser.remainingSql());
+
+        cmd = parser.nextCommand();
+
+        assertTrue(cmd instanceof SqlCreateIndexCommand);
+        assertEquals("CREATE INDEX IDX2 on TABLE2(id)", parser.lastCommandSql());
+        assertEquals(" \n -- All done.", parser.remainingSql());
+
+        cmd = parser.nextCommand();
+
+        assertNull(cmd);
+        assertEquals(null, parser.lastCommandSql());
+        assertEquals(null, parser.remainingSql());
+    }
+
+    @Test
+    public void testEmptyTransaction() {
         String beginSql = "BEGIN  ;" + "  \n";
         String noteSql = "  -- Let's start an empty transaction; $1M idea!\n";
         String commitSql = "COMMIT;" + ";;";
diff --git a/modules/indexing/src/test/java/org/apache/ignite/internal/processors/query/MultipleStatementsSqlQuerySelfTest.java b/modules/indexing/src/test/java/org/apache/ignite/internal/processors/query/MultipleStatementsSqlQuerySelfTest.java
index b3006120398..90381aa01a7 100644
--- a/modules/indexing/src/test/java/org/apache/ignite/internal/processors/query/MultipleStatementsSqlQuerySelfTest.java
+++ b/modules/indexing/src/test/java/org/apache/ignite/internal/processors/query/MultipleStatementsSqlQuerySelfTest.java
@@ -54,11 +54,15 @@ public class MultipleStatementsSqlQuerySelfTest extends AbstractIndexingCommonTe
         GridQueryProcessor qryProc = node.context().query();
 
         SqlFieldsQuery qry = new SqlFieldsQuery(
+            "-- Creating table \n" +
             "create table test(ID int primary key, NAME varchar(20)); " +
+            " -- Adding data \n" +
                 "insert into test (ID, NAME) values (1, 'name_1');" +
                 "insert into test (ID, NAME) values (2, 'name_2'), (3, 'name_3');" +
-                "select * from test;")
-            .setSchema("PUBLIC");
+            " -- Query data \n" +
+            "select * from test;" +
+            " -- All done."
+        ).setSchema("PUBLIC");
 
         List<FieldsQueryCursor<List<?>>> res = qryProc.querySqlFields(qry, true, false);