Function test_save_with_header

tests/_server/api/endpoints/test_files.py:159–205

(client: TestClient)

)
@with_session(SESSION_ID)
def test_save_with_header(client: TestClient) -> None:
    filename = get_session_manager(client).workspace.get_unique_file_key()
    assert filename
    path = Path(filename)
    assert path.exists()

    copyright_year = datetime.now().year
    header = (
        '"""This is a docstring"""\n\n'
        + f"# Copyright {copyright_year}\n# Linter ignore\n"
    )
    # Prepend a header to the file
    contents = path.read_text()
    contents = header + contents
    path.write_text(contents, encoding="UTF-8")

    response = client.post(
        "/api/kernel/save",
        headers=HEADERS,
        json={
            "cellIds": ["1"],
            "filename": str(path),
            "codes": ["import marimo as mo"],
            "names": ["my_cell"],
            "configs": [
                {
                    "hide_code": True,
                    "disabled": False,
                }
            ],
        },
    )

    assert response.status_code == 200, response.text
    assert "import marimo" in response.text

    def _assert_contents():
        file_contents = path.read_text()
        assert "import marimo as mo" in file_contents
        # Race condition with uv (seen in python 3.10)
        if file_contents.startswith("# ///"):
            file_contents = file_contents.split("# ///")[2].lstrip()
        assert file_contents.startswith(header.rstrip()), "Header was removed"
        assert "@app.cell(hide_code=True)" in file_contents
        assert "my_cell" in file_contents

    try_assert_n_times(5, _assert_contents)


@pytest.mark.flaky(reruns=5)