[shardingsphere] branch master updated: [Issue #20384]-Added more unit tests in CreateIndexStatementHandlerTest (#20792)

[du...@apache.org]

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

duanzhengqiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new f6f4240d4da [Issue #20384]-Added more unit tests in CreateIndexStatementHandlerTest (#20792)
f6f4240d4da is described below

commit f6f4240d4da797019ea63ae93d54251de735d283
Author: Abhinav Koppula <ab...@gmail.com>
AuthorDate: Tue Sep 6 05:31:16 2022 +0530

    [Issue #20384]-Added more unit tests in CreateIndexStatementHandlerTest (#20792)
---
 .../ddl/CreateIndexStatementHandlerTest.java       | 34 ++++++++++++++++++++--
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/shardingsphere-sql-parser/shardingsphere-sql-parser-statement/src/test/java/org/apache/shardingsphere/sql/parser/sql/dialect/handler/ddl/CreateIndexStatementHandlerTest.java b/shardingsphere-sql-parser/shardingsphere-sql-parser-statement/src/test/java/org/apache/shardingsphere/sql/parser/sql/dialect/handler/ddl/CreateIndexStatementHandlerTest.java
index 9236a6dcec1..0f02eb2848a 100644
--- a/shardingsphere-sql-parser/shardingsphere-sql-parser-statement/src/test/java/org/apache/shardingsphere/sql/parser/sql/dialect/handler/ddl/CreateIndexStatementHandlerTest.java
+++ b/shardingsphere-sql-parser/shardingsphere-sql-parser-statement/src/test/java/org/apache/shardingsphere/sql/parser/sql/dialect/handler/ddl/CreateIndexStatementHandlerTest.java
@@ -17,6 +17,12 @@
 
 package org.apache.shardingsphere.sql.parser.sql.dialect.handler.ddl;
 
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+import java.util.Optional;
 import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.ddl.MySQLCreateIndexStatement;
 import org.apache.shardingsphere.sql.parser.sql.dialect.statement.opengauss.ddl.OpenGaussCreateIndexStatement;
 import org.apache.shardingsphere.sql.parser.sql.dialect.statement.oracle.ddl.OracleCreateIndexStatement;
@@ -24,9 +30,6 @@ import org.apache.shardingsphere.sql.parser.sql.dialect.statement.postgresql.ddl
 import org.apache.shardingsphere.sql.parser.sql.dialect.statement.sqlserver.ddl.SQLServerCreateIndexStatement;
 import org.junit.Test;
 
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
-
 public final class CreateIndexStatementHandlerTest {
     
     @Test
@@ -37,4 +40,29 @@ public final class CreateIndexStatementHandlerTest {
         assertFalse(CreateIndexStatementHandler.ifNotExists(new OracleCreateIndexStatement()));
         assertFalse(CreateIndexStatementHandler.ifNotExists(new SQLServerCreateIndexStatement()));
     }
+
+    @Test
+    public void assertGeneratedIndexStartIndexForPostgres() {
+        PostgreSQLCreateIndexStatement postgreSQLCreateIndexStatement = new PostgreSQLCreateIndexStatement(true);
+        postgreSQLCreateIndexStatement.setGeneratedIndexStartIndex(2);
+        Optional<Integer> actual = CreateIndexStatementHandler.getGeneratedIndexStartIndex(postgreSQLCreateIndexStatement);
+        assertTrue(actual.isPresent());
+        assertThat(actual.get(), is(2));
+    }
+
+    @Test
+    public void assertGeneratedIndexStartIndexForOpenGauss() {
+        OpenGaussCreateIndexStatement openGaussCreateIndexStatement = new OpenGaussCreateIndexStatement(true);
+        openGaussCreateIndexStatement.setGeneratedIndexStartIndex(2);
+        Optional<Integer> actual = CreateIndexStatementHandler.getGeneratedIndexStartIndex(openGaussCreateIndexStatement);
+        assertTrue(actual.isPresent());
+        assertThat(actual.get(), is(2));
+    }
+
+    @Test
+    public void assertGeneratedIndexStartIndexForOtherDatabases() {
+        assertFalse(CreateIndexStatementHandler.getGeneratedIndexStartIndex(new MySQLCreateIndexStatement()).isPresent());
+        assertFalse(CreateIndexStatementHandler.getGeneratedIndexStartIndex(new OracleCreateIndexStatement()).isPresent());
+        assertFalse(CreateIndexStatementHandler.getGeneratedIndexStartIndex(new SQLServerCreateIndexStatement()).isPresent());
+    }
 }