Python模拟数据库连接/单元测试中的未知类型

Question

此处为python的Newby。我的类使用数据库连接来包装一些函数。我已经成功地想出了一些基本的例子。对于我正在使用的更复杂的库，我找不到模仿数据库连接的近义词。在我的

class DBSAccess():
def __init__(self, db_con):
self.db_con = db_con
def get_db_perm(self, target_user):
## this is where I start having trouble
with self.db_con.cursor() as cursor:
cursor.execute("SELECT CAST(sum(maxperm) AS bigint) 
FROM dbc.diskspace 
WHERE  databasename = '%s' 
GROUP BY databasename" % (target_user))
res = cursor.fetchone()
if res is not None:
return res[0]
else:
msg = target_user + " does not exist"
return msg

其中db_con是teradata。UdaExec返回一个连接

udaExec = teradata.UdaExec (appName="whatever", version="1.0", logConsole=True)
db_con = udaExec.connect(method="odbc", system='my_sys', username='my_name', password='my_pswd')
dbc_instance = tdtestpy.DBSaccess (db_con)

因此，为了让我的测试不使用任何真正的连接，我必须模拟出一些东西。我试过这个组合：

class DBAccessTest(unittest.TestCase):
def test_get_db_free_perm_expects_500(self):
uda_exec = mock.Mock(spec=teradata.UdaExec)
db_con = MagicMock(return_value=None)
db_con.cursor.fetchone.return_value = [500]
uda_exec.connect.return_value = db_con
self.dbc_instance = DBSAccess(db_con)
self.assertEqual(self.dbc_instance.get_db_free_perm("dbc"), 500)

但我的结果是一团糟，因为fetchone返回的是一个mock，而不是我所期望的[500]一项列表：

AssertionError: <MagicMock name='mock.connect().cursor().[54 chars]312'> != 500

我发现了一些例子，其中有一个"with block"用于测试操作系统操作，但没有数据库。另外，我不知道db_con.cursor是什么数据类型，所以我不能准确地指定它——我认为游标在Teradata/PyTd的UdaExecConnection.cursor()中找到。

我需要知道如何模拟响应，以便测试方法中的逻辑。

Answer 1

问题的根源在以下行：

with self.db_con.cursor() as cursor:

with行调用__enter__方法，该方法在您的情况下生成一个新的mock。

解决方案是模拟__enter__方法：

db_con.cursor.return_value.__enter__.return_value = cursor

您的测试：

class DBAccessTest(unittest.TestCase):
def test_get_db_free_perm_expects_500(self):
db_con = MagicMock(UdaExecConnection)
cursor = MagicMock(UdaExecCursor)
cursor.fetchone.return_value = [500]
db_con.cursor.return_value.__enter__.return_value = cursor
self.dbc_instance = DBSAccess(db_con)
self.assertEqual(self.dbc_instance.get_db_perm("dbc"), 500)
def test_get_db_free_perm_expects_None(self):
db_con = MagicMock(UdaExecConnection)
cursor = MagicMock(UdaExecCursor)
cursor.fetchone.return_value = None
db_con.cursor.return_value.__enter__.return_value = cursor
self.dbc_instance = DBSAccess(db_con)
self.assertEqual(self.dbc_instance.get_db_perm("dbc"), "dbc does not exist")