Co-authored-by: Albert Jiang <aj@mistral.ai>
Co-authored-by: Clément Drouin <clement.drouin@mistral.ai>
Co-authored-by: Laure Hugo <201583486+laure0303@users.noreply.github.com>
Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai>
Co-authored-by: Mert Unsal <mert.unsal@mistral.ai>
Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com>
Co-authored-by: Paul VEZIA <166131032+le-codeur-rapide@users.noreply.github.com>
Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai>
Co-authored-by: Quentin <quentin.torroba@mistral.ai>
Co-authored-by: Mistral Vibe <vibe@mistral.ai>
This commit is contained in:
maiengineering 2026-07-01 19:03:09 +02:00 committed by GitHub
parent 4e495f658d
commit ac8f1a09fd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
83 changed files with 1979 additions and 572 deletions

View file

@ -45,9 +45,10 @@ REPLACE_ALL_CONTENT = "\n".join([
" count = count + 1",
" return count",
"",
"def reset():",
" count = 0",
" return count",
"class Counter:",
" def reset(self):",
" count = 0 # start over",
" return count",
])
@ -69,6 +70,28 @@ class EditReplaceAllApprovalApp(BaseSnapshotTestApp):
await self._switch_to_approval_app("edit", args)
LONG_OLD = " message = " + " + ".join(f'"word_{i}"' for i in range(40))
LONG_NEW = " message = " + " + ".join(f'"token_{i}"' for i in range(40))
OVERFLOW_CONTENT = "\n".join(["def build_message():", LONG_OLD, " return message"])
class EditOverflowApprovalApp(BaseSnapshotTestApp):
_diff_theme: str = "tokyo-night"
async def on_ready(self) -> None:
await super().on_ready()
self.theme = self._diff_theme
path = Path("src/message.py")
path.parent.mkdir(parents=True, exist_ok=True)
path.write_text(OVERFLOW_CONTENT)
args = EditArgs(
file_path="src/message.py",
old_string=f"{LONG_OLD}\n return message",
new_string=f"{LONG_NEW}\n return message.upper()",
)
await self._switch_to_approval_app("edit", args)
def test_snapshot_edit_approval_diff(snap_compare: SnapCompare) -> None:
async def run_before(pilot: Pilot) -> None:
await pilot.pause(0.3)
@ -100,3 +123,16 @@ def test_snapshot_edit_approval_diff_replace_all(snap_compare: SnapCompare) -> N
terminal_size=(100, 30),
run_before=run_before,
)
def test_snapshot_edit_approval_diff_horizontal_overflow(
snap_compare: SnapCompare,
) -> None:
async def run_before(pilot: Pilot) -> None:
await pilot.pause(0.3)
assert snap_compare(
"test_ui_snapshot_edit_diff.py:EditOverflowApprovalApp",
terminal_size=(100, 30),
run_before=run_before,
)