16

我使用 Anaconda 发行版中的 Python 3.4。在这个发行版中，我找到了pymysql连接到现有 MySQL 数据库的库，该数据库位于另一台计算机上。

import pymysql
config = {
      'user': 'my_user',
      'passwd': 'my_passwd',
      'host': 'my_host',
      'port': my_port
    }

    try:
        cnx = pymysql.connect(**config)
    except pymysql.err.OperationalError : 
        sys.exit("Invalid Input: Wrong username/database or password")

我现在想为我的应用程序编写测试代码，我想在setUp每个测试用例中创建一个非常小的数据库，最好是在内存中。但是，当我突然尝试使用 时pymysql，它无法建立连接。

def setUp(self):
    config = {
      'user': 'test_user',
      'passwd': 'test_passwd',
      'host': 'localhost'
    }
    cnx = pymysql.connect(**config)

pymysql.err.OperationalError: (2003, "Can't connect to MySQL server on 'localhost' ([Errno 61] Connection refused)")

我一直在谷歌搜索，发现了一些关于SQLiteand的东西MySQLdb。我有以下问题：

  1. 是否适合在内存sqlite3中MySQLdb快速创建数据库？
  2. 如何MySQLdb在 Anaconda 包中安装？
  3. 是否有在 中创建的测试数据库的示例setUp？这甚至是个好主意吗？

我的计算机上没有本地运行的 MySQL 服务器。

4

2 回答 2

14

您可以使用testing.mysqld ( pip install testing.mysqld)模拟 mysql 数据库

由于出现了一些嘈杂的错误日志，我在测试时喜欢这个设置：

import testing.mysqld
from sqlalchemy import create_engine

# prevent generating brand new db every time.  Speeds up tests.
MYSQLD_FACTORY = testing.mysqld.MysqldFactory(cache_initialized_db=True, port=7531)


def tearDownModule():
    """Tear down databases after test script has run.
    https://docs.python.org/3/library/unittest.html#setupclass-and-teardownclass
    """
    MYSQLD_FACTORY.clear_cache()


class TestWhatever(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.mysql = MYSQLD_FACTORY()
        cls.db_conn = create_engine(cls.mysql.url()).connect()

    def setUp(self):
        self.mysql.start()
        self.db_conn.execute("""CREATE TABLE `foo` (blah)""")

    def tearDown(self):
        self.db_conn.execute("DROP TABLE foo")

    @classmethod
    def tearDownClass(cls):
        cls.mysql.stop()  # from source code we can see this kills the pid

    def test_something(self):
        # something useful
于 2019-05-09T02:49:35.087 回答
10

pymysql、MySQLdb 和 sqlite 都需要一个真正的数据库来连接。如果你只想测试你的代码，你应该在你想要测试的模块上模拟 pymysql 模块，并相应地使用它（在你的测试代码中：你可以设置模拟对象以将硬编码的结果返回到预定义的 SQL 语句）

在以下位置查看有关本机 Python 模拟库的文档： https ://docs.python.org/3/library/unittest.mock.html

或者，对于 Python 2： https ://pypi.python.org/pypi/mock

于 2015-02-10T13:16:33.540 回答