python - Django：是否有可以将断言作为子测试运行的 assertNumQueries 方法？

Question

我想运行一段代码并确保生成的查询量得到检查，但是，如果此检查失败，我希望测试代码继续进行。

伪代码：

def assertNumQueriesSubTest(func, expected_number):
    with self.countNumQueries as query_count:
        func()

        with self.subTest('query check'):
             self.assertEqual(queries_count, expected_number)

使用经典assertNumQueries，如果查询计数与预期值不匹配，则测试代码会停止，但是，这会阻止我正确调试测试，因为查询计数更改的原因可能并不明显，所以我想知道是否至少行为匹配是否仍然相同。

标签： pythondjangodjango-testing

Answer 1

我为 OpenWISP 编写了一个解决方案，它在 subtest 中执行断言，它记录在这里：openwisp_utils.tests.AssertNumQueriesSubTestMixin。