数据库场景用例

以MySQL的增删操作为例，编写功能测试用例。

import pytest
import pymysql
'''
案例测试软件为：MySQL
预置条件：
1. MySQL已安装并启动
'''
# 测试用例一：测试数据添加功能
def test_insert():
    db = pymysql.connect(host='localhost', user='root', password='Password@123', database='test')
    try:
        cursor = db.cursor()
        for i in range(0, 5):
            sql = "INSERT INTO TEST_INFO(DATA) VALUES('data%s');" % str(i)
            cursor.execute(sql)
        db.commit()
        cursor.execute("SELECT COUNT(*) FROM TEST_INFO")
        data_num = cursor.fetchone()[0]
    finally:
        db.close()
    assert 5 == data_num
# 测试用例二：测试数据删除功能
def test_delete():
    db = pymysql.connect(host='localhost', user='root', password='Password@123', database='test')
    try:
        cursor = db.cursor()
        sql = "DELETE FROM TEST_INFO;"
        cursor.execute(sql)
        db.commit()
        cursor.execute("SELECT COUNT(*) FROM TEST_INFO")
        data_num = cursor.fetchone()[0]
    finally:
        db.close()
    assert 0 == data_num
# 案例代码文件名：mysql_testing.py
if __name__ == '__main__':
    pytest.main(["-s", "mysql_testing.py"])