Edit File: test_replacedialog.py
"""Unittest for idlelib.ReplaceDialog""" from test.support import requires requires('gui') import unittest from unittest.mock import Mock from tkinter import Tk, Text from idlelib.idle_test.mock_tk import Mbox import idlelib.SearchEngine as se import idlelib.ReplaceDialog as rd orig_mbox = se.tkMessageBox showerror = Mbox.showerror class ReplaceDialogTest(unittest.TestCase): @classmethod def setUpClass(cls): cls.root = Tk() cls.root.withdraw() se.tkMessageBox = Mbox cls.engine = se.SearchEngine(cls.root) cls.dialog = rd.ReplaceDialog(cls.root, cls.engine) cls.dialog.ok = Mock() cls.text = Text(cls.root) cls.text.undo_block_start = Mock() cls.text.undo_block_stop = Mock() cls.dialog.text = cls.text @classmethod def tearDownClass(cls): se.tkMessageBox = orig_mbox del cls.text, cls.dialog, cls.engine cls.root.destroy() del cls.root def setUp(self): self.text.insert('insert', 'This is a sample sTring') def tearDown(self): self.engine.patvar.set('') self.dialog.replvar.set('') self.engine.wordvar.set(False) self.engine.casevar.set(False) self.engine.revar.set(False) self.engine.wrapvar.set(True) self.engine.backvar.set(False) showerror.title = '' showerror.message = '' self.text.delete('1.0', 'end') def test_replace_simple(self): # Test replace function with all options at default setting. # Wrap around - True # Regular Expression - False # Match case - False # Match word - False # Direction - Forwards text = self.text equal = self.assertEqual pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it # test accessor method self.engine.setpat('asdf') equal(self.engine.getpat(), pv.get()) # text found and replaced pv.set('a') rv.set('asdf') self.dialog.open(self.text) replace() equal(text.get('1.8', '1.12'), 'asdf') # dont "match word" case text.mark_set('insert', '1.0') pv.set('is') rv.set('hello') replace() equal(text.get('1.2', '1.7'), 'hello') # dont "match case" case pv.set('string') rv.set('world') replace() equal(text.get('1.23', '1.28'), 'world') # without "regular expression" case text.mark_set('insert', 'end') text.insert('insert', '\nline42:') before_text = text.get('1.0', 'end') pv.set('[a-z][\d]+') replace() after_text = text.get('1.0', 'end') equal(before_text, after_text) # test with wrap around selected and complete a cycle text.mark_set('insert', '1.9') pv.set('i') rv.set('j') replace() equal(text.get('1.8'), 'i') equal(text.get('2.1'), 'j') replace() equal(text.get('2.1'), 'j') equal(text.get('1.8'), 'j') before_text = text.get('1.0', 'end') replace() after_text = text.get('1.0', 'end') equal(before_text, after_text) # text not found before_text = text.get('1.0', 'end') pv.set('foobar') replace() after_text = text.get('1.0', 'end') equal(before_text, after_text) # test access method self.dialog.find_it(0) def test_replace_wrap_around(self): text = self.text equal = self.assertEqual pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it self.engine.wrapvar.set(False) # replace candidate found both after and before 'insert' text.mark_set('insert', '1.4') pv.set('i') rv.set('j') replace() equal(text.get('1.2'), 'i') equal(text.get('1.5'), 'j') replace() equal(text.get('1.2'), 'i') equal(text.get('1.20'), 'j') replace() equal(text.get('1.2'), 'i') # replace candidate found only before 'insert' text.mark_set('insert', '1.8') pv.set('is') before_text = text.get('1.0', 'end') replace() after_text = text.get('1.0', 'end') equal(before_text, after_text) def test_replace_whole_word(self): text = self.text equal = self.assertEqual pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it self.engine.wordvar.set(True) pv.set('is') rv.set('hello') replace() equal(text.get('1.0', '1.4'), 'This') equal(text.get('1.5', '1.10'), 'hello') def test_replace_match_case(self): equal = self.assertEqual text = self.text pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it self.engine.casevar.set(True) before_text = self.text.get('1.0', 'end') pv.set('this') rv.set('that') replace() after_text = self.text.get('1.0', 'end') equal(before_text, after_text) pv.set('This') replace() equal(text.get('1.0', '1.4'), 'that') def test_replace_regex(self): equal = self.assertEqual text = self.text pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it self.engine.revar.set(True) before_text = text.get('1.0', 'end') pv.set('[a-z][\d]+') rv.set('hello') replace() after_text = text.get('1.0', 'end') equal(before_text, after_text) text.insert('insert', '\nline42') replace() equal(text.get('2.0', '2.8'), 'linhello') pv.set('') replace() self.assertIn('error', showerror.title) self.assertIn('Empty', showerror.message) pv.set('[\d') replace() self.assertIn('error', showerror.title) self.assertIn('Pattern', showerror.message) showerror.title = '' showerror.message = '' pv.set('[a]') rv.set('test\\') replace() self.assertIn('error', showerror.title) self.assertIn('Invalid Replace Expression', showerror.message) # test access method self.engine.setcookedpat("\'") equal(pv.get(), "\\'") def test_replace_backwards(self): equal = self.assertEqual text = self.text pv = self.engine.patvar rv = self.dialog.replvar replace = self.dialog.replace_it self.engine.backvar.set(True) text.insert('insert', '\nis as ') pv.set('is') rv.set('was') replace() equal(text.get('1.2', '1.4'), 'is') equal(text.get('2.0', '2.3'), 'was') replace() equal(text.get('1.5', '1.8'), 'was') replace() equal(text.get('1.2', '1.5'), 'was') def test_replace_all(self): text = self.text pv = self.engine.patvar rv = self.dialog.replvar replace_all = self.dialog.replace_all text.insert('insert', '\n') text.insert('insert', text.get('1.0', 'end')*100) pv.set('is') rv.set('was') replace_all() self.assertNotIn('is', text.get('1.0', 'end')) self.engine.revar.set(True) pv.set('') replace_all() self.assertIn('error', showerror.title) self.assertIn('Empty', showerror.message) pv.set('[s][T]') rv.set('\\') replace_all() self.engine.revar.set(False) pv.set('text which is not present') rv.set('foobar') replace_all() def test_default_command(self): text = self.text pv = self.engine.patvar rv = self.dialog.replvar replace_find = self.dialog.default_command equal = self.assertEqual pv.set('This') rv.set('was') replace_find() equal(text.get('sel.first', 'sel.last'), 'was') self.engine.revar.set(True) pv.set('') replace_find() if __name__ == '__main__': unittest.main(verbosity=2)