具有从配置文件读取 configparser 的单元测试 python 代码

Question

我是 python 单元测试的新手。我学习并进行了示例单元测试，其中方法接受输入并返回输出。

但是对于下面提到的代码，我有一些问题。

1. 如何在 unittest 中模拟 init 方法的 configparser？路径 /config/program.cfg 位于生产服务器上，并且在 dev 目录中不存在。 program.cfg 文件存在于代码目录中的其他位置。有没有办法在单元测试中处理它？
2. 如何在 unittest 中发送或跳过硬编码路径之类的内容，例如/var/log/info_server.log

3. 如果可能的话，你能告诉我你将如何使用 pytest 模块为下面的代码编写 unittest 吗？这将有助于理解流程，我可以使用其余代码来做到这一点。

   def __init__(self):
       self.setup_logger()
   
       # Read config parameters
       config = configparser.ConfigParser()
       config.read("/config/program.cfg")
       self.host_ip = config.get('default','HostIP')
       self.redis_ip = config.get('default','RedisIP')
       self.redis_port = config.get('default','RedisPort')
       self.info_port = config.get('default','InfoPort')
       self.sqlite_db_file = config.get('default','SQLiteDbFile')
   
       self.connect_redis()
   
   def setup_logger(self):
       #Initialize logger
       self.logger = logging.getLogger(__name__)
       self.logger.setLevel(logging.INFO)
   
       # Create a file handler
       handler = logging.FileHandler('/var/log/info_server.log')
       handler.setLevel(logging.INFO)
   
       # Create a logging format
       formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
       handler.setFormatter(formatter)
   
       # Add the handlers to the logger
       self.logger.addHandler(handler)
   
   def connect_redis(self):
       # Start up a Redis instance
       self.logger.info("Start Redis instance")
       self.ad_info = redis.StrictRedis(host=self.redis_ip, port=self.redis_port, db=0)
   
       if self.ad_info is None:
           self.logger.error("Failed to start Redis instance")
       else:
           self.logger.info("Started Redis instance")
   

Answer 1

当然，我们可以模拟很多东西，但是简单地重构被测试的类，添加配置和日志，会让生活变得更轻松。

def __init__(self, cfg='/config/program.cfg', log='/var/log/info_server.log'):

在/config/program.cfg 不存在的开发服务器中，您可以只是

TheClass(cfg='~/dev.cfg', log='/tmp/dev.log')

self.server_obj.connect_redis() log_mock.logger.info.assert_called_with('Started Redis instance') 不工作

import logging
logging.getLogger

# <function getLogger at 0x7f3d2ed427b8>
logger = logging.getLogger()
type(logger)
# <class 'logging.RootLogger'>
logger.info
# <bound method Logger.info of <RootLogger root (WARNING)>>

补丁应该是这样的：

with mock.patch.object(logging.RootLogger, 'info') as mock_info:

    server_obj = Server(cfg='sth', log='sth')
    mock_info.assert_called_with('xxx')