"""Tests the merge cell api."""


from .utils import EDITOR_PAGE


INITIAL_CELLS = [
    "foo = 5",
    "bar = 10",
    "baz = 15",
    "print(foo)",
    "print(bar)",
    "print(baz)",
]


def test_merge_cells(prefill_notebook):
    notebook_frontend = prefill_notebook(INITIAL_CELLS)
    a, b, c, d, e, f = INITIAL_CELLS

    # Before merging, there are 6 separate cells
    assert notebook_frontend.get_cells_contents() == [a, b, c, d, e, f]

    # Focus on the second cell and merge it with the cell above
    notebook_frontend.focus_cell(1)
    notebook_frontend.evaluate("Jupyter.notebook.merge_cell_above();", EDITOR_PAGE)
    merged_a_b = f"{a}\n\n{b}"
    assert notebook_frontend.get_cells_contents() == [merged_a_b, c, d, e, f]

    # Focus on the second cell and merge it with the cell below
    notebook_frontend.focus_cell(1)
    notebook_frontend.evaluate("Jupyter.notebook.merge_cell_below();", EDITOR_PAGE)
    merged_c_d = f"{c}\n\n{d}"
    assert notebook_frontend.get_cells_contents() == [merged_a_b, merged_c_d, e, f]

    # Merge everything down to a single cell with selected cells
    notebook_frontend.select_cell_range(0, 3)
    notebook_frontend.evaluate("Jupyter.notebook.merge_selected_cells();", EDITOR_PAGE)
    merged_all = f"{merged_a_b}\n\n{merged_c_d}\n\n{e}\n\n{f}"
    assert notebook_frontend.get_cells_contents() == [merged_all]