如何正確地對FileNotFoundError進行單元測試

Question

使用unittest模塊，打開文件時，如何測試FileNotFoundError是否引發如下：

func(filename):
    try:
        file = open(filename)
    except FileNotFoundError:
        exit()

單元測試模塊

class TestFunc(unittest.TestCase):

    def test_func(self):
        self.assertRaises(FileNotFoundError, func, "no_exist.txt")

結果錯誤

AssertionError：函數未引發FileNotFoundError

Answer 1

您正在吞沒異常處理程序中的異常，因此調用代碼（單元測試）無法知道錯誤已引發，所看到的只是代碼已運行並退出。

考慮重新籌款，如：

func(filename):
    try:
        file = open(filename)
    except FileNotFoundError:
        # Whatever else you want to do 
        raise

更好的是，您的單元測試不應真正依賴於不存在的文件-單元測試應按定義自包含。 考慮模擬open方法，以便您擁有更多控制權。

Answer 2

使用上下文管理器處理異常

在2.7版中進行了更改：添加了使用assertRaises（）作為上下文管理器的功能。

如果僅給出exception參數，則返回一個上下文管理器，以便可以將內嵌代碼而不是作為函數編寫被測代碼：

with self.assertRaises(SomeException):
    do_something()