""" Test idlelib.browser. Coverage: 88% (Higher, because should exclude 3 lines that .coveragerc won't exclude.) """ import os.path import unittest import pyclbr from idlelib import browser, filelist from idlelib.tree import TreeNode from test.support import requires from unittest import mock from tkinter import Tk from idlelib.idle_test.mock_idle import Func from collections import deque class ModuleBrowserTest(unittest.TestCase): @classmethod def setUpClass(cls): requires('gui') cls.root = Tk() cls.root.withdraw() cls.flist = filelist.FileList(cls.root) cls.file = __file__ cls.path = os.path.dirname(cls.file) cls.module = os.path.basename(cls.file).rstrip('.py') cls.mb = browser.ModuleBrowser(cls.flist, cls.module, [cls.path], _utest=True) @classmethod def tearDownClass(cls): cls.mb.close() cls.root.destroy() del cls.root, cls.flist, cls.mb def test_init(self): mb = self.mb eq = self.assertEqual eq(mb.name, self.module) eq(mb.file, self.file) eq(mb.flist, self.flist) eq(pyclbr._modules, {}) self.assertIsInstance(mb.node, TreeNode) def test_settitle(self): mb = self.mb self.assertIn(self.module, mb.top.title()) self.assertEqual(mb.top.iconname(), 'Module Browser') def test_rootnode(self): mb = self.mb rn = mb.rootnode() self.assertIsInstance(rn, browser.ModuleBrowserTreeItem) def test_close(self): mb = self.mb mb.top.destroy = Func() mb.node.destroy = Func() mb.close() self.assertTrue(mb.top.destroy.called) self.assertTrue(mb.node.destroy.called) del mb.top.destroy, mb.node.destroy # Nested tree same as in test_pyclbr.py except for supers on C0. C1. mb = pyclbr module, fname = 'test', 'test.py' f0 = mb.Function(module, 'f0', fname, 1) f1 = mb._nest_function(f0, 'f1', 2) f2 = mb._nest_function(f1, 'f2', 3) c1 = mb._nest_class(f0, 'c1', 5) C0 = mb.Class(module, 'C0', ['base'], fname, 6) F1 = mb._nest_function(C0, 'F1', 8) C1 = mb._nest_class(C0, 'C1', 11, ['']) C2 = mb._nest_class(C1, 'C2', 12) F3 = mb._nest_function(C2, 'F3', 14) mock_pyclbr_tree = {'f0': f0, 'C0': C0} # Adjust C0.name, C1.name so tests do not depend on order. browser.transform_children(mock_pyclbr_tree, 'test') # C0(base) browser.transform_children(C0.children) # C1() # The class below checks that the calls above are correct # and that duplicate calls have no effect. class TransformChildrenTest(unittest.TestCase): def test_transform_module_children(self): eq = self.assertEqual transform = browser.transform_children # Parameter matches tree module. tcl = list(transform(mock_pyclbr_tree, 'test')) eq(tcl, [f0, C0]) eq(tcl[0].name, 'f0') eq(tcl[1].name, 'C0(base)') # Check that second call does not change suffix. tcl = list(transform(mock_pyclbr_tree, 'test')) eq(tcl[1].name, 'C0(base)') # Nothing to traverse if parameter name isn't same as tree module. tcl = list(transform(mock_pyclbr_tree, 'different name')) eq(tcl, []) def test_transform_node_children(self): eq = self.assertEqual transform = browser.transform_children # Class with two children, one name altered. tcl = list(transform(C0.children)) eq(tcl, [F1, C1]) eq(tcl[0].name, 'F1') eq(tcl[1].name, 'C1()') tcl = list(transform(C0.children)) eq(tcl[1].name, 'C1()') # Function with two children. eq(list(transform(f0.children)), [f1, c1]) class ModuleBrowserTreeItemTest(unittest.TestCase): @classmethod def setUpClass(cls): cls.mbt = browser.ModuleBrowserTreeItem(fname) def test_init(self): self.assertEqual(self.mbt.file, fname) def test_gettext(self): self.assertEqual(self.mbt.GetText(), fname) def test_geticonname(self): self.assertEqual(self.mbt.GetIconName(), 'python') def test_isexpandable(self): self.assertTrue(self.mbt.IsExpandable()) def test_listchildren(self): save_rex = browser.pyclbr.readmodule_ex save_tc = browser.transform_children browser.pyclbr.readmodule_ex = Func(result=mock_pyclbr_tree) browser.transform_children = Func(result=[f0, C0]) try: self.assertEqual(self.mbt.listchildren(), [f0, C0]) finally: browser.pyclbr.readmodule_ex = save_rex browser.transform_children = save_tc def test_getsublist(self): mbt = self.mbt mbt.listchildren = Func(result=[f0, C0]) sub0, sub1 = mbt.GetSubList() del mbt.listchildren self.assertIsInstance(sub0, browser.ChildBrowserTreeItem) self.assertIsInstance(sub1, browser.ChildBrowserTreeItem) self.assertEqual(sub0.name, 'f0') self.assertEqual(sub1.name, 'C0(base)') def test_ondoubleclick(self): mbt = self.mbt fopen = browser.file_open = mock.Mock() with mock.patch('os.path.exists', return_value=False): mbt.OnDoubleClick() fopen.assert_not_called() with mock.patch('os.path.exists', return_value=True): mbt.OnDoubleClick() fopen.assert_called() fopen.called_with(fname) del browser.file_open class ChildBrowserTreeItemTest(unittest.TestCase): @classmethod def setUpClass(cls): CBT = browser.ChildBrowserTreeItem cls.cbt_f1 = CBT(f1) cls.cbt_C1 = CBT(C1) cls.cbt_F1 = CBT(F1) @classmethod def tearDownClass(cls): del cls.cbt_C1, cls.cbt_f1, cls.cbt_F1 def test_init(self): eq = self.assertEqual eq(self.cbt_C1.name, 'C1()') self.assertFalse(self.cbt_C1.isfunction) eq(self.cbt_f1.name, 'f1') self.assertTrue(self.cbt_f1.isfunction) def test_gettext(self): self.assertEqual(self.cbt_C1.GetText(), 'class C1()') self.assertEqual(self.cbt_f1.GetText(), 'def f1(...)') def test_geticonname(self): self.assertEqual(self.cbt_C1.GetIconName(), 'folder') self.assertEqual(self.cbt_f1.GetIconName(), 'python') def test_isexpandable(self): self.assertTrue(self.cbt_C1.IsExpandable()) self.assertTrue(self.cbt_f1.IsExpandable()) self.assertFalse(self.cbt_F1.IsExpandable()) def test_getsublist(self): eq = self.assertEqual CBT = browser.ChildBrowserTreeItem f1sublist = self.cbt_f1.GetSubList() self.assertIsInstance(f1sublist[0], CBT) eq(len(f1sublist), 1) eq(f1sublist[0].name, 'f2') eq(self.cbt_F1.GetSubList(), []) def test_ondoubleclick(self): fopen = browser.file_open = mock.Mock() goto = fopen.return_value.gotoline = mock.Mock() self.cbt_F1.OnDoubleClick() fopen.assert_called() goto.assert_called() goto.assert_called_with(self.cbt_F1.obj.lineno) del browser.file_open # Failure test would have to raise OSError or AttributeError. class NestedChildrenTest(unittest.TestCase): "Test that all the nodes in a nested tree are added to the BrowserTree." def test_nested(self): queue = deque() actual_names = [] # The tree items are processed in breadth first order. # Verify that processing each sublist hits every node and # in the right order. expected_names = ['f0', 'C0(base)', 'f1', 'c1', 'F1', 'C1()', 'f2', 'C2', 'F3'] CBT = browser.ChildBrowserTreeItem queue.extend((CBT(f0), CBT(C0))) while queue: cb = queue.popleft() sublist = cb.GetSubList() queue.extend(sublist) self.assertIn(cb.name, cb.GetText()) self.assertIn(cb.GetIconName(), ('python', 'folder')) self.assertIs(cb.IsExpandable(), sublist != []) actual_names.append(cb.name) self.assertEqual(actual_names, expected_names) if __name__ == '__main__': unittest.main(verbosity=2)