单元测试时云 NDB 上下文的包装？

如何解决单元测试时云 NDB 上下文的包装？

我正在将 Python 2 GAE 应用程序迁移到 Python 3，并且正在将 ndb 替换为 cloud-ndb。这涉及改变行，如

do_something_with_ndb()

到

with client.context() as context:
    do_something_with_ndb()

对于 Flask 应用程序，您可以使用 Wsgi 中间件（请参阅 here）来自动处理此问题，这非常棒。

是否可以为单元测试做类似的事情？在整个测试代码中添加上下文很乏味。

为 nose 和 pytest 寻找解决方案，因为我两者都有。

解决方法

在整个测试代码中添加 with client.context() as context: 的练习之后，我得出的结论是，您可能不希望在测试代码中对上下文进行任何类型的自动包装。

对于单元测试，它们通常在 Flask 应用程序之外完成，因此您需要将每个单元文本包装在上下文中。

对于路由（或处理程序）测试，大部分处理都在 Flask 应用程序中进行，因此您无法将测试包装在上下文中（双重包装会引发错误）。

但是，对于单元测试和路由测试，设置测试和其他事情有很多边缘情况。因此，您需要灵活地根据需要添加上下文，而全局包装器将不起作用。