Pythonでデータベース接続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方法であることができますunittestの最後の

//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',...))