为什么嘲弄＆＃39;打开＆＃39;并返回FileNotFoundError引发AttributeError：__ exit__？

Question

使用open加注FileNotFoundError模拟AttributeError: __exit__进行测试。为什么会发生这种情况，我该怎么做才能解决它？

以下代码打开一个简单的文本文件。如果文件丢失，则会生成默认值。它已通过常规运行进行检查，看起来效果很好。

so_main.py

import os

import so_config


def load_savelocation():
    path = os.path.join(so_config.ROOT, so_config.SAVELOCATION_FN)
    savelocation_path = os.path.normpath(path)
    try:
        with open(savelocation_path) as f:
            so_config.SAVELOCATION_PATH = f.readline()
    except FileNotFoundError:
        so_config.SAVELOCATION_PATH = so_config.ROOT

so_config.py

import os

ROOT, _ = os.path.split(__file__)
SAVELOCATION_PATH = None
SAVELOCATION_FN = 'savelocation.ini'

单元测试是一个不同的故事。我在open中嘲笑了so.main命令。 test_so_main.py有两个测试：一个用于正常打开存在的文件，另一个用于测试FileNotFoundError处理。

常规文件打开test_read_path_from_disk_file_into_config_py的第一个测试工作正常。

第二次测试失败，因为FileNotFoundError引发了AttributeError: __exit__。 我可以将self.mock_open.return_value设置为FileNotFoundError，也可以将其设置为'garbage'。它没有任何区别。

test_so_main.py

import unittest
import unittest.mock as mock

import so_config
import so_main


class TestReadSaveLocation(unittest.TestCase):
    def setUp(self):
        self.savelocation_path = so_config.SAVELOCATION_PATH
        self.root = so_config.ROOT
        so_config.ROOT = 'program root'
        p = mock.patch('so_main.open')
        self.mock_open = p.start()
        self.addCleanup(p.stop)

    def tearDown(self):
        so_config.SAVELOCATION_PATH = self.savelocation_path
        so_config.ROOT = self.root

    def test_read_path_from_disk_file_into_config_py(self):
        self.mock_open().__enter__().readline.return_value = 'data files location'
        so_main.load_savelocation()
        self.assertEqual('data files location', so_config.SAVELOCATION_PATH)

    def test_missing_file_defaults_savelocation_to_program_root(self):
        self.mock_open.return_value = FileNotFoundError
        so_main.load_savelocation()
        self.assertEqual('program root', so_config.SAVELOCATION_PATH)

我在Windows 7计算机上通过PyCharm 2016.2.1运行Python 3.5.2。

Answer 1

你模拟了返回异常的函数而不是 raise 它。

尝试使用side_effect（docs）：

self.mock_open.side_effect = FileNotFoundError