mirror of
https://github.com/python/cpython.git
synced 2025-07-17 00:05:20 +00:00
4d92158f4c
Part 3 of 3, continuing PR #7689. This covers 14 idlelib modules and their tests, rpc to zoomheight except for run (already done) and tooltip (being done separately).
403 lines
14 KiB
Python
403 lines
14 KiB
Python
"Test codecontext, coverage 100%"
|
|
|
|
from idlelib import codecontext
|
|
import unittest
|
|
from test.support import requires
|
|
from tkinter import Tk, Frame, Text, TclError
|
|
|
|
from unittest import mock
|
|
import re
|
|
from idlelib import config
|
|
|
|
|
|
usercfg = codecontext.idleConf.userCfg
|
|
testcfg = {
|
|
'main': config.IdleUserConfParser(''),
|
|
'highlight': config.IdleUserConfParser(''),
|
|
'keys': config.IdleUserConfParser(''),
|
|
'extensions': config.IdleUserConfParser(''),
|
|
}
|
|
code_sample = """\
|
|
|
|
class C1():
|
|
# Class comment.
|
|
def __init__(self, a, b):
|
|
self.a = a
|
|
self.b = b
|
|
def compare(self):
|
|
if a > b:
|
|
return a
|
|
elif a < b:
|
|
return b
|
|
else:
|
|
return None
|
|
"""
|
|
|
|
|
|
class DummyEditwin:
|
|
def __init__(self, root, frame, text):
|
|
self.root = root
|
|
self.top = root
|
|
self.text_frame = frame
|
|
self.text = text
|
|
|
|
|
|
class CodeContextTest(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
requires('gui')
|
|
root = cls.root = Tk()
|
|
root.withdraw()
|
|
frame = cls.frame = Frame(root)
|
|
text = cls.text = Text(frame)
|
|
text.insert('1.0', code_sample)
|
|
# Need to pack for creation of code context text widget.
|
|
frame.pack(side='left', fill='both', expand=1)
|
|
text.pack(side='top', fill='both', expand=1)
|
|
cls.editor = DummyEditwin(root, frame, text)
|
|
codecontext.idleConf.userCfg = testcfg
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
codecontext.idleConf.userCfg = usercfg
|
|
cls.editor.text.delete('1.0', 'end')
|
|
del cls.editor, cls.frame, cls.text
|
|
cls.root.update_idletasks()
|
|
cls.root.destroy()
|
|
del cls.root
|
|
|
|
def setUp(self):
|
|
self.text.yview(0)
|
|
self.cc = codecontext.CodeContext(self.editor)
|
|
|
|
def tearDown(self):
|
|
if self.cc.context:
|
|
self.cc.context.destroy()
|
|
# Explicitly call __del__ to remove scheduled scripts.
|
|
self.cc.__del__()
|
|
del self.cc.context, self.cc
|
|
|
|
def test_init(self):
|
|
eq = self.assertEqual
|
|
ed = self.editor
|
|
cc = self.cc
|
|
|
|
eq(cc.editwin, ed)
|
|
eq(cc.text, ed.text)
|
|
eq(cc.textfont, ed.text['font'])
|
|
self.assertIsNone(cc.context)
|
|
eq(cc.info, [(0, -1, '', False)])
|
|
eq(cc.topvisible, 1)
|
|
eq(self.root.tk.call('after', 'info', self.cc.t1)[1], 'timer')
|
|
eq(self.root.tk.call('after', 'info', self.cc.t2)[1], 'timer')
|
|
|
|
def test_del(self):
|
|
self.cc.__del__()
|
|
with self.assertRaises(TclError) as msg:
|
|
self.root.tk.call('after', 'info', self.cc.t1)
|
|
self.assertIn("doesn't exist", msg)
|
|
with self.assertRaises(TclError) as msg:
|
|
self.root.tk.call('after', 'info', self.cc.t2)
|
|
self.assertIn("doesn't exist", msg)
|
|
# For coverage on the except. Have to delete because the
|
|
# above Tcl error is caught by after_cancel.
|
|
del self.cc.t1, self.cc.t2
|
|
self.cc.__del__()
|
|
|
|
def test_reload(self):
|
|
codecontext.CodeContext.reload()
|
|
self.assertEqual(self.cc.colors, {'background': 'lightgray',
|
|
'foreground': '#000000'})
|
|
self.assertEqual(self.cc.context_depth, 15)
|
|
|
|
def test_toggle_code_context_event(self):
|
|
eq = self.assertEqual
|
|
cc = self.cc
|
|
toggle = cc.toggle_code_context_event
|
|
|
|
# Make sure code context is off.
|
|
if cc.context:
|
|
toggle()
|
|
|
|
# Toggle on.
|
|
eq(toggle(), 'break')
|
|
self.assertIsNotNone(cc.context)
|
|
eq(cc.context['font'], cc.textfont)
|
|
eq(cc.context['fg'], cc.colors['foreground'])
|
|
eq(cc.context['bg'], cc.colors['background'])
|
|
eq(cc.context.get('1.0', 'end-1c'), '')
|
|
|
|
# Toggle off.
|
|
eq(toggle(), 'break')
|
|
self.assertIsNone(cc.context)
|
|
|
|
def test_get_context(self):
|
|
eq = self.assertEqual
|
|
gc = self.cc.get_context
|
|
|
|
# stopline must be greater than 0.
|
|
with self.assertRaises(AssertionError):
|
|
gc(1, stopline=0)
|
|
|
|
eq(gc(3), ([(2, 0, 'class C1():', 'class')], 0))
|
|
|
|
# Don't return comment.
|
|
eq(gc(4), ([(2, 0, 'class C1():', 'class')], 0))
|
|
|
|
# Two indentation levels and no comment.
|
|
eq(gc(5), ([(2, 0, 'class C1():', 'class'),
|
|
(4, 4, ' def __init__(self, a, b):', 'def')], 0))
|
|
|
|
# Only one 'def' is returned, not both at the same indent level.
|
|
eq(gc(10), ([(2, 0, 'class C1():', 'class'),
|
|
(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if')], 0))
|
|
|
|
# With 'elif', also show the 'if' even though it's at the same level.
|
|
eq(gc(11), ([(2, 0, 'class C1():', 'class'),
|
|
(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 0))
|
|
|
|
# Set stop_line to not go back to first line in source code.
|
|
# Return includes stop_line.
|
|
eq(gc(11, stopline=2), ([(2, 0, 'class C1():', 'class'),
|
|
(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 0))
|
|
eq(gc(11, stopline=3), ([(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 4))
|
|
eq(gc(11, stopline=8), ([(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 8))
|
|
|
|
# Set stop_indent to test indent level to stop at.
|
|
eq(gc(11, stopindent=4), ([(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 4))
|
|
# Check that the 'if' is included.
|
|
eq(gc(11, stopindent=8), ([(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')], 8))
|
|
|
|
def test_update_code_context(self):
|
|
eq = self.assertEqual
|
|
cc = self.cc
|
|
# Ensure code context is active.
|
|
if not cc.context:
|
|
cc.toggle_code_context_event()
|
|
|
|
# Invoke update_code_context without scrolling - nothing happens.
|
|
self.assertIsNone(cc.update_code_context())
|
|
eq(cc.info, [(0, -1, '', False)])
|
|
eq(cc.topvisible, 1)
|
|
|
|
# Scroll down to line 1.
|
|
cc.text.yview(1)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False)])
|
|
eq(cc.topvisible, 2)
|
|
eq(cc.context.get('1.0', 'end-1c'), '')
|
|
|
|
# Scroll down to line 2.
|
|
cc.text.yview(2)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False), (2, 0, 'class C1():', 'class')])
|
|
eq(cc.topvisible, 3)
|
|
eq(cc.context.get('1.0', 'end-1c'), 'class C1():')
|
|
|
|
# Scroll down to line 3. Since it's a comment, nothing changes.
|
|
cc.text.yview(3)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False), (2, 0, 'class C1():', 'class')])
|
|
eq(cc.topvisible, 4)
|
|
eq(cc.context.get('1.0', 'end-1c'), 'class C1():')
|
|
|
|
# Scroll down to line 4.
|
|
cc.text.yview(4)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False),
|
|
(2, 0, 'class C1():', 'class'),
|
|
(4, 4, ' def __init__(self, a, b):', 'def')])
|
|
eq(cc.topvisible, 5)
|
|
eq(cc.context.get('1.0', 'end-1c'), 'class C1():\n'
|
|
' def __init__(self, a, b):')
|
|
|
|
# Scroll down to line 11. Last 'def' is removed.
|
|
cc.text.yview(11)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False),
|
|
(2, 0, 'class C1():', 'class'),
|
|
(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')])
|
|
eq(cc.topvisible, 12)
|
|
eq(cc.context.get('1.0', 'end-1c'), 'class C1():\n'
|
|
' def compare(self):\n'
|
|
' if a > b:\n'
|
|
' elif a < b:')
|
|
|
|
# No scroll. No update, even though context_depth changed.
|
|
cc.update_code_context()
|
|
cc.context_depth = 1
|
|
eq(cc.info, [(0, -1, '', False),
|
|
(2, 0, 'class C1():', 'class'),
|
|
(7, 4, ' def compare(self):', 'def'),
|
|
(8, 8, ' if a > b:', 'if'),
|
|
(10, 8, ' elif a < b:', 'elif')])
|
|
eq(cc.topvisible, 12)
|
|
eq(cc.context.get('1.0', 'end-1c'), 'class C1():\n'
|
|
' def compare(self):\n'
|
|
' if a > b:\n'
|
|
' elif a < b:')
|
|
|
|
# Scroll up.
|
|
cc.text.yview(5)
|
|
cc.update_code_context()
|
|
eq(cc.info, [(0, -1, '', False),
|
|
(2, 0, 'class C1():', 'class'),
|
|
(4, 4, ' def __init__(self, a, b):', 'def')])
|
|
eq(cc.topvisible, 6)
|
|
# context_depth is 1.
|
|
eq(cc.context.get('1.0', 'end-1c'), ' def __init__(self, a, b):')
|
|
|
|
def test_jumptoline(self):
|
|
eq = self.assertEqual
|
|
cc = self.cc
|
|
jump = cc.jumptoline
|
|
|
|
if not cc.context:
|
|
cc.toggle_code_context_event()
|
|
|
|
# Empty context.
|
|
cc.text.yview(f'{2}.0')
|
|
cc.update_code_context()
|
|
eq(cc.topvisible, 2)
|
|
cc.context.mark_set('insert', '1.5')
|
|
jump()
|
|
eq(cc.topvisible, 1)
|
|
|
|
# 4 lines of context showing.
|
|
cc.text.yview(f'{12}.0')
|
|
cc.update_code_context()
|
|
eq(cc.topvisible, 12)
|
|
cc.context.mark_set('insert', '3.0')
|
|
jump()
|
|
eq(cc.topvisible, 8)
|
|
|
|
# More context lines than limit.
|
|
cc.context_depth = 2
|
|
cc.text.yview(f'{12}.0')
|
|
cc.update_code_context()
|
|
eq(cc.topvisible, 12)
|
|
cc.context.mark_set('insert', '1.0')
|
|
jump()
|
|
eq(cc.topvisible, 8)
|
|
|
|
@mock.patch.object(codecontext.CodeContext, 'update_code_context')
|
|
def test_timer_event(self, mock_update):
|
|
# Ensure code context is not active.
|
|
if self.cc.context:
|
|
self.cc.toggle_code_context_event()
|
|
self.cc.timer_event()
|
|
mock_update.assert_not_called()
|
|
|
|
# Activate code context.
|
|
self.cc.toggle_code_context_event()
|
|
self.cc.timer_event()
|
|
mock_update.assert_called()
|
|
|
|
def test_config_timer_event(self):
|
|
eq = self.assertEqual
|
|
cc = self.cc
|
|
save_font = cc.text['font']
|
|
save_colors = codecontext.CodeContext.colors
|
|
test_font = 'FakeFont'
|
|
test_colors = {'background': '#222222', 'foreground': '#ffff00'}
|
|
|
|
# Ensure code context is not active.
|
|
if cc.context:
|
|
cc.toggle_code_context_event()
|
|
|
|
# Nothing updates on inactive code context.
|
|
cc.text['font'] = test_font
|
|
codecontext.CodeContext.colors = test_colors
|
|
cc.config_timer_event()
|
|
eq(cc.textfont, save_font)
|
|
eq(cc.contextcolors, save_colors)
|
|
|
|
# Activate code context, but no change to font or color.
|
|
cc.toggle_code_context_event()
|
|
cc.text['font'] = save_font
|
|
codecontext.CodeContext.colors = save_colors
|
|
cc.config_timer_event()
|
|
eq(cc.textfont, save_font)
|
|
eq(cc.contextcolors, save_colors)
|
|
eq(cc.context['font'], save_font)
|
|
eq(cc.context['background'], save_colors['background'])
|
|
eq(cc.context['foreground'], save_colors['foreground'])
|
|
|
|
# Active code context, change font.
|
|
cc.text['font'] = test_font
|
|
cc.config_timer_event()
|
|
eq(cc.textfont, test_font)
|
|
eq(cc.contextcolors, save_colors)
|
|
eq(cc.context['font'], test_font)
|
|
eq(cc.context['background'], save_colors['background'])
|
|
eq(cc.context['foreground'], save_colors['foreground'])
|
|
|
|
# Active code context, change color.
|
|
cc.text['font'] = save_font
|
|
codecontext.CodeContext.colors = test_colors
|
|
cc.config_timer_event()
|
|
eq(cc.textfont, save_font)
|
|
eq(cc.contextcolors, test_colors)
|
|
eq(cc.context['font'], save_font)
|
|
eq(cc.context['background'], test_colors['background'])
|
|
eq(cc.context['foreground'], test_colors['foreground'])
|
|
codecontext.CodeContext.colors = save_colors
|
|
cc.config_timer_event()
|
|
|
|
|
|
class HelperFunctionText(unittest.TestCase):
|
|
|
|
def test_get_spaces_firstword(self):
|
|
get = codecontext.get_spaces_firstword
|
|
test_lines = (
|
|
(' first word', (' ', 'first')),
|
|
('\tfirst word', ('\t', 'first')),
|
|
(' \u19D4\u19D2: ', (' ', '\u19D4\u19D2')),
|
|
('no spaces', ('', 'no')),
|
|
('', ('', '')),
|
|
('# TEST COMMENT', ('', '')),
|
|
(' (continuation)', (' ', ''))
|
|
)
|
|
for line, expected_output in test_lines:
|
|
self.assertEqual(get(line), expected_output)
|
|
|
|
# Send the pattern in the call.
|
|
self.assertEqual(get(' (continuation)',
|
|
c=re.compile(r'^(\s*)([^\s]*)')),
|
|
(' ', '(continuation)'))
|
|
|
|
def test_get_line_info(self):
|
|
eq = self.assertEqual
|
|
gli = codecontext.get_line_info
|
|
lines = code_sample.splitlines()
|
|
|
|
# Line 1 is not a BLOCKOPENER.
|
|
eq(gli(lines[0]), (codecontext.INFINITY, '', False))
|
|
# Line 2 is a BLOCKOPENER without an indent.
|
|
eq(gli(lines[1]), (0, 'class C1():', 'class'))
|
|
# Line 3 is not a BLOCKOPENER and does not return the indent level.
|
|
eq(gli(lines[2]), (codecontext.INFINITY, ' # Class comment.', False))
|
|
# Line 4 is a BLOCKOPENER and is indented.
|
|
eq(gli(lines[3]), (4, ' def __init__(self, a, b):', 'def'))
|
|
# Line 8 is a different BLOCKOPENER and is indented.
|
|
eq(gli(lines[7]), (8, ' if a > b:', 'if'))
|
|
# Test tab.
|
|
eq(gli('\tif a == b:'), (1, '\tif a == b:', 'if'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|