Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-12-08 06:10:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

refactor(tests): Use mock.patch for messagebox dialogs

Browse source
This commit is contained in:
Shamil Abdulaev 2025-11-16 01:18:43 +03:00
parent 66e0495276
commit a154c9f344
No known key found for this signature in database
GPG key ID: F9F7726024C005E7
1 changed files with 4 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Lib/idlelib/idle_test/test_iomenu.py
View file

@ -119,16 +119,12 @@ def test_reload_with_unsaved_changes_cancel(self):
io.set_saved(False)
# Mock askokcancel to return False (cancel)
orig_askokcancel = iomenu.messagebox.askokcancel
iomenu.messagebox.askokcancel = lambda *args, **kwargs: False
try:
with patch('idlelib.iomenu.messagebox.askokcancel', return_value=False) as mock_askokcancel:
result = io.reload(None)
self.assertEqual(result, "break")
# Content should not change
self.assertIn("# Unsaved change", text.get('1.0', 'end-1c'))
finally:
iomenu.messagebox.askokcancel = orig_askokcancel
mock_askokcancel.assert_called_once()
def test_reload_with_unsaved_changes_confirm(self):
# Test reload with unsaved changes and user confirms
@ -149,16 +145,12 @@ def test_reload_with_unsaved_changes_confirm(self):
io.set_saved(False)
# Mock askokcancel to return True (confirm)
orig_askokcancel = iomenu.messagebox.askokcancel
iomenu.messagebox.askokcancel = lambda *args, **kwargs: True
try:
with patch('idlelib.iomenu.messagebox.askokcancel', return_value=True) as mock_askokcancel:
result = io.reload(None)
self.assertEqual(result, "break")
# Content should be reverted to original
self.assertEqual(text.get('1.0', 'end-1c'), "# Original content\n")
finally:
iomenu.messagebox.askokcancel = orig_askokcancel
mock_askokcancel.assert_called_once()
def _extension_in_filetypes(extension):