test.py

import json
from copy import copy

def test_json():
    tainted_string = TAINTED
    tainted_json = json.loads(tainted_string)
    a = tainted_json["x"]
    b = a.get("y")
    c = b["z"]

def test_str():
    tainted_string = TAINTED
    a = tainted_string.ljust(8)
    b = tainted_string.copy()
    c = tainted_string[:]
    d = tainted_string[::2]
    e = reversed(tainted_string)
    f = copy(tainted_string)
    h = tainted_string.strip()

def test_const_sanitizer1():
    tainted_string = TAINTED
    if tainted_string == "OK":
        not_tainted(tainted_string)
    else:
        still_tainted(tainted_string)

def test_const_sanitizer2():
    tainted_string = TAINTED
    if tainted_string == "OK" or tainted_string == "ALSO_OK":
        not_tainted(tainted_string)
    else:
        still_tainted(tainted_string)

def test_str2():
    tainted_string = TAINTED
    a = str(tainted_string)
    b = bytes(tainted_string) # This is an error in Python 3
    c = bytes(tainted_string, encoding="utf8") # This is an error in Python 2

def cross_over(func, taint):
    return func(taint)

def test_exc_info():
    info = TAINTED_EXCEPTION_INFO
    res = cross_over(exc_info_call, info)

def exc_info_call(arg):
    return arg

def test_untrusted():
    ext = TAINTED_EXTERNAL_STRING
    res = cross_over(untrusted_call, ext)

def exc_untrusted_call(arg):
    return arg