어떻게 단위 테스트 파이썬에서 데이터베이스 연결 pymysql?

Question

데이터베이스를 조롱하여 다음 방법을 테스트하기 위해 단위 테스트 케이스를 작성하려고합니다. 실제 데이터베이스 서버에 실제로 연결하지 않고 데이터베이스 연결을 조롱하는 방법. 나는 샘플 테스트 케이스를 시도했다. 그것이 올바른 방법인지 확실하지 않습니다. 내가 틀렸다면 나를 바로 잡아주세요.

//MySQL.py 
    class MySQL(): 

     retry_interval = 20000 

     def link(self, server,port): 
      try: 
       return pymysql.connect(server, username='name', passwd='pwd', db='demo', int(port)) 
       sys.stdout.write('Connected to database at {}:{}\n'.format(server,port)) 
      except: 
       sys.stderr.write('Retry database connection in {} ms...\n'.format(self.retry_interval)) 
       sleep(self.retry_interval/1000.0) 
       continue 

//test.py 
from unittest.mock import Mock, MagicMock 
from unittest.mock import patch 
import MySQL 

    @patch('MySQL.pymysql') 
    def testLink(self, mysql_mock): 
     mock_cursor = Mock.MagicMock() 
     mysql_mock.connect.return_value.cursor.return_value.__enter__.return_value = mock_cursor 

Answer 1

유일한 유용한 것은 당신은뿐만 아니라 직접 그 패치를 할 수 있도록 당신이 정말로 MySQL.pymysql의 테스트는 connect 방법입니다 수 있습니다하십시오 유닛 테스트 끝의

//test.py 
from unittest.mock import Mock, MagicMock 
from unittest.mock import patch, call 
import MySQL 

@patch('MySQL.pymysql.connect') 
def testLink(self, connect_mock): 
    # You can configure a return value for the connection if you need to... 
    connect_mock.return_value = MagicMock(name='connection_return', return_value=...) 

    # You can now test that your connection is being called... 
    self.assertEqual(1, connect_mock.call_count) 

    # You can also check the call parameters... 
    self.assertEqual(connect_mock.call_args_list[0], call(server, username='name', passwd='pwd',...))