python - 有没有办法判断函数是否在单元测试中执行？

Question

我正在使用一个配置文件来获取我的数据库的信息。它总是获取主机名，然后从该配置文件中找出要使用的数据库选项。我想知道我是否在单元测试中，而是使用内存中的 sqlite 数据库。有没有办法判断我是否在单元测试中，或者我是否必须找到不同的方法？

Answer 1

使用 IoC 将您的数据库依赖项注入(inject)到您的类中。这应该通过在您的类的构造函数中将一个存储库对象传递给它来完成。请注意，您不一定需要 IoC 容器来执行此操作。您只需要一个存储库接口(interface)和存储库的两个实现。

注意:在 Python 中，IoC 的工作方式略有不同。参见 http://code.activestate.com/recipes/413268/获取更多信息。