使用 mysql.connector 数据库连接时要修补哪个目标？

Question

修补数据库查询：

导入 mysql.connector 从烧瓶导入 g

@before_request
def con():  
    g.db=mysql.connector.connect("credentials")  #database connection

@route('/insert')   #passing data for insertion  as a 'json' string
def insert():
    cursor = g.db.cursor()
    cursor.execute('Insert into  test("ID,name")')
    cursor.close()
    g.db.close()

test.py

def test_conn():
 with patch(app.mysql.connector) as mock_mysql:
  conn()
  mock_mysql.connect.assert_called_with("credentials")

def test_insert():
 with patch(target =?) as mock:
  #mock execute() and close()?

我是否必须修补我的（conn() 和 g）或（mysql.connector 和 g）作为目标？

如何模拟execute() 和close()？

Answer 1

你可以在这里模拟mysql.connector.connect，模拟库从那里接管：

with patch(target='mysql.connector.connect') as mock:

然后代码调用connect 模拟，返回一个存储在g.db 中的新 模拟对象。您可以在测试函数中访问这些对象：

connection = mock.return_value
cursor = connection.cursor.return_value

然后你可以测试connection.close和cursor.execute和cursor.close是否被调用。