Function process_taint

tests/test_taint.py:14–25

(src: str, pattern: str, cache_mock, taint: str="tainted")

@patch("aura.cache.ASTPatternCache.proxy")
def process_taint(src: str, pattern: str, cache_mock, taint: str="tainted"):
    tree = collect(dedent(src), minimal=True)
    loc = ScanLocation(location="<unknown>")
    p = ASTPattern({
        "pattern": pattern,
        "taint": taint
    })

    cache_mock.return_value = [p]

    v = Visitor.run_stages(location=loc,  ast_tree=tree)
    return v.tree[-1]