1'''Unittests for idlelib/SearchDialogBase.py
2
3Coverage: 99%. The only thing not covered is inconsequential --
4testing skipping of suite when self.needwrapbutton is false.
5
6'''
7import unittest
8from test.test_support import requires
9from Tkinter import Tk, Toplevel, Frame ## BooleanVar, StringVar
10from idlelib import SearchEngine as se
11from idlelib import SearchDialogBase as sdb
12from idlelib.idle_test.mock_idle import Func
13##from idlelib.idle_test.mock_tk import Var
14
15# The ## imports above & following could help make some tests gui-free.# However, they currently make radiobutton tests fail.
16##def setUpModule():
17##    # Replace tk objects used to initialize se.SearchEngine.
18##    se.BooleanVar = Var
19##    se.StringVar = Var
20##
21##def tearDownModule():
22##    se.BooleanVar = BooleanVar
23##    se.StringVar = StringVar
24
25class SearchDialogBaseTest(unittest.TestCase):
26
27    @classmethod
28    def setUpClass(cls):
29        requires('gui')
30        cls.root = Tk()
31
32    @classmethod
33    def tearDownClass(cls):
34        cls.root.destroy()
35        del cls.root
36
37    def setUp(self):
38        self.engine = se.SearchEngine(self.root)  # None also seems to work
39        self.dialog = sdb.SearchDialogBase(root=self.root, engine=self.engine)
40
41    def tearDown(self):
42        self.dialog.close()
43
44    def test_open_and_close(self):
45        # open calls create_widgets, which needs default_command
46        self.dialog.default_command = None
47
48        # Since text parameter of .open is not used in base class,
49        # pass dummy 'text' instead of tk.Text().
50        self.dialog.open('text')
51        self.assertEqual(self.dialog.top.state(), 'normal')
52        self.dialog.close()
53        self.assertEqual(self.dialog.top.state(), 'withdrawn')
54
55        self.dialog.open('text', searchphrase="hello")
56        self.assertEqual(self.dialog.ent.get(), 'hello')
57        self.dialog.close()
58
59    def test_create_widgets(self):
60        self.dialog.create_entries = Func()
61        self.dialog.create_option_buttons = Func()
62        self.dialog.create_other_buttons = Func()
63        self.dialog.create_command_buttons = Func()
64
65        self.dialog.default_command = None
66        self.dialog.create_widgets()
67
68        self.assertTrue(self.dialog.create_entries.called)
69        self.assertTrue(self.dialog.create_option_buttons.called)
70        self.assertTrue(self.dialog.create_other_buttons.called)
71        self.assertTrue(self.dialog.create_command_buttons.called)
72
73    def test_make_entry(self):
74        equal = self.assertEqual
75        self.dialog.row = 0
76        self.dialog.top = Toplevel(self.root)
77        entry, label = self.dialog.make_entry("Test:", 'hello')
78        equal(label['text'], 'Test:')
79
80        self.assertIn(entry.get(), 'hello')
81        egi = entry.grid_info()
82        equal(int(egi['row']), 0)
83        equal(int(egi['column']), 1)
84        equal(int(egi['rowspan']), 1)
85        equal(int(egi['columnspan']), 1)
86        equal(self.dialog.row, 1)
87
88    def test_create_entries(self):
89        self.dialog.row = 0
90        self.engine.setpat('hello')
91        self.dialog.create_entries()
92        self.assertIn(self.dialog.ent.get(), 'hello')
93
94    def test_make_frame(self):
95        self.dialog.row = 0
96        self.dialog.top = Toplevel(self.root)
97        frame, label = self.dialog.make_frame()
98        self.assertEqual(label, '')
99        self.assertIsInstance(frame, Frame)
100
101        frame, label = self.dialog.make_frame('testlabel')
102        self.assertEqual(label['text'], 'testlabel')
103        self.assertIsInstance(frame, Frame)
104
105    def btn_test_setup(self, meth):
106        self.dialog.top = Toplevel(self.root)
107        self.dialog.row = 0
108        return meth()
109
110    def test_create_option_buttons(self):
111        e = self.engine
112        for state in (0, 1):
113            for var in (e.revar, e.casevar, e.wordvar, e.wrapvar):
114                var.set(state)
115            frame, options = self.btn_test_setup(
116                    self.dialog.create_option_buttons)
117            for spec, button in zip (options, frame.pack_slaves()):
118                var, label = spec
119                self.assertEqual(button['text'], label)
120                self.assertEqual(var.get(), state)
121                if state == 1:
122                    button.deselect()
123                else:
124                    button.select()
125                self.assertEqual(var.get(), 1 - state)
126
127    def test_create_other_buttons(self):
128        for state in (False, True):
129            var = self.engine.backvar
130            var.set(state)
131            frame, others = self.btn_test_setup(
132                self.dialog.create_other_buttons)
133            buttons = frame.pack_slaves()
134            for spec, button in zip(others, buttons):
135                val, label = spec
136                self.assertEqual(button['text'], label)
137                if val == state:
138                    # hit other button, then this one
139                    # indexes depend on button order
140                    self.assertEqual(var.get(), state)
141                    buttons[val].select()
142                    self.assertEqual(var.get(), 1 - state)
143                    buttons[1-val].select()
144                    self.assertEqual(var.get(), state)
145
146    def test_make_button(self):
147        self.dialog.top = Toplevel(self.root)
148        self.dialog.buttonframe = Frame(self.dialog.top)
149        btn = self.dialog.make_button('Test', self.dialog.close)
150        self.assertEqual(btn['text'], 'Test')
151
152    def test_create_command_buttons(self):
153        self.dialog.create_command_buttons()
154        # Look for close button command in buttonframe
155        closebuttoncommand = ''
156        for child in self.dialog.buttonframe.winfo_children():
157            if child['text'] == 'close':
158                closebuttoncommand = child['command']
159        self.assertIn('close', closebuttoncommand)
160
161
162
163if __name__ == '__main__':
164    unittest.main(verbosity=2, exit=2)
165