/pytest-db

Pytest plugin for mysql query or change

Primary LanguagePythonMIT LicenseMIT

pytest-db

Languate - Python PyPI - License PyPI PyPI - Downloads

Pytest操作数据库

如何使用

  1. 安装 pytest-db

使用pip从github安装

pip install pytest-db
  1. 使用方法 在环境变量中添加
export DB_URI=mysql://root:password@localhost:3306/test

或在pytest.ini中配置

[pytest]
db_uri = mysql://root:password@localhost:3306/test

或命令行传入

$ pytest --db-uri=mysql://root:password@localhost:3306/test

或使用Fixture配置

import pytest


@pytest.fixture
def db_uri():
    return 'mysql://root:password@localhost:3306/test'
  1. 使用fixture函数: db
def test_a(db):
    print(db.query('select id from user limit 1;')

使用pytest -s 运行，查看结果

...
[{'id': 123321336}]
...

游标使用pymysql.cursors.DictCursor，结果使用fetchone,返回字典格式的结果

  1. db对象支持的方法
  • db.query(sql): 执行查询sql，返回一条结果
  • db.query_all(sql): 执行查询sql，返回所有结果
  • db.execute(sql): 执行修改sql