python mock for mysql

Question

나는 코드가있는 다음 file1.py를 가지고있다. 내가 run_q()

이 file1.py 아래

def exec_mysql(query): 
    mysql_conn = MySqlActions(..) 
    .. 
    cur.execute(query) 
    mysql_conn.commit() 
    mysql_conn.close() 

def run_q(): 
    qa = "delete from table where dts = '%s'" % val 
    exec_mysql(qa) 

테스트하기 위해 모의 테스트를 만들려고하고있어 모의 코드입니다. run_q() 메소드의 모의 표현 방법을 잘 모릅니다. 이것을 제시하는 것이 올바른 방법일까요?

test_file1.py는

import mock 
@mock.patch('file1.exec_mysql') 
def test_run(mysql_mock) 
    run_q = mock.Mock() 
    query = "delete from table where dts = '2015-01-01'" 
    mysql_mock.assert_called_with(query) 

Answer 1

당신은 거의 바로 있어요. run_q을 모의 할 필요가 없습니다. 테스트 중에 호출해야합니다.

동작 예 :

app.py

def exec_mysql(query): 
    # do something 
    return query 


def run_q(): 
    qa = 'blahblahblah' 
    exec_mysql(qa) 

tests.py

from unittest import mock 
from app import run_q 

@mock.patch('app.exec_mysql') 
def test_run_q(mysql_mock): 
    run_q() 
    mysql_mock.assert_called_with('blahblahblah') 

테스트 실행 :

$ pytest -vvv tests.py 
===================== test session starts ===================== 
platform linux -- Python 3.5.2, pytest-3.2.1, py-1.4.34 
cachedir: .cache 
rootdir: /home/kris/projects/tmp, inifile: 
plugins: mock-1.6.2, celery-4.1.0 
collected 1 item            

tests.py::test_run_q PASSED 

================== 1 passed in 0.00 seconds ===================