我在一个类中有两个以下函数,我需要模拟数据库连接和游标结果。 cx_Oracle.connect 已成功修补,它按预期工作。但是光标、callfunc、fetch_all 不会被模拟。知道正确的语法是什么吗?
class dbconnect:
def db_connect(self, connection_details):
connection = cx_Oracle.connect(user_name,pwd,<connection_string>)
return connection
def execute_function(self, sqlFunction, args):
cursor = self.connection.cursor()
res=cursor.callfunc(sqlFunction, cursor.var(cx_Oracle.CURSOR), args)
results = {'headers' : [x[0] for x in res.description],
'data': res.fetchall()
}
cursor.close()
return results
我的模拟代码
with mock.patch('dbconnect.cx_Oracle.connect') as mockOracle:
result_set = {}
mockOracle.cursor.callfunc.fetch_all = result_set
实际上,您根本不必“模拟”连接和光标!您可以简单地自己对连接和光标进行子类化,如下所示:
class MyConnection(cx_Oracle.Connection):
def cursor(self, scrollable=False):
return MyCursor(self, scrollable)
class MyCursor(cx_Oracle.Cursor):
def execute(self, sql, args):
result = super(MyConnection, self).execute(sql, args)
print("My mock execute...")
return result
我不确定这是否是您的意图,或者您是否知道这种可能性。有了它,您可以添加新功能,并且可以覆盖或扩展现有功能。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)