Improve coverage ratio.

Author: ralgond

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='mybatis',
-    version='0.0.16',
+    version='0.0.17',
     description='A python ORM like mybatis.',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',  # 如果你使用的是Markdown格式的README

test/test_db_rollback.py

@@ -6,7 +6,7 @@
 
 
 @pytest.fixture(scope="function")
-def db_connection():
+def mysql_db_connection():
     # 配置数据库连接
     connection = ConnectionFactory.get_connection(
         dbms_name="mysql",
@@ -36,8 +36,67 @@ def db_connection():
     # 清理数据和关闭连接
     connection.close()
 
-def test_db_rollback_error_table(db_connection):
-    mb = Mybatis(db_connection, "mapper")
+@pytest.fixture(scope="function")
+def postgresql_db_connection():
+    # 配置数据库连接
+    connection = ConnectionFactory.get_connection(
+        dbms_name="mysql",
+        host="localhost",
+        user="mybatis",
+        password="mybatis",
+        database="mybatis"
+    )
+    connection.start_transaction()
+    cursor = connection.cursor()
+    cursor.execute("DROP TABLE IF EXISTS fruits")
+    create_table_sql = '''CREATE TABLE IF NOT EXISTS fruits (
+        id SERIAL PRIMARY KEY, 
+        name VARCHAR(100),
+        category VARCHAR(100),
+        price int)
+    '''
+    # 在测试开始前准备数据
+    cursor.execute(create_table_sql)
+    cursor.execute("INSERT INTO fruits (name, category, price) VALUES ('Alice', 'A', 100)")
+    cursor.execute("INSERT INTO fruits (name, category, price) VALUES ('Bob', 'B', 200)")
+    connection.commit()
+
+    # 提供数据库连接给测试用例
+    yield connection
+
+    # 清理数据和关闭连接
+    connection.close()
+
+@pytest.fixture(scope="function")
+def sqlite3_db_connection():
+    # 配置数据库连接
+    connection = ConnectionFactory.get_connection(
+        dbms_name="sqlite3",
+        db_path="./test.db"
+    )
+    connection.start_transaction()
+    cursor = connection.cursor()
+    cursor.execute("DROP TABLE IF EXISTS fruits")
+    create_table_sql = '''CREATE TABLE IF NOT EXISTS fruits (
+        id INTEGER PRIMARY KEY AUTOINCREMENT, 
+        name VARCHAR,
+        category VARCHAR,
+        price INTEGER)
+    '''
+    # 在测试开始前准备数据
+    cursor.execute(create_table_sql)
+    cursor.execute("INSERT INTO fruits (name, category, price) VALUES ('Alice', 'A', 100)")
+    cursor.execute("INSERT INTO fruits (name, category, price) VALUES ('Bob', 'B', 200)")
+    connection.commit()
+
+    # 提供数据库连接给测试用例
+    yield connection
+
+    # 清理数据和关闭连接
+    connection.close()
+
+def test_db_rollback_error_table(mysql_db_connection):
+    mb = Mybatis(mysql_db_connection, "mapper")
     @mb.Insert("INSERT INTO fruits2 (name, category, price) VALUES ('Candy', 'B', 500)")
     def insert():
         pass
@@ -65,8 +124,8 @@ def update():
     except mybatis.errors.DatabaseError as err:
         assert True
 
-def test_db_rollback_error_table2(db_connection):
-    mb = Mybatis(db_connection, "mapper")
+def test_db_rollback_error_table2(mysql_db_connection):
+    mb = Mybatis(mysql_db_connection, "mapper")
     try:
         mb.insert("testDBRollbackInsert", {})
     except mybatis.errors.DatabaseError as err:
@@ -77,6 +136,99 @@ def test_db_rollback_error_table2(db_connection):
     except mybatis.errors.DatabaseError as err:
         assert True
 
+    try:
+        mb.update("testDBRollbackUpdate", {})
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+
+def test_postgresql_db_rollback_error_table(postgresql_db_connection):
+    mb = Mybatis(postgresql_db_connection, "mapper")
+    @mb.Insert("INSERT INTO fruits2 (name, category, price) VALUES ('Candy', 'B', 500)", primary_key="id")
+    def insert():
+        pass
+
+    try:
+        insert()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    @mb.Delete("DELETE FROM fruits2 WHERE id=1")
+    def delete():
+        pass
+
+    try:
+        delete()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    @mb.Delete("UPDATE fruits2 SET name='Candy' WHERE id=1")
+    def update():
+        pass
+
+    try:
+        update()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+def test_postgresql_db_rollback_error_table2(postgresql_db_connection):
+    mb = Mybatis(postgresql_db_connection, "mapper")
+    try:
+        mb.insert("testDBRollbackInsert", {}, primary_key="id")
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    try:
+        mb.delete("testDBRollbackDelete", {})
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    try:
+        mb.update("testDBRollbackUpdate", {})
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+def test_sqlite3_db_rollback_error_table(sqlite3_db_connection):
+    mb = Mybatis(sqlite3_db_connection, "mapper")
+    @mb.Insert("INSERT INTO fruits2 (name, category, price) VALUES ('Candy', 'B', 500)", primary_key="id")
+    def insert():
+        pass
+
+    try:
+        insert()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    @mb.Delete("DELETE FROM fruits2 WHERE id=1")
+    def delete():
+        pass
+
+    try:
+        delete()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    @mb.Delete("UPDATE fruits2 SET name='Candy' WHERE id=1")
+    def update():
+        pass
+
+    try:
+        update()
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+def test_sqlite3_db_rollback_error_table2(sqlite3_db_connection):
+    mb = Mybatis(sqlite3_db_connection, "mapper")
+    try:
+        mb.insert("testDBRollbackInsert", {}, primary_key="id")
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
+    try:
+        mb.delete("testDBRollbackDelete", {})
+    except mybatis.errors.DatabaseError as err:
+        assert True
+
     try:
         mb.update("testDBRollbackUpdate", {})
     except mybatis.errors.DatabaseError as err: