Skip to content

test_profiling#

test_count_queries#

Source code in tests/unit/apps/common/test_profiling.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
24
def test_count_queries():
    with count_queries() as counts:
        spatial = Spatial.objects.create(geographic_name="paikka")
        spatial.refresh_from_db()
        Provenance.objects.create(title={"en": "hello"}, spatial=spatial)
        Provenance.objects.create(title={"en": "world"})
        Provenance.objects.filter(title__en="hello").update(title={"fi": "moro"})
        Provenance.objects.all().delete()
    assert counts == {
        "SQLCompiler": Counter({"total": 1, "Spatial": 1}),
        "SQLInsertCompiler": Counter({"total": 3, "Provenance": 2, "Spatial": 1}),
        "SQLUpdateCompiler": Counter({"total": 4, "Provenance": 4}),
        "total": 8,
    }

test_count_queries_log#

Source code in tests/unit/apps/common/test_profiling.py 
27
28
29
30
31
32
33
def test_count_queries_log(caplog):
    logging.disable(logging.NOTSET)
    factories.DatasetFactory()
    with count_queries(log=True):
        Dataset.objects.count()
    assert len(caplog.messages) == 1
    assert "{'total': 1, 'SQLCompiler': Counter({'total': 1, 'Dataset': 1})}" in caplog.messages[0]