Function test_custom_object

tests/integration/test_local_variables.py:396–445

(generate_threads, python, tmpdir)

@ALL_PYTHONS
@ALL_SOURCES
def test_custom_object(generate_threads, python, tmpdir):
    # GIVEN
    _, python_executable = python

    program_template = """
import sys
import time

class A:
    pass

def first_func():
    the_local = A()
    second_func(the_local)

def second_func(the_argument):
    fifo = sys.argv[1]
    with open(sys.argv[1], "w") as fifo:
        fifo.write("ready")
    time.sleep(1000)

first_func()
    """

    script = tmpdir / "the_script.py"
    script.write_text(program_template, encoding="utf-8")

    # WHEN

    threads = generate_threads(python_executable, script, tmpdir)

    # THEN

    assert len(threads) == 1
    (thread,) = threads

    frames = list(thread.frames)
    assert (len(frames)) == 3

    repr_regex = re.compile(r"<(?:A|instance|\?\?\?) at 0[xX][0-9a-fA-F]+>")

    first_func_frame = find_frame(frames, "first_func")
    assert "the_local" in first_func_frame.locals
    argument_repr = first_func_frame.locals["the_local"]
    assert repr_regex.match(argument_repr)

    second_func_frame = find_frame(frames, "second_func")
    assert "the_argument" in second_func_frame.arguments
    argument_repr = second_func_frame.arguments["the_argument"]
    assert repr_regex.match(argument_repr)


@pytest.mark.parametrize(