Function assert_result_equal

test/pytests/test_sqlexecute.py:19–50

Assert that an sqlexecute.run() result matches the expected values.

(
    result,
    preamble=None,
    header=None,
    rows=None,
    status=None,
    status_plain=None,
    postamble=None,
    auto_status=True,
    assert_contains=False,
)

def assert_result_equal(
    result,
    preamble=None,
    header=None,
    rows=None,
    status=None,
    status_plain=None,
    postamble=None,
    auto_status=True,
    assert_contains=False,
):
    """Assert that an sqlexecute.run() result matches the expected values."""
    if status_plain is None and auto_status and rows:
        status_plain = f"{len(rows)} row{'s' if len(rows) > 1 else ''} in set"
        status = FormattedText([('', status_plain)])
    fields = {
        "preamble": preamble,
        "header": header,
        "rows": rows,
        "postamble": postamble,
        "status": status,
        "status_plain": status_plain,
    }

    if assert_contains:
        # Do a loose match on the results using the *in* operator.
        for key, field in fields.items():
            if field:
                assert field in result[0][key]
    else:
        # Do an exact match on the fields.
        assert result == [fields]


@dbtest