(shardingsphere) branch master updated: Refactor GenericSQLRewriteEngineTest (#31847)

[zhangliang]

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

zhangliang 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 d2ae62b6fb9 Refactor GenericSQLRewriteEngineTest (#31847)
d2ae62b6fb9 is described below

commit d2ae62b6fb93671b1d4a1b85805ea6ab5c014726
Author: Liang Zhang <zhangli...@apache.org>
AuthorDate: Tue Jun 25 21:54:46 2024 +0800

    Refactor GenericSQLRewriteEngineTest (#31847)
---
 .../infra/rewrite/engine/GenericSQLRewriteEngineTest.java            | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git 
a/infra/rewrite/src/test/java/org/apache/shardingsphere/infra/rewrite/engine/GenericSQLRewriteEngineTest.java
 
b/infra/rewrite/src/test/java/org/apache/shardingsphere/infra/rewrite/engine/GenericSQLRewriteEngineTest.java
index 046f3a90f04..4c3f2f437d0 100644
--- 
a/infra/rewrite/src/test/java/org/apache/shardingsphere/infra/rewrite/engine/GenericSQLRewriteEngineTest.java
+++ 
b/infra/rewrite/src/test/java/org/apache/shardingsphere/infra/rewrite/engine/GenericSQLRewriteEngineTest.java
@@ -73,11 +73,10 @@ class GenericSQLRewriteEngineTest {
         CommonSQLStatementContext sqlStatementContext = 
mock(CommonSQLStatementContext.class);
         DatabaseType databaseType = mock(DatabaseType.class);
         when(sqlStatementContext.getDatabaseType()).thenReturn(databaseType);
-        QueryContext queryContext = mock(QueryContext.class);
+        QueryContext queryContext = mock(QueryContext.class, 
RETURNS_DEEP_STUBS);
         
when(queryContext.getSqlStatementContext()).thenReturn(sqlStatementContext);
         GenericSQLRewriteResult actual = new GenericSQLRewriteEngine(rule, 
database, mock(RuleMetaData.class))
-                .rewrite(new SQLRewriteContext(database, sqlStatementContext, 
"SELECT 1", Collections.emptyList(), mock(ConnectionContext.class),
-                        new HintValueContext()), queryContext);
+                .rewrite(new SQLRewriteContext(database, sqlStatementContext, 
"SELECT 1", Collections.emptyList(), mock(ConnectionContext.class), new 
HintValueContext()), queryContext);
         assertThat(actual.getSqlRewriteUnit().getSql(), is("SELECT 1"));
         assertThat(actual.getSqlRewriteUnit().getParameters(), 
is(Collections.emptyList()));
     }