有些函数需要连接数据库,可以把连接数据库和断开数据的操作放在,指定的函数里. 这样每次测试函数的时候,都调取这个链接数据库的函数即可
@pytest
.fixture
()
def db():
print('Connection successful')
yield
print('Connection closed')
def search_user(user_id
):
d
= {
'001': 'xiaoming'
}
return d
[user_id
]
def test_search(db
):
assert search_user
('001') == 'xiaoming'
参考: https://learning-pytest.readthedocs.io/zh/latest/doc/fixture/setup-and-teardown.html
