Function test_should_collect_sources

tests/sync/test_tracing.py:82–128

(
    context: BrowserContext,
    page: Page,
    server: Server,
    tmp_path: Path,
    show_trace_viewer: Callable[[Path], ContextManager[TraceViewerPage]],
)

def test_should_collect_sources(
    context: BrowserContext,
    page: Page,
    server: Server,
    tmp_path: Path,
    show_trace_viewer: Callable[[Path], ContextManager[TraceViewerPage]],
) -> None:
    context.tracing.start(sources=True)
    page.goto(server.EMPTY_PAGE)
    page.set_content("<button>Click</button>")

    def my_method_outer() -> None:
        def my_method_inner() -> None:
            page.get_by_text("Click").click()

        my_method_inner()

    my_method_outer()
    path = tmp_path / "trace.zip"
    context.tracing.stop(path=path)

    with show_trace_viewer(path) as trace_viewer:
        expect(trace_viewer.action_titles).to_have_text(
            [
                re.compile(r'Navigate to "/empty\.html"'),
                re.compile(r"Set content"),
                re.compile(r"Click"),
            ]
        )
        trace_viewer.show_source_tab()
        # Check that stack frames are shown (they might be anonymous in Python)
        expect(trace_viewer.stack_frames).to_contain_text(
            [
                re.compile(r"my_method_inner"),
                re.compile(r"my_method_outer"),
                re.compile(r"test_should_collect_sources"),
            ]
        )

        trace_viewer.select_action("Set content")
        # Check that the source file is shown
        expect(trace_viewer.page.locator(".source-tab-file-name")).to_have_attribute(
            "title", re.compile(r".*test_.*\.py")
        )
        expect(trace_viewer.page.locator(".source-line-running")).to_contain_text(
            'page.set_content("<button>Click</button>")'
        )


def test_should_collect_trace_with_resources_but_no_js(