miemie 5 anos atrás
pai
commit
1dc11d1d72

+ 9 - 2
mybatis-plus-extension/src/test/java/com/baomidou/mybatisplus/extension/plugins/tenant/TenantSqlParserTest.java

@@ -37,10 +37,17 @@ public class TenantSqlParserTest {
 
 
     @Test
     @Test
     public void processSelectBody() throws JSQLParserException {
     public void processSelectBody() throws JSQLParserException {
-        m("select * from user", "select * from user where user.t_id = 1");
+        select("select * from user",
+            "select * from user where user.t_id = 1");
+        select("select * from user where id in (select id from user)",
+            "select * from user where id in (select id from user where user.t_id = 1) and user.t_id = 1");
+        select("select * from user where id = 1 and id in (select id from user)",
+            "select * from user where id = 1 and id in (select id from user where user.t_id = 1) and user.t_id = 1");
+        select("select * from user where id = 1 or id in (select id from user)",
+            "select * from user where (id = 1 or id in (select id from user where user.t_id = 1)) and user.t_id = 1");
     }
     }
 
 
-    private void m(String sql, String target) throws JSQLParserException {
+    private void select(String sql, String target) throws JSQLParserException {
         Statements statement = CCJSqlParserUtil.parseStatements(sql);
         Statements statement = CCJSqlParserUtil.parseStatements(sql);
         Select select = (Select) statement.getStatements().get(0);
         Select select = (Select) statement.getStatements().get(0);
         parser.processSelectBody(select.getSelectBody());
         parser.processSelectBody(select.getSelectBody());