当前位置:  开发笔记 > 编程语言 > 正文

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

如何解决《Python单元测试中的Mocking数据库连接/未知类型》经验,为你挑选了1个好方法。

这里是蟒蛇新手.我的类使用数据库连接来包装一些函数.我已经成功找到了一些基本的例子.对于我正在使用的更复杂的库,我找不到模拟数据库连接的近似示例.在我的,

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正在返回一个模拟,而不是我预期的[500]一个项目列表:

AssertionError:  != 500

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

我需要知道如何模拟响应,这将允许我在我的方法中测试逻辑.



1> Old Fox..:

您的问题的来源如下:

with self.db_con.cursor() as cursor:

withlines调用__enter__方法,在您的情况下生成一个新的模拟.

解决方案是模拟__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")

推荐阅读
刘美娥94662
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有