Fixup and test edit prediction mocks (#52991)

Follow-up of https://github.com/zed-industries/zed/pull/40185

Release Notes:

- N/A
f8d646794dKirill Bulatov committed on 4/2/2026, 1:53:37 PM· committed by GitHubparentc4d7004
1 file changedLine totals unavailable