test_函数执行一些操作,然后调用另一个函数

调用其他函数将执行某些操作,然后返回结果并

在完成测试后,等待函数“完成”

当然,这不适用于字面返回,所以我的问题是,如果有可能我看不到。

pytest-Test配置中的Test函数是一个fixture:

def test_load_into_database(test_config):

logger.info('Testing load_into_database')

data = {

'filename': 'file_0'

}

accessor = list(database_accessor(test_config))[0]

access_data = accessor['session'].query(accessor['table']).all()

assert access_data[0].filename != data['filename']

accessor['session'].add(accessor['table'](

filename='file_0'

))

accessor['session'].commit()

access_data = accessor['session'].query(accessor['table']).all()

assert access_data[0].filename == data['filename']

def database_accessor(cfg):

setup = 'mysql+pymysql://{}:{}@{}/{}'.format(

cfg['database_user'],

cfg['database_passwd'],

cfg['database_host'],

cfg['database_db']

)

Base = automap_base()

engine = create_engine(setup, echo=False)

Base.prepare(engine, reflect=True)

table = Base.classes.get(cfg['database_table'])

session = Session(engine)

->return and wait

'session': session,

'table': table

}

session.close()

with engine.connect() as con:

con.execution_options(autocommit=True).execute("TRUNCATE TABLE {}".format(cfg['database_table']))

我想要的是,数据库访问器返回字典,然后等待底层函数(在本例中是test_函数)完成,然后继续。

这样,就可以使用具有相同数据库访问器的可变数量的测试函数,而不必在数据库访问器中执行所有不同的测试函数。

回拨

我知道有一种方法,使用回调函数,但这会使我的函数加倍,这是我不想要的。例如。

def test_load_into_database():

database_accessor(load_into_database_1, var1, var2)

def database_accessor(function, args*):

# do stuff

function(args, stuff)

# do other stuff

def load_into_database_1(args, stuff):

# do something