Question

我尝试编写一个单元测试，用于验证在将错误的密钥传递给字典时创建KeyError。

引发异常的代码：

connections = SettingsManager().get_connections()
try:
    connection = connections[self.conn_name]
except Exception:
    self.log.error("Connection %s does not exist, exiting." % conn_name)
    self.log.error(sys.exc_info()[0])
    raise

我看过并发现使用lambda的KeyError测试，但我没有太多运气。这是我到目前为止的测试，但它与实际的KeyError错误。

def test_bad_connection(self):
    #Testing to see if a non existant connection will fail gracefully.
    records = [1, 2]
    given_result = DataConnector("BadConnection").generate_data(self.table, records)
    expected_result = "BadConnection"

    self.assertRaises(KeyError, given_result[:1])

Answer 1

assertRaises()将为您调用该函数，并声明该调用会引发异常：

records = [1, 2]
connector = DataConnector("BadConnection")

self.assertRaises(KeyError, connector.generate_data, self.table, records)

或者，使用assertRaises()作为上下文管理器：

with self.assertRaises(KeyError) as raises:
    DataConnector("BadConnection").generate_data(self.table, records)

它具有额外的优势，即上下文管理器允许您访问引发的异常：

self.assertEqual(raises.exception.message, "BadConnection")

Answer 2

self.assertRaise()只需要一个可调用的，所以

self.assertRaises(KeyError, given_result[:1])
在测试时会给你一个实际的KeyError，

self.assertRaises(KeyError, lambda: given_result[:1])
应该管用。

一般来说：
不起作用：self.assertRaises(KeyError, mydict[mykey])测试中的#KeyError
工作：self.assertRaises(KeyError, lambda: mydict[mykey])
是否有效：self.assertRaises(KeyError, mydict.__getitem__, mykey) #but需要一个实际的dict，而不是函数

测试KeyError

2 个答案: