119 lines
4.0 KiB
Python
119 lines
4.0 KiB
Python
'''Test percolator.py.'''
|
|
from test.support import requires
|
|
requires('gui')
|
|
|
|
import unittest
|
|
from tkinter import Text, Tk, END
|
|
from idlelib.percolator import Percolator, Delegator
|
|
|
|
|
|
class MyFilter(Delegator):
|
|
def __init__(self):
|
|
Delegator.__init__(self, None)
|
|
|
|
def insert(self, *args):
|
|
self.insert_called_with = args
|
|
self.delegate.insert(*args)
|
|
|
|
def delete(self, *args):
|
|
self.delete_called_with = args
|
|
self.delegate.delete(*args)
|
|
|
|
def uppercase_insert(self, index, chars, tags=None):
|
|
chars = chars.upper()
|
|
self.delegate.insert(index, chars)
|
|
|
|
def lowercase_insert(self, index, chars, tags=None):
|
|
chars = chars.lower()
|
|
self.delegate.insert(index, chars)
|
|
|
|
def dont_insert(self, index, chars, tags=None):
|
|
pass
|
|
|
|
|
|
class PercolatorTest(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.root = Tk()
|
|
cls.text = Text(cls.root)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
del cls.text
|
|
cls.root.destroy()
|
|
del cls.root
|
|
|
|
def setUp(self):
|
|
self.percolator = Percolator(self.text)
|
|
self.filter_one = MyFilter()
|
|
self.filter_two = MyFilter()
|
|
self.percolator.insertfilter(self.filter_one)
|
|
self.percolator.insertfilter(self.filter_two)
|
|
|
|
def tearDown(self):
|
|
self.percolator.close()
|
|
self.text.delete('1.0', END)
|
|
|
|
def test_insertfilter(self):
|
|
self.assertIsNotNone(self.filter_one.delegate)
|
|
self.assertEqual(self.percolator.top, self.filter_two)
|
|
self.assertEqual(self.filter_two.delegate, self.filter_one)
|
|
self.assertEqual(self.filter_one.delegate, self.percolator.bottom)
|
|
|
|
def test_removefilter(self):
|
|
filter_three = MyFilter()
|
|
self.percolator.removefilter(self.filter_two)
|
|
self.assertEqual(self.percolator.top, self.filter_one)
|
|
self.assertIsNone(self.filter_two.delegate)
|
|
|
|
filter_three = MyFilter()
|
|
self.percolator.insertfilter(self.filter_two)
|
|
self.percolator.insertfilter(filter_three)
|
|
self.percolator.removefilter(self.filter_one)
|
|
self.assertEqual(self.percolator.top, filter_three)
|
|
self.assertEqual(filter_three.delegate, self.filter_two)
|
|
self.assertEqual(self.filter_two.delegate, self.percolator.bottom)
|
|
self.assertIsNone(self.filter_one.delegate)
|
|
|
|
def test_insert(self):
|
|
self.text.insert('insert', 'foo')
|
|
self.assertEqual(self.text.get('1.0', END), 'foo\n')
|
|
self.assertTupleEqual(self.filter_one.insert_called_with,
|
|
('insert', 'foo', None))
|
|
|
|
def test_modify_insert(self):
|
|
self.filter_one.insert = self.filter_one.uppercase_insert
|
|
self.text.insert('insert', 'bAr')
|
|
self.assertEqual(self.text.get('1.0', END), 'BAR\n')
|
|
|
|
def test_modify_chain_insert(self):
|
|
filter_three = MyFilter()
|
|
self.percolator.insertfilter(filter_three)
|
|
self.filter_two.insert = self.filter_two.uppercase_insert
|
|
self.filter_one.insert = self.filter_one.lowercase_insert
|
|
self.text.insert('insert', 'BaR')
|
|
self.assertEqual(self.text.get('1.0', END), 'bar\n')
|
|
|
|
def test_dont_insert(self):
|
|
self.filter_one.insert = self.filter_one.dont_insert
|
|
self.text.insert('insert', 'foo bar')
|
|
self.assertEqual(self.text.get('1.0', END), '\n')
|
|
self.filter_one.insert = self.filter_one.dont_insert
|
|
self.text.insert('insert', 'foo bar')
|
|
self.assertEqual(self.text.get('1.0', END), '\n')
|
|
|
|
def test_without_filter(self):
|
|
self.text.insert('insert', 'hello')
|
|
self.assertEqual(self.text.get('1.0', 'end'), 'hello\n')
|
|
|
|
def test_delete(self):
|
|
self.text.insert('insert', 'foo')
|
|
self.text.delete('1.0', '1.2')
|
|
self.assertEqual(self.text.get('1.0', END), 'o\n')
|
|
self.assertTupleEqual(self.filter_one.delete_called_with,
|
|
('1.0', '1.2'))
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|