添加单元测试

hhyo · Sep 19, 2024 · 5db6b33 · 5db6b33
1 parent a1a3cd1
commit 5db6b33
Show file tree

Hide file tree

Showing 6 changed files with 458 additions and 342 deletions.
diff --git a/sql/engines/__init__.py b/sql/engines/__init__.py
@@ -1,6 +1,7 @@
 """engine base库, 包含一个``EngineBase`` class和一个get_engine函数"""
 
 import importlib
+import re
 from sql.engines.models import ResultSet, ReviewSet
 from sql.models import Instance
 from sql.utils.ssh_tunnel import SSHConnection

diff --git a/sql/engines/mysql.py b/sql/engines/mysql.py
@@ -13,7 +13,6 @@
 
 from sql.engines.goinception import GoInceptionEngine
 from sql.utils.sql_utils import (
-    filter_db_list,
     get_syntax_type,
     remove_comments,
 )

diff --git a/sql/utils/sql_utils.py b/sql/utils/sql_utils.py
@@ -414,3 +414,5 @@ def filter_db_list(db_list, db_name_regex: str, is_match_regex: bool, key="value
         if (is_match_regex and is_match) or (not is_match_regex and not is_match):
             filtered_list.append(db)
     return filtered_list
+
+